Add Mock keyword to integration test cases.