added mockito to the unit test