Refactor IntegrationTests.java