Return the 'as-test' function.

After .as-test was introduced, the tests weren't updated to actually use that, instead we had a test function that didn't executed anything. This fixes that by just returning the .as-test function to use as the test function in mocha rather than wrapping that in another function.