Added tests with mockito