Added Mockito tests