Add integration test support