Mockito style mocking for wake