Use Mockito for testing