Set unit when there is no argument in a lambda
Start using token-mock in lambda tests