Update mocks to reflect Heka API changes.