Use iterated value in input generator mock