WIP - inching closer to proper iterator mock