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))
