Replaced EasyMock with mockito