Update mocking for consumer test.