def _add_to_streaming_element(self, element, path): if element not in ['mapper', 'reducer']: raise ex.NotFoundException(element, message='"%s" child cannot be added to ' 'streaming element') x.get_and_create_if_not_exist(self.doc, self.tag_name, 'streaming') x.add_text_element_to_tag(self.doc, 'streaming', element, path)
def _add_to_prepare_element(self, element, paths): if element not in ["delete", "mkdir"]: raise ex.NotFoundException(element, message='"%s" child cannot be added to ' "prepare element") prop = x.get_and_create_if_not_exist(self.doc, self.tag_name, "prepare") for path in paths: elem = xml.parseString('<%s path="%s"/>' % (element, path)) prop.appendChild(elem.firstChild)
def test_get_if_not_exist_and_add_text_element(self): doc = self.create_default_doc() x.get_and_create_if_not_exist(doc, 'test', 'tag_to_add') self.assertEqual( doc.toprettyxml(indent=" "), """<?xml version="1.0" ?> <test> <tag_to_add/> </test> """) x.add_text_element_to_tag(doc, 'tag_to_add', 'p', 'v') self.assertEqual( doc.toprettyxml(indent=" "), """<?xml version="1.0" ?> <test> <tag_to_add> <p>v</p> </tag_to_add> </test> """)
def test_get_if_not_exist_and_add_text_element(self): doc = self.create_default_doc() x.get_and_create_if_not_exist(doc, 'test', 'tag_to_add') self.assertEqual(doc.toprettyxml(indent=" "), """<?xml version="1.0" ?> <test> <tag_to_add/> </test> """) x.add_text_element_to_tag(doc, 'tag_to_add', 'p', 'v') self.assertEqual(doc.toprettyxml(indent=" "), """<?xml version="1.0" ?> <test> <tag_to_add> <p>v</p> </tag_to_add> </test> """)
def _add_to_prepare_element(self, element, paths): if element not in ['delete', 'mkdir']: raise ex.NotFoundException(element, message='"%s" child cannot be added to ' 'prepare element') prop = x.get_and_create_if_not_exist(self.doc, self.tag_name, 'prepare') for path in paths: elem = xml.parseString('<%s path="%s"/>' % (element, path)) prop.appendChild(elem.firstChild)