Make IterationModel Unit Testable with mocks