Add api mocking support