Change from EasyMock to Mockito