Added a mockito test