def test_get_children(self): next_step1 = NextStep() names = ['sub1_top_flag', 'mod1_flag1', 'Top Flag'] for name in names: self.assertIsNone(next_step1.get_children([name])) self.assertDictEqual(next_step1.get_children([]), next_step1.as_json(with_children=False)) flags = [ Flag('sub1_top_flag'), Flag(action='mod1_flag1'), Flag(action='Top Flag') ] next_step2 = NextStep(flags=flags) for i, name in enumerate(names): self.assertDictEqual(next_step2.get_children([name]), flags[i].as_json()) self.assertDictEqual(next_step2.get_children([]), next_step2.as_json(with_children=False))
def test_get_children(self): next_step1 = NextStep() names = ['', 'action1', 'action2'] for name in names: self.assertIsNone(next_step1.get_children([name])) self.assertDictEqual(next_step1.get_children([]), next_step1.as_json(with_children=False)) flags = [Flag(), Flag(action='action1'), Flag(action='action2')] next_step2 = NextStep(flags=flags) for i, name in enumerate(names): self.assertDictEqual(next_step2.get_children([name]), flags[i].as_json()) self.assertDictEqual(next_step2.get_children([]), next_step2.as_json(with_children=False)) filters = [ Filter(action='filter1'), Filter(), Filter(action='filter2') ] flags = [Flag(action='action1', filters=filters)] names = ['filter1', '', 'filter2'] next_step3 = NextStep(flags=flags) ancestries = [[name, 'action1'] for name in names] for i, ancestry in enumerate(ancestries): self.assertDictEqual(next_step3.get_children(ancestry), filters[i].as_json()) self.assertDictEqual(next_step3.get_children([]), next_step3.as_json(with_children=False))