Handle mocks with arguments.
Checks arity not that the exact value of the arg matches.