def test_single_iteration_if_table_empty(self): parser = PageParser(table_parser=None, description_parser=MagicMock()) parser._name = 'name' parser._position = 'position' parser._description = 'description' parser._stages = iter([]) generator = parser._generator() actual = list(generator) self.assertEqual([Stage(name='name', coordinates='position', description='description', tasks=[])], actual)
def test_generator_iterates_over_stages_once(self): parser = PageParser(table_parser=None, description_parser=MagicMock()) parser._name = 'cache name' parser._position = 'anchor position' parser._description = 'cache description' parser._stages = iter( [{'name': 'stage name', 'coordinates': 'stage position', 'description': 'stage description', 'tasks': {}}]) generator = parser._generator() actual = list(generator) self.assertEqual([Stage(name='cache name', coordinates='anchor position', description='cache description', tasks=[]), Stage(name='stage name', coordinates='stage position', description='stage description', tasks=[])], actual)