Support fns as arguments to mocks
since anything is a function we no longer need a special case.