Change AppTester.input(fn) to set with the fn's result instead of update