public void PrintMenu() { var pancakeIterator = _pancakeHouseMenu.CreateIterator(); var dinerIterator = _dinerMenu.CreateIterator(); Console.WriteLine("MENU\n----\nBREAKFAST"); PrintMenu(pancakeIterator); Console.WriteLine("\nLUNCH"); PrintMenu(dinerIterator); }
public void MenuIterator_GetIterator() { var pancakeHouseMenu = new PancakeHouseMenu(); var dinerMenu = new DinerMenu(); IIterator pancakeIterator = pancakeHouseMenu.CreateIterator(); IIterator dinerIterator = dinerMenu.CreateIterator(); var pancakeIteratorResult = PrintMenu(pancakeIterator); pancakeIteratorResult.OfType <IEnumerable>().Should().HaveCount(4); var dinerIteratorResult = PrintMenu(dinerIterator); dinerIteratorResult.OfType <IEnumerable>().Should().HaveCount(6); }