Move test to Mockito