Added a mock and expectation.