Added mockito test