Updated integration tests with mockito