Adding support of a function as a mock