class TestXmlWriter(unittest.TestCase):
def setUp(self):
self.writer = XmlWriter(PATH)
def tearDown(self):
self.writer.close()
os.remove(PATH)
def test_write_element_in_pieces(self):
self.writer.start('name', {'attr': 'value'}, newline=False)
self.writer.content('Some content here!!')
self.writer.end('name')
self._verify_node(None, 'name', 'Some content here!!', {'attr': 'value'})
def test_calling_content_multiple_times(self):
self.writer.start(u'robot-log', newline=False)
self.writer.content(u'Hello world!\n')
self.writer.content(u'Hi again!')
self.writer.content('\tMy name is John')
self.writer.end('robot-log')
self._verify_node(None, 'robot-log',
'Hello world!\nHi again!\tMy name is John')
def test_write_element(self):
self.writer.element('foo', 'Node\n content',
{'a1': 'attr1', 'a2': 'attr2'})
self._verify_node(None, 'foo', 'Node\n content',
{'a1': 'attr1', 'a2': 'attr2'})
def test_write_many_elements(self):
self.writer.start('root', {'version': 'test'})
self.writer.start('child1', {'my-attr': 'my value'})
self.writer.element('leaf1.1', 'leaf content', {'type': 'kw'})
self.writer.element('leaf1.2')
self.writer.end('child1')
self.writer.element('child2', attrs={'class': 'foo'})
self.writer.end('root')
root = self._get_root()
self._verify_node(root, 'root', attrs={'version': 'test'})
self._verify_node(root.find('child1'), 'child1', attrs={'my-attr': 'my value'})
self._verify_node(root.find('child1/leaf1.1'), 'leaf1.1',
'leaf content', {'type': 'kw'})
self._verify_node(root.find('child1/leaf1.2'), 'leaf1.2')
self._verify_node(root.find('child2'), 'child2', attrs={'class': 'foo'})
def test_newline_insertion(self):
self.writer.start('root')
self.writer.start('suite', {'type': 'directory_suite'})
self.writer.element('test', attrs={'name': 'my_test'}, newline=False)
self.writer.element('test', attrs={'name': 'my_2nd_test'})
self.writer.end('suite', False)
self.writer.start('suite', {'name': 'another suite'}, newline=False)
self.writer.content('Suite 2 content')
self.writer.end('suite')
self.writer.end('root')
content = self._get_content()
lines = [line for line in content.splitlines() if line != '\n']
assert_equal(len(lines), 6)
def test_none_content(self):
self.writer.element(u'robot-log', None)
self._verify_node(None, 'robot-log')
def test_content_with_invalid_command_char(self):
self.writer.element('robot-log', '\033[31m\033[32m\033[33m\033[m')
self._verify_node(None, 'robot-log', '[31m[32m[33m[m')
def test_content_with_invalid_command_char_unicode(self):
self.writer.element('robot-log', u'\x1b[31m\x1b[32m\x1b[33m\x1b[m')
self._verify_node(None, 'robot-log', '[31m[32m[33m[m')
def test_content_with_non_ascii(self):
self.writer.start('root')
self.writer.element(u'e', u'Circle is 360\xB0')
self.writer.element(u'f', u'Hyv\xE4\xE4 \xFC\xF6t\xE4')
self.writer.end('root')
root = self._get_root()
self._verify_node(root.find('e'), 'e', u'Circle is 360\xB0')
self._verify_node(root.find('f'), 'f', u'Hyv\xE4\xE4 \xFC\xF6t\xE4')
def test_content_with_entities(self):
self.writer.element(u'robot-log', 'Me, Myself & I > you')
self._verify_content('Me, Myself & I > you')
def test_remove_illegal_chars(self):
assert_equals(self.writer._escape(u'\x1b[31m'), '[31m')
assert_equals(self.writer._escape(u'\x00'), '')
def test_ioerror_when_file_is_invalid(self):
assert_raises(IOError, XmlWriter, os.path.dirname(__file__))
def test_custom_encoding(self):
self.writer.close()
self.writer = XmlWriter(PATH, encoding='ISO-8859-1')
self.writer.element('test', u'hyv\xe4')
self._verify_content('encoding="ISO-8859-1"')
self._verify_node(None, 'test', u'hyv\xe4')
def _verify_node(self, node, name, text=None, attrs={}):
if node is None:
node = self._get_root()
assert_equals(node.tag, name)
if text is not None:
assert_equals(node.text, text)
assert_equals(node.attrib, attrs)
def _verify_content(self, expected):
content = self._get_content()
assert_true(expected in content,
'Failed to find:\n%s\n\nfrom:\n%s' % (expected, content))
def _get_root(self):
self.writer.close()
with ETSource(PATH) as source:
return ET.parse(source).getroot()
def _get_content(self):
self.writer.close()
with open(PATH) as f:
return f.read()
class TestXmlWriter(unittest.TestCase):
def setUp(self):
self.writer = XmlWriter(PATH)
def tearDown(self):
self.writer.close()
os.remove(PATH)
def test_write_element_in_pieces(self):
self.writer.start('name', {'attr': 'value'}, newline=False)
self.writer.content('Some content here!!')
self.writer.end('name')
self._verify_node(None, 'name', 'Some content here!!',
{'attr': 'value'})
def test_calling_content_multiple_times(self):
self.writer.start(u'robot-log', newline=False)
self.writer.content(u'Hello world!\n')
self.writer.content(u'Hi again!')
self.writer.content('\tMy name is John')
self.writer.end('robot-log')
self._verify_node(None, 'robot-log',
'Hello world!\nHi again!\tMy name is John')
def test_write_element(self):
self.writer.element('foo', 'Node\n content', {
'a1': 'attr1',
'a2': 'attr2'
})
self._verify_node(None, 'foo', 'Node\n content', {
'a1': 'attr1',
'a2': 'attr2'
})
def test_write_many_elements(self):
self.writer.start('root', {'version': 'test'})
self.writer.start('child1', {'my-attr': 'my value'})
self.writer.element('leaf1.1', 'leaf content', {'type': 'kw'})
self.writer.element('leaf1.2')
self.writer.end('child1')
self.writer.element('child2', attrs={'class': 'foo'})
self.writer.end('root')
root = self._get_root()
self._verify_node(root, 'root', attrs={'version': 'test'})
self._verify_node(root.find('child1'),
'child1',
attrs={'my-attr': 'my value'})
self._verify_node(root.find('child1/leaf1.1'), 'leaf1.1',
'leaf content', {'type': 'kw'})
self._verify_node(root.find('child1/leaf1.2'), 'leaf1.2')
self._verify_node(root.find('child2'),
'child2',
attrs={'class': 'foo'})
def test_newline_insertion(self):
self.writer.start('root')
self.writer.start('suite', {'type': 'directory_suite'})
self.writer.element('test', attrs={'name': 'my_test'}, newline=False)
self.writer.element('test', attrs={'name': 'my_2nd_test'})
self.writer.end('suite', False)
self.writer.start('suite', {'name': 'another suite'}, newline=False)
self.writer.content('Suite 2 content')
self.writer.end('suite')
self.writer.end('root')
content = self._get_content()
lines = [line for line in content.splitlines() if line != '\n']
assert_equal(len(lines), 6)
def test_none_content(self):
self.writer.element(u'robot-log', None)
self._verify_node(None, 'robot-log')
def test_content_with_invalid_command_char(self):
self.writer.element('robot-log', '\033[31m\033[32m\033[33m\033[m')
self._verify_node(None, 'robot-log', '[31m[32m[33m[m')
def test_content_with_invalid_command_char_unicode(self):
self.writer.element('robot-log', u'\x1b[31m\x1b[32m\x1b[33m\x1b[m')
self._verify_node(None, 'robot-log', '[31m[32m[33m[m')
def test_content_with_non_ascii(self):
self.writer.start('root')
self.writer.element(u'e', u'Circle is 360\xB0')
self.writer.element(u'f', u'Hyv\xE4\xE4 \xFC\xF6t\xE4')
self.writer.end('root')
root = self._get_root()
self._verify_node(root.find('e'), 'e', u'Circle is 360\xB0')
self._verify_node(root.find('f'), 'f', u'Hyv\xE4\xE4 \xFC\xF6t\xE4')
def test_content_with_entities(self):
self.writer.element(u'robot-log', 'Me, Myself & I > you')
self._verify_content('Me, Myself & I > you')
def test_remove_illegal_chars(self):
assert_equals(self.writer._escape(u'\x1b[31m'), '[31m')
assert_equals(self.writer._escape(u'\x00'), '')
def test_ioerror_when_file_is_invalid(self):
assert_raises(IOError, XmlWriter, os.path.dirname(__file__))
def test_custom_encoding(self):
self.writer.close()
self.writer = XmlWriter(PATH, encoding='ISO-8859-1')
self.writer.element('test', u'hyv\xe4')
self._verify_content('encoding="ISO-8859-1"')
self._verify_node(None, 'test', u'hyv\xe4')
def _verify_node(self, node, name, text=None, attrs={}):
if node is None:
node = self._get_root()
assert_equals(node.tag, name)
if text is not None:
assert_equals(node.text, text)
assert_equals(node.attrib, attrs)
def _verify_content(self, expected):
content = self._get_content()
assert_true(expected in content,
'Failed to find:\n%s\n\nfrom:\n%s' % (expected, content))
def _get_root(self):
self.writer.close()
with ETSource(PATH) as source:
return ET.parse(source).getroot()
def _get_content(self):
self.writer.close()
with open(PATH) as f:
return f.read()
class TestXmlWriter(unittest.TestCase):
def setUp(self):
self.writer = XmlWriter(PATH, encoding="UTF-8")
def tearDown(self):
self.writer.close()
os.remove(PATH)
def test_write_element_in_pieces(self):
self.writer.start("name", {"attr": "value"}, newline=False)
self.writer.content("Some content here!!")
self.writer.end("name")
self.writer.close()
self._verify_node(None, "name", "Some content here!!", {"attr": "value"})
def test_calling_content_multiple_times(self):
self.writer.start(u"robot-log", newline=False)
self.writer.content(u"Hello world!\n")
self.writer.content(u"Hi again!")
self.writer.content("\tMy name is John")
self.writer.end("robot-log")
self.writer.close()
self._verify_node(None, "robot-log", "Hello world!\nHi again!\tMy name is John")
def test_write_element(self):
self.writer.element("foo", "Node\n content", {"a1": "attr1", "a2": "attr2"})
self.writer.close()
self._verify_node(None, "foo", "Node\n content", {"a1": "attr1", "a2": "attr2"})
def test_write_many_elements(self):
self.writer.start("root", {"version": "test"})
self.writer.start("child1", {"my-attr": "my value"})
self.writer.element("leaf1.1", "leaf content", {"type": "kw"})
self.writer.element("leaf1.2")
self.writer.end("child1")
self.writer.element("child2", attrs={"class": "foo"})
self.writer.end("root")
self.writer.close()
with ETSource(PATH) as source:
root = ET.parse(source).getroot()
self._verify_node(root, "root", attrs={"version": "test"})
self._verify_node(root.find("child1"), "child1", attrs={"my-attr": "my value"})
self._verify_node(root.find("child1/leaf1.1"), "leaf1.1", "leaf content", {"type": "kw"})
self._verify_node(root.find("child1/leaf1.2"), "leaf1.2")
self._verify_node(root.find("child2"), "child2", attrs={"class": "foo"})
def test_newline_insertion(self):
self.writer.start("root")
self.writer.start("suite", {"type": "directory_suite"})
self.writer.element("test", attrs={"name": "my_test"}, newline=False)
self.writer.element("test", attrs={"name": "my_2nd_test"})
self.writer.end("suite", False)
self.writer.start("suite", {"name": "another suite"}, newline=False)
self.writer.content("Suite 2 content")
self.writer.end("suite")
self.writer.end("root")
self.writer.close()
with open(PATH) as file:
lines = [line for line in file if line != "\n"]
assert_equal(len(lines), 6)
def test_none_content(self):
self.writer.element(u"robot-log", None)
self.writer.close()
self._verify_node(None, "robot-log")
def test_content_with_invalid_command_char(self):
self.writer.element("robot-log", "\033[31m\033[32m\033[33m\033[m")
self.writer.close()
self._verify_node(None, "robot-log", "[31m[32m[33m[m")
def test_content_with_invalid_command_char_unicode(self):
self.writer.element("robot-log", u"\x1b[31m\x1b[32m\x1b[33m\x1b[m")
self.writer.close()
self._verify_node(None, "robot-log", "[31m[32m[33m[m")
def test_content_with_unicode(self):
self.writer.start("root")
self.writer.element(u"e", u"Circle is 360\u00B0")
self.writer.element(u"f", u"Hyv\u00E4\u00E4 \u00FC\u00F6t\u00E4")
self.writer.end("root")
self.writer.close()
with ETSource(PATH) as source:
root = ET.parse(source).getroot()
self._verify_node(root.find("e"), "e", u"Circle is 360\u00B0")
self._verify_node(root.find("f"), "f", u"Hyv\u00E4\u00E4 \u00FC\u00F6t\u00E4")
def test_content_with_entities(self):
self.writer.element(u"robot-log", "Me, Myself & I > you")
self.writer.close()
with open(PATH) as file:
content = file.read()
assert_true("Me, Myself & I > you" in content)
def test_remove_illegal_chars(self):
assert_equals(self.writer._escape(u"\x1b[31m"), "[31m")
assert_equals(self.writer._escape(u"\x00"), "")
def test_ioerror_when_file_is_invalid(self):
assert_raises(IOError, XmlWriter, os.path.dirname(__file__))
def _verify_node(self, node, name, text=None, attrs={}):
if node is None:
with ETSource(PATH) as source:
node = ET.parse(source).getroot()
assert_equals(node.tag, name)
if text is not None:
assert_equals(node.text, text)
assert_equals(node.attrib, attrs)
