1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
|
#!/usr/bin/python
# -*- coding: UTF-8 -*-
import unittest
import libxml2
from pyxmpp import xmlextra
from stream_reader import xml_elements_equal
input_xml = """<?xml version="1.0" ?>
<root xmlns="http://pyxmpp.jajcus.net/xmlns/test" xmlns:prefix="http://pyxmpp.jajcus.net/xmlns/test1">
<a> <a1/> <a2/> </a>
<b xmlns="http://pyxmpp.jajcus.net/xmlns/test2">
<c/>
<prefix:d/>
<g xmlns="http://pyxmpp.jajcus.net/xmlns/test3" type="ble"/>
</b>
<prefix:e/>
<f/>
</root>
"""
input_doc = libxml2.parseDoc(input_xml)
input_root = input_doc.getRootElement()
input_xml2 = """<?xml version="1.0" ?>
<root xmlns:prefix="http://pyxmpp.jajcus.net/xmlns/test1">
<a> <a1/> <a2/> </a>
<b xmlns="http://pyxmpp.jajcus.net/xmlns/test2">
<c/>
<prefix:d/>
<g xmlns="http://pyxmpp.jajcus.net/xmlns/test3" type="ble"/>
</b>
<prefix:e/>
<f/>
</root>
"""
input_doc2 = libxml2.parseDoc(input_xml2)
input_root2 = input_doc2.getRootElement()
output_xml = """<?xml version="1.0" ?>
<root xmlns="http://pyxmpp.jajcus.net/xmlns/common" xmlns:prefix="http://pyxmpp.jajcus.net/xmlns/test1">
<a> <a1/> <a2/> </a>
<b xmlns="http://pyxmpp.jajcus.net/xmlns/test2">
<c/>
<prefix:d/>
<g xmlns="http://pyxmpp.jajcus.net/xmlns/test3" type="ble"/>
</b>
<prefix:e/>
<f/>
</root>
"""
output_doc = libxml2.parseDoc(output_xml)
output_root = output_doc.getRootElement()
class TestReplaceNs(unittest.TestCase):
def test_replace_ns(self):
doc = libxml2.newDoc("1.0")
root = doc.newChild(None, "root", None)
common_ns = root.newNs(xmlextra.COMMON_NS, None)
root.setNs(common_ns)
doc.setRootElement(root)
n = input_doc.getRootElement()
try:
input_ns = n.ns()
except libxml2.treeError:
input_ns = None
n = n.children
while n:
n1 = n.docCopyNode(doc, 1)
root.addChild(n1)
if n1.type == 'element':
try:
n1_ns = n1.ns()
except libxml2.treeError:
ns1_ns = None
if n1_ns.content == input_ns.content:
xmlextra.replace_ns(n1, n1_ns, common_ns)
n = n.next
self.failUnless(xml_elements_equal(root, output_root))
def test_replace_null_ns(self):
doc = libxml2.newDoc("1.0")
root = doc.newChild(None, "root", None)
common_ns = root.newNs(xmlextra.COMMON_NS, None)
root.setNs(common_ns)
doc.setRootElement(root)
n = input_doc2.getRootElement()
try:
input_ns = n.ns()
except libxml2.treeError:
input_ns = None
n = n.children
while n:
n1 = n.docCopyNode(doc, 1)
root.addChild(n1)
if n1.type == 'element':
try:
n1_ns = n1.ns()
except libxml2.treeError:
n1_ns = None
if n1_ns is None:
xmlextra.replace_ns(n1, n1_ns, common_ns)
n = n.next
self.failUnless(xml_elements_equal(root, output_root))
def test_safe_serialize(self):
s1 = """<a xmlns="http://pyxmpp.jajcus.net/xmlns/test"><b a1="v1" xmlns="http://pyxmpp.jajcus.net/xmlns/test1" a2="v2"/></a>"""
doc1 = libxml2.parseDoc(s1)
root1 = doc1.getRootElement()
el1 = root1.children
try:
root1_ns = root1.ns()
except libxml2.treeError:
root1_ns = None
el1.setNs(root1_ns)
#s = el1.serialize()
s = xmlextra.safe_serialize(el1)
s2 = '<a xmlns="http://pyxmpp.jajcus.net/xmlns/test">%s</a>' % (s,)
doc2 = libxml2.parseDoc(s2)
root2 = doc2.getRootElement()
self.failUnless(xml_elements_equal(root1, root2))
def suite():
suite = unittest.TestSuite()
suite.addTest(unittest.makeSuite(TestReplaceNs))
return suite
if __name__ == '__main__':
unittest.TextTestRunner(verbosity=2).run(suite())
# vi: sts=4 et sw=4
|
