Use Mockito instead of custom mock object