Add a mock for the API tests