adding some tests

Author: Arzaroth Lekva

README.md

@@ -6,20 +6,21 @@ A library providing python bindings for rapidxml
 
     import rapidxml
 
-    r = rapidxml.RapidXml("<test/><test2>foo</test2><test></test>") # parsing from string
+    r = rapidxml.RapidXml(b"<test/><test2>foo</test2><test></test>") # parsing from bytes
     test = r.first_node("test") # get first node named test
     test.name = "foo" # changing node's name to foo
     r.first_node("test2").value = "bar" # changing node's value to bar
 
-    print(str(r)) # will output prettified string of the xml document
+    print(str(r)) # will output a prettified string of the xml document
+    print(r.unparse(pretty=False, raw=True)) # will output the xml document as flat bytes
     print(test) # also works for nodes
 
     with open('dump.xml', 'w') as f:
         f.write(str(r))
     r = rapidxml.RapidXml("dump.xml", True) # loading from file
 
-    assert(str(r) == r.unparse(True)) # is always True
-    assert(repr(r) == r.unparse()) # also always True, returns flat version
+    assert(str(r) == r.unparse(True, False)) # is always True
+    assert(repr(r) == r.unparse(pretty=False, raw=False)) # also always True
 
 
 ### Install

tests/conftest.py

@@ -11,7 +11,7 @@
 
 @pytest.fixture(scope="function")
 def init_rapidxml():
-    r = rapidxml.RapidXml(bytearray(b"""
+    r = rapidxml.RapidXml(b"""
 <root>
   <test attr1="one" attr2="two" attr3="three" />
   <test2>
@@ -20,5 +20,5 @@ def init_rapidxml():
     <node id="3"/>
   </test2>
   <test>some text</test>
-</root>"""))
+</root>""")
     return r

tests/test_attributes.py

@@ -21,3 +21,35 @@ def test_last_attribute(init_rapidxml):
     attr2 = test.last_attribute("attr2")
     assert attr2.name == "attr2"
     assert attr2.value == "two"
+
+def test_append_attribute(init_rapidxml):
+    root = init_rapidxml.first_node()
+    root.append_attribute(init_rapidxml.allocate_attribute())
+    assert root.last_attribute().name == ""
+    assert root.last_attribute().value == ""
+    root.append_attribute(init_rapidxml.allocate_attribute("test"))
+    assert root.last_attribute().name == "test"
+    assert root.last_attribute().value == ""
+    root.append_attribute(init_rapidxml.allocate_attribute(value="test"))
+    assert root.last_attribute().name == ""
+    assert root.last_attribute().value == "test"
+    root.append_attribute(init_rapidxml.allocate_attribute("test", "test"))
+    assert root.last_attribute().name == "test"
+    assert root.last_attribute().value == "test"
+
+def test_prepend_attribute(init_rapidxml):
+    root = init_rapidxml.first_node()
+    root.prepend_attribute(init_rapidxml.allocate_attribute("test", "test"))
+    assert root.first_attribute().name == "test"
+    assert root.first_attribute().value == "test"
+    root.prepend_attribute(init_rapidxml.allocate_attribute("test2", "test2"))
+    assert root.first_attribute().name == "test2"
+    assert root.first_attribute().value == "test2"
+
+def test_insert_attribute(init_rapidxml):
+    test = init_rapidxml.first_node().first_node()
+    attr2 = test.first_attribute("attr2")
+    test.insert_attribute(attr2, init_rapidxml.allocate_attribute("test", "test"))
+    test = attr2.previous_attribute()
+    assert test.name == "test"
+    assert test.value == "test"

tests/test_basic.py

@@ -22,7 +22,7 @@ def test_parse(init_rapidxml):
         data = init_rapidxml.unparse().encode('utf-8')
     except UnicodeDecodeError:
         data = init_rapidxml.unparse()
-    r.parse(bytearray(data))
+    r.parse(data)
     assert str(r) == str(init_rapidxml)
 
 def test_parse_from_file(init_rapidxml, tmpdir):

tests/test_utf8.py

@@ -0,0 +1,21 @@
+#!/usr/bin/env python
+# -*- coding: utf-8 -*-
+#
+# File: test_utf8.py
+# by Arzaroth Lekva
+# [email protected]
+#
+
+import sys
+import rapidxml
+
+def test_utf8():
+    data = "<root>éè</root>"
+    if sys.version_info.major >= 3:
+        data = data.encode('utf-8')
+    r = rapidxml.RapidXml(data)
+    assert str(r) == r.unparse(pretty=True)
+    assert repr(r) == r.unparse()
+    assert data == r.unparse(raw=True)
+    r.first_node().value = b'\x85'
+    assert r.unparse(raw=True).decode('cp1252').encode('utf-8') == b"<root>\xe2\x80\xa6</root>"