Add some mocking helpers