Using mockito for tests