Creating tests using Mockito to mock DB