def test_remove(self): element = etree.Element("element", attr="value") alert._update_optional_attribute(element, "attr", "") self.assertTrue(element.get("attr") is None)
def test_update(self): element = etree.Element("element", attr="value") alert._update_optional_attribute(element, "attr", "value1") self.assertEqual(element.get("attr"), "value1")