def test_to_xml_error(self):
flags = [Flag(action='Top Flag'), Flag(action='mod1_flag1')]
next_step = NextStep(name='name', flags=flags)
xml = next_step.to_xml(tag='error')
self.assertEqual(xml.tag, 'error')
self.assertEqual(xml.get('step'), 'name')
self.assertEqual(len(xml.findall('flag')), 2)
def test_to_xml_with_status(self):
next_step = NextStep(name='name', status='test_status')
xml = next_step.to_xml()
self.assertEqual(xml.tag, 'next')
self.assertEqual(xml.get('step'), 'name')
status_xml = xml.findall('status')
self.assertEqual(len(status_xml), 1)
self.assertEqual(status_xml[0].text, 'test_status')
self.assertEqual(len(xml.findall('flag')), 0)
def test_to_xml_none(self):
next_step = NextStep()
next_step.name = None
self.assertIsNone(next_step.to_xml())
def test_to_xml_error_no_flags(self):
next_step = NextStep(name='name')
xml = next_step.to_xml(tag='error')
self.assertEqual(xml.tag, 'error')
self.assertEqual(xml.get('step'), 'name')
self.assertEqual(len(xml.findall('flag')), 0)
