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)