Add a mock integration for testing