Modified how mock expectations are set to make it clearer what is passed
to setMockExpectations helper function