Refactored the IntegrationTests class.