Commit | Line | Data |
---|---|---|
86530b38 AT |
1 | # test for xml.dom.minidom |
2 | ||
3 | import os | |
4 | import sys | |
5 | import pickle | |
6 | import traceback | |
7 | from StringIO import StringIO | |
8 | from test.test_support import verbose | |
9 | ||
10 | import xml.dom | |
11 | import xml.dom.minidom | |
12 | import xml.parsers.expat | |
13 | ||
14 | from xml.dom.minidom import parse, Node, Document, parseString | |
15 | from xml.dom.minidom import getDOMImplementation | |
16 | ||
17 | ||
18 | if __name__ == "__main__": | |
19 | base = sys.argv[0] | |
20 | else: | |
21 | base = __file__ | |
22 | tstfile = os.path.join(os.path.dirname(base), "test"+os.extsep+"xml") | |
23 | del base | |
24 | ||
25 | def confirm(test, testname = "Test"): | |
26 | if not test: | |
27 | print "Failed " + testname | |
28 | raise Exception | |
29 | ||
30 | def testParseFromFile(): | |
31 | dom = parse(StringIO(open(tstfile).read())) | |
32 | dom.unlink() | |
33 | confirm(isinstance(dom,Document)) | |
34 | ||
35 | def testGetElementsByTagName(): | |
36 | dom = parse(tstfile) | |
37 | confirm(dom.getElementsByTagName("LI") == \ | |
38 | dom.documentElement.getElementsByTagName("LI")) | |
39 | dom.unlink() | |
40 | ||
41 | def testInsertBefore(): | |
42 | dom = parseString("<doc><foo/></doc>") | |
43 | root = dom.documentElement | |
44 | elem = root.childNodes[0] | |
45 | nelem = dom.createElement("element") | |
46 | root.insertBefore(nelem, elem) | |
47 | confirm(len(root.childNodes) == 2 | |
48 | and root.childNodes.length == 2 | |
49 | and root.childNodes[0] is nelem | |
50 | and root.childNodes.item(0) is nelem | |
51 | and root.childNodes[1] is elem | |
52 | and root.childNodes.item(1) is elem | |
53 | and root.firstChild is nelem | |
54 | and root.lastChild is elem | |
55 | and root.toxml() == "<doc><element/><foo/></doc>" | |
56 | , "testInsertBefore -- node properly placed in tree") | |
57 | nelem = dom.createElement("element") | |
58 | root.insertBefore(nelem, None) | |
59 | confirm(len(root.childNodes) == 3 | |
60 | and root.childNodes.length == 3 | |
61 | and root.childNodes[1] is elem | |
62 | and root.childNodes.item(1) is elem | |
63 | and root.childNodes[2] is nelem | |
64 | and root.childNodes.item(2) is nelem | |
65 | and root.lastChild is nelem | |
66 | and nelem.previousSibling is elem | |
67 | and root.toxml() == "<doc><element/><foo/><element/></doc>" | |
68 | , "testInsertBefore -- node properly placed in tree") | |
69 | nelem2 = dom.createElement("bar") | |
70 | root.insertBefore(nelem2, nelem) | |
71 | confirm(len(root.childNodes) == 4 | |
72 | and root.childNodes.length == 4 | |
73 | and root.childNodes[2] is nelem2 | |
74 | and root.childNodes.item(2) is nelem2 | |
75 | and root.childNodes[3] is nelem | |
76 | and root.childNodes.item(3) is nelem | |
77 | and nelem2.nextSibling is nelem | |
78 | and nelem.previousSibling is nelem2 | |
79 | and root.toxml() == "<doc><element/><foo/><bar/><element/></doc>" | |
80 | , "testInsertBefore -- node properly placed in tree") | |
81 | dom.unlink() | |
82 | ||
83 | def _create_fragment_test_nodes(): | |
84 | dom = parseString("<doc/>") | |
85 | orig = dom.createTextNode("original") | |
86 | c1 = dom.createTextNode("foo") | |
87 | c2 = dom.createTextNode("bar") | |
88 | c3 = dom.createTextNode("bat") | |
89 | dom.documentElement.appendChild(orig) | |
90 | frag = dom.createDocumentFragment() | |
91 | frag.appendChild(c1) | |
92 | frag.appendChild(c2) | |
93 | frag.appendChild(c3) | |
94 | return dom, orig, c1, c2, c3, frag | |
95 | ||
96 | def testInsertBeforeFragment(): | |
97 | dom, orig, c1, c2, c3, frag = _create_fragment_test_nodes() | |
98 | dom.documentElement.insertBefore(frag, None) | |
99 | confirm(tuple(dom.documentElement.childNodes) == (orig, c1, c2, c3), | |
100 | "insertBefore(<fragment>, None)") | |
101 | frag.unlink() | |
102 | dom.unlink() | |
103 | # | |
104 | dom, orig, c1, c2, c3, frag = _create_fragment_test_nodes() | |
105 | dom.documentElement.insertBefore(frag, orig) | |
106 | confirm(tuple(dom.documentElement.childNodes) == (c1, c2, c3, orig), | |
107 | "insertBefore(<fragment>, orig)") | |
108 | frag.unlink() | |
109 | dom.unlink() | |
110 | ||
111 | def testAppendChild(): | |
112 | dom = parse(tstfile) | |
113 | dom.documentElement.appendChild(dom.createComment(u"Hello")) | |
114 | confirm(dom.documentElement.childNodes[-1].nodeName == "#comment") | |
115 | confirm(dom.documentElement.childNodes[-1].data == "Hello") | |
116 | dom.unlink() | |
117 | ||
118 | def testAppendChildFragment(): | |
119 | dom, orig, c1, c2, c3, frag = _create_fragment_test_nodes() | |
120 | dom.documentElement.appendChild(frag) | |
121 | confirm(tuple(dom.documentElement.childNodes) == (orig, c1, c2, c3), | |
122 | "appendChild(<fragment>)") | |
123 | frag.unlink() | |
124 | dom.unlink() | |
125 | ||
126 | def testReplaceChildFragment(): | |
127 | dom, orig, c1, c2, c3, frag = _create_fragment_test_nodes() | |
128 | dom.documentElement.replaceChild(frag, orig) | |
129 | orig.unlink() | |
130 | confirm(tuple(dom.documentElement.childNodes) == (c1, c2, c3), | |
131 | "replaceChild(<fragment>)") | |
132 | frag.unlink() | |
133 | dom.unlink() | |
134 | ||
135 | def testLegalChildren(): | |
136 | dom = Document() | |
137 | elem = dom.createElement('element') | |
138 | text = dom.createTextNode('text') | |
139 | ||
140 | try: dom.appendChild(text) | |
141 | except xml.dom.HierarchyRequestErr: pass | |
142 | else: | |
143 | print "dom.appendChild didn't raise HierarchyRequestErr" | |
144 | ||
145 | dom.appendChild(elem) | |
146 | try: dom.insertBefore(text, elem) | |
147 | except xml.dom.HierarchyRequestErr: pass | |
148 | else: | |
149 | print "dom.appendChild didn't raise HierarchyRequestErr" | |
150 | ||
151 | try: dom.replaceChild(text, elem) | |
152 | except xml.dom.HierarchyRequestErr: pass | |
153 | else: | |
154 | print "dom.appendChild didn't raise HierarchyRequestErr" | |
155 | ||
156 | nodemap = elem.attributes | |
157 | try: nodemap.setNamedItem(text) | |
158 | except xml.dom.HierarchyRequestErr: pass | |
159 | else: | |
160 | print "NamedNodeMap.setNamedItem didn't raise HierarchyRequestErr" | |
161 | ||
162 | try: nodemap.setNamedItemNS(text) | |
163 | except xml.dom.HierarchyRequestErr: pass | |
164 | else: | |
165 | print "NamedNodeMap.setNamedItemNS didn't raise HierarchyRequestErr" | |
166 | ||
167 | elem.appendChild(text) | |
168 | dom.unlink() | |
169 | ||
170 | def testNamedNodeMapSetItem(): | |
171 | dom = Document() | |
172 | elem = dom.createElement('element') | |
173 | attrs = elem.attributes | |
174 | attrs["foo"] = "bar" | |
175 | a = attrs.item(0) | |
176 | confirm(a.ownerDocument is dom, | |
177 | "NamedNodeMap.__setitem__() sets ownerDocument") | |
178 | confirm(a.ownerElement is elem, | |
179 | "NamedNodeMap.__setitem__() sets ownerElement") | |
180 | confirm(a.value == "bar", | |
181 | "NamedNodeMap.__setitem__() sets value") | |
182 | confirm(a.nodeValue == "bar", | |
183 | "NamedNodeMap.__setitem__() sets nodeValue") | |
184 | elem.unlink() | |
185 | dom.unlink() | |
186 | ||
187 | def testNonZero(): | |
188 | dom = parse(tstfile) | |
189 | confirm(dom)# should not be zero | |
190 | dom.appendChild(dom.createComment("foo")) | |
191 | confirm(not dom.childNodes[-1].childNodes) | |
192 | dom.unlink() | |
193 | ||
194 | def testUnlink(): | |
195 | dom = parse(tstfile) | |
196 | dom.unlink() | |
197 | ||
198 | def testElement(): | |
199 | dom = Document() | |
200 | dom.appendChild(dom.createElement("abc")) | |
201 | confirm(dom.documentElement) | |
202 | dom.unlink() | |
203 | ||
204 | def testAAA(): | |
205 | dom = parseString("<abc/>") | |
206 | el = dom.documentElement | |
207 | el.setAttribute("spam", "jam2") | |
208 | confirm(el.toxml() == '<abc spam="jam2"/>', "testAAA") | |
209 | a = el.getAttributeNode("spam") | |
210 | confirm(a.ownerDocument is dom, | |
211 | "setAttribute() sets ownerDocument") | |
212 | confirm(a.ownerElement is dom.documentElement, | |
213 | "setAttribute() sets ownerElement") | |
214 | dom.unlink() | |
215 | ||
216 | def testAAB(): | |
217 | dom = parseString("<abc/>") | |
218 | el = dom.documentElement | |
219 | el.setAttribute("spam", "jam") | |
220 | el.setAttribute("spam", "jam2") | |
221 | confirm(el.toxml() == '<abc spam="jam2"/>', "testAAB") | |
222 | dom.unlink() | |
223 | ||
224 | def testAddAttr(): | |
225 | dom = Document() | |
226 | child = dom.appendChild(dom.createElement("abc")) | |
227 | ||
228 | child.setAttribute("def", "ghi") | |
229 | confirm(child.getAttribute("def") == "ghi") | |
230 | confirm(child.attributes["def"].value == "ghi") | |
231 | ||
232 | child.setAttribute("jkl", "mno") | |
233 | confirm(child.getAttribute("jkl") == "mno") | |
234 | confirm(child.attributes["jkl"].value == "mno") | |
235 | ||
236 | confirm(len(child.attributes) == 2) | |
237 | ||
238 | child.setAttribute("def", "newval") | |
239 | confirm(child.getAttribute("def") == "newval") | |
240 | confirm(child.attributes["def"].value == "newval") | |
241 | ||
242 | confirm(len(child.attributes) == 2) | |
243 | dom.unlink() | |
244 | ||
245 | def testDeleteAttr(): | |
246 | dom = Document() | |
247 | child = dom.appendChild(dom.createElement("abc")) | |
248 | ||
249 | confirm(len(child.attributes) == 0) | |
250 | child.setAttribute("def", "ghi") | |
251 | confirm(len(child.attributes) == 1) | |
252 | del child.attributes["def"] | |
253 | confirm(len(child.attributes) == 0) | |
254 | dom.unlink() | |
255 | ||
256 | def testRemoveAttr(): | |
257 | dom = Document() | |
258 | child = dom.appendChild(dom.createElement("abc")) | |
259 | ||
260 | child.setAttribute("def", "ghi") | |
261 | confirm(len(child.attributes) == 1) | |
262 | child.removeAttribute("def") | |
263 | confirm(len(child.attributes) == 0) | |
264 | ||
265 | dom.unlink() | |
266 | ||
267 | def testRemoveAttrNS(): | |
268 | dom = Document() | |
269 | child = dom.appendChild( | |
270 | dom.createElementNS("http://www.python.org", "python:abc")) | |
271 | child.setAttributeNS("http://www.w3.org", "xmlns:python", | |
272 | "http://www.python.org") | |
273 | child.setAttributeNS("http://www.python.org", "python:abcattr", "foo") | |
274 | confirm(len(child.attributes) == 2) | |
275 | child.removeAttributeNS("http://www.python.org", "abcattr") | |
276 | confirm(len(child.attributes) == 1) | |
277 | ||
278 | dom.unlink() | |
279 | ||
280 | def testRemoveAttributeNode(): | |
281 | dom = Document() | |
282 | child = dom.appendChild(dom.createElement("foo")) | |
283 | child.setAttribute("spam", "jam") | |
284 | confirm(len(child.attributes) == 1) | |
285 | node = child.getAttributeNode("spam") | |
286 | child.removeAttributeNode(node) | |
287 | confirm(len(child.attributes) == 0 | |
288 | and child.getAttributeNode("spam") is None) | |
289 | ||
290 | dom.unlink() | |
291 | ||
292 | def testChangeAttr(): | |
293 | dom = parseString("<abc/>") | |
294 | el = dom.documentElement | |
295 | el.setAttribute("spam", "jam") | |
296 | confirm(len(el.attributes) == 1) | |
297 | el.setAttribute("spam", "bam") | |
298 | # Set this attribute to be an ID and make sure that doesn't change | |
299 | # when changing the value: | |
300 | el.setIdAttribute("spam") | |
301 | confirm(len(el.attributes) == 1 | |
302 | and el.attributes["spam"].value == "bam" | |
303 | and el.attributes["spam"].nodeValue == "bam" | |
304 | and el.getAttribute("spam") == "bam" | |
305 | and el.getAttributeNode("spam").isId) | |
306 | el.attributes["spam"] = "ham" | |
307 | confirm(len(el.attributes) == 1 | |
308 | and el.attributes["spam"].value == "ham" | |
309 | and el.attributes["spam"].nodeValue == "ham" | |
310 | and el.getAttribute("spam") == "ham" | |
311 | and el.attributes["spam"].isId) | |
312 | el.setAttribute("spam2", "bam") | |
313 | confirm(len(el.attributes) == 2 | |
314 | and el.attributes["spam"].value == "ham" | |
315 | and el.attributes["spam"].nodeValue == "ham" | |
316 | and el.getAttribute("spam") == "ham" | |
317 | and el.attributes["spam2"].value == "bam" | |
318 | and el.attributes["spam2"].nodeValue == "bam" | |
319 | and el.getAttribute("spam2") == "bam") | |
320 | el.attributes["spam2"] = "bam2" | |
321 | confirm(len(el.attributes) == 2 | |
322 | and el.attributes["spam"].value == "ham" | |
323 | and el.attributes["spam"].nodeValue == "ham" | |
324 | and el.getAttribute("spam") == "ham" | |
325 | and el.attributes["spam2"].value == "bam2" | |
326 | and el.attributes["spam2"].nodeValue == "bam2" | |
327 | and el.getAttribute("spam2") == "bam2") | |
328 | dom.unlink() | |
329 | ||
330 | def testGetAttrList(): | |
331 | pass | |
332 | ||
333 | def testGetAttrValues(): pass | |
334 | ||
335 | def testGetAttrLength(): pass | |
336 | ||
337 | def testGetAttribute(): pass | |
338 | ||
339 | def testGetAttributeNS(): pass | |
340 | ||
341 | def testGetAttributeNode(): pass | |
342 | ||
343 | def testGetElementsByTagNameNS(): | |
344 | d="""<foo xmlns:minidom='http://pyxml.sf.net/minidom'> | |
345 | <minidom:myelem/> | |
346 | </foo>""" | |
347 | dom = parseString(d) | |
348 | elems = dom.getElementsByTagNameNS("http://pyxml.sf.net/minidom", "myelem") | |
349 | confirm(len(elems) == 1 | |
350 | and elems[0].namespaceURI == "http://pyxml.sf.net/minidom" | |
351 | and elems[0].localName == "myelem" | |
352 | and elems[0].prefix == "minidom" | |
353 | and elems[0].tagName == "minidom:myelem" | |
354 | and elems[0].nodeName == "minidom:myelem") | |
355 | dom.unlink() | |
356 | ||
357 | def get_empty_nodelist_from_elements_by_tagName_ns_helper(doc, nsuri, lname): | |
358 | nodelist = doc.getElementsByTagNameNS(nsuri, lname) | |
359 | confirm(len(nodelist) == 0) | |
360 | ||
361 | def testGetEmptyNodeListFromElementsByTagNameNS(): | |
362 | doc = parseString('<doc/>') | |
363 | get_empty_nodelist_from_elements_by_tagName_ns_helper( | |
364 | doc, 'http://xml.python.org/namespaces/a', 'localname') | |
365 | get_empty_nodelist_from_elements_by_tagName_ns_helper( | |
366 | doc, '*', 'splat') | |
367 | get_empty_nodelist_from_elements_by_tagName_ns_helper( | |
368 | doc, 'http://xml.python.org/namespaces/a', '*') | |
369 | ||
370 | doc = parseString('<doc xmlns="http://xml.python.org/splat"><e/></doc>') | |
371 | get_empty_nodelist_from_elements_by_tagName_ns_helper( | |
372 | doc, "http://xml.python.org/splat", "not-there") | |
373 | get_empty_nodelist_from_elements_by_tagName_ns_helper( | |
374 | doc, "*", "not-there") | |
375 | get_empty_nodelist_from_elements_by_tagName_ns_helper( | |
376 | doc, "http://somewhere.else.net/not-there", "e") | |
377 | ||
378 | def testElementReprAndStr(): | |
379 | dom = Document() | |
380 | el = dom.appendChild(dom.createElement("abc")) | |
381 | string1 = repr(el) | |
382 | string2 = str(el) | |
383 | confirm(string1 == string2) | |
384 | dom.unlink() | |
385 | ||
386 | # commented out until Fredrick's fix is checked in | |
387 | def _testElementReprAndStrUnicode(): | |
388 | dom = Document() | |
389 | el = dom.appendChild(dom.createElement(u"abc")) | |
390 | string1 = repr(el) | |
391 | string2 = str(el) | |
392 | confirm(string1 == string2) | |
393 | dom.unlink() | |
394 | ||
395 | # commented out until Fredrick's fix is checked in | |
396 | def _testElementReprAndStrUnicodeNS(): | |
397 | dom = Document() | |
398 | el = dom.appendChild( | |
399 | dom.createElementNS(u"http://www.slashdot.org", u"slash:abc")) | |
400 | string1 = repr(el) | |
401 | string2 = str(el) | |
402 | confirm(string1 == string2) | |
403 | confirm(string1.find("slash:abc") != -1) | |
404 | dom.unlink() | |
405 | ||
406 | def testAttributeRepr(): | |
407 | dom = Document() | |
408 | el = dom.appendChild(dom.createElement(u"abc")) | |
409 | node = el.setAttribute("abc", "def") | |
410 | confirm(str(node) == repr(node)) | |
411 | dom.unlink() | |
412 | ||
413 | def testTextNodeRepr(): pass | |
414 | ||
415 | def testWriteXML(): | |
416 | str = '<?xml version="1.0" ?><a b="c"/>' | |
417 | dom = parseString(str) | |
418 | domstr = dom.toxml() | |
419 | dom.unlink() | |
420 | confirm(str == domstr) | |
421 | ||
422 | def testAltNewline(): | |
423 | str = '<?xml version="1.0" ?>\n<a b="c"/>\n' | |
424 | dom = parseString(str) | |
425 | domstr = dom.toprettyxml(newl="\r\n") | |
426 | dom.unlink() | |
427 | confirm(domstr == str.replace("\n", "\r\n")) | |
428 | ||
429 | def testProcessingInstruction(): | |
430 | dom = parseString('<e><?mypi \t\n data \t\n ?></e>') | |
431 | pi = dom.documentElement.firstChild | |
432 | confirm(pi.target == "mypi" | |
433 | and pi.data == "data \t\n " | |
434 | and pi.nodeName == "mypi" | |
435 | and pi.nodeType == Node.PROCESSING_INSTRUCTION_NODE | |
436 | and pi.attributes is None | |
437 | and not pi.hasChildNodes() | |
438 | and len(pi.childNodes) == 0 | |
439 | and pi.firstChild is None | |
440 | and pi.lastChild is None | |
441 | and pi.localName is None | |
442 | and pi.namespaceURI == xml.dom.EMPTY_NAMESPACE) | |
443 | ||
444 | def testProcessingInstructionRepr(): pass | |
445 | ||
446 | def testTextRepr(): pass | |
447 | ||
448 | def testWriteText(): pass | |
449 | ||
450 | def testDocumentElement(): pass | |
451 | ||
452 | def testTooManyDocumentElements(): | |
453 | doc = parseString("<doc/>") | |
454 | elem = doc.createElement("extra") | |
455 | try: | |
456 | doc.appendChild(elem) | |
457 | except xml.dom.HierarchyRequestErr: | |
458 | pass | |
459 | else: | |
460 | print "Failed to catch expected exception when" \ | |
461 | " adding extra document element." | |
462 | elem.unlink() | |
463 | doc.unlink() | |
464 | ||
465 | def testCreateElementNS(): pass | |
466 | ||
467 | def testCreateAttributeNS(): pass | |
468 | ||
469 | def testParse(): pass | |
470 | ||
471 | def testParseString(): pass | |
472 | ||
473 | def testComment(): pass | |
474 | ||
475 | def testAttrListItem(): pass | |
476 | ||
477 | def testAttrListItems(): pass | |
478 | ||
479 | def testAttrListItemNS(): pass | |
480 | ||
481 | def testAttrListKeys(): pass | |
482 | ||
483 | def testAttrListKeysNS(): pass | |
484 | ||
485 | def testRemoveNamedItem(): | |
486 | doc = parseString("<doc a=''/>") | |
487 | e = doc.documentElement | |
488 | attrs = e.attributes | |
489 | a1 = e.getAttributeNode("a") | |
490 | a2 = attrs.removeNamedItem("a") | |
491 | confirm(a1.isSameNode(a2)) | |
492 | try: | |
493 | attrs.removeNamedItem("a") | |
494 | except xml.dom.NotFoundErr: | |
495 | pass | |
496 | ||
497 | def testRemoveNamedItemNS(): | |
498 | doc = parseString("<doc xmlns:a='http://xml.python.org/' a:b=''/>") | |
499 | e = doc.documentElement | |
500 | attrs = e.attributes | |
501 | a1 = e.getAttributeNodeNS("http://xml.python.org/", "b") | |
502 | a2 = attrs.removeNamedItemNS("http://xml.python.org/", "b") | |
503 | confirm(a1.isSameNode(a2)) | |
504 | try: | |
505 | attrs.removeNamedItemNS("http://xml.python.org/", "b") | |
506 | except xml.dom.NotFoundErr: | |
507 | pass | |
508 | ||
509 | def testAttrListValues(): pass | |
510 | ||
511 | def testAttrListLength(): pass | |
512 | ||
513 | def testAttrList__getitem__(): pass | |
514 | ||
515 | def testAttrList__setitem__(): pass | |
516 | ||
517 | def testSetAttrValueandNodeValue(): pass | |
518 | ||
519 | def testParseElement(): pass | |
520 | ||
521 | def testParseAttributes(): pass | |
522 | ||
523 | def testParseElementNamespaces(): pass | |
524 | ||
525 | def testParseAttributeNamespaces(): pass | |
526 | ||
527 | def testParseProcessingInstructions(): pass | |
528 | ||
529 | def testChildNodes(): pass | |
530 | ||
531 | def testFirstChild(): pass | |
532 | ||
533 | def testHasChildNodes(): pass | |
534 | ||
535 | def testCloneElementShallow(): | |
536 | dom, clone = _setupCloneElement(0) | |
537 | confirm(len(clone.childNodes) == 0 | |
538 | and clone.childNodes.length == 0 | |
539 | and clone.parentNode is None | |
540 | and clone.toxml() == '<doc attr="value"/>' | |
541 | , "testCloneElementShallow") | |
542 | dom.unlink() | |
543 | ||
544 | def testCloneElementDeep(): | |
545 | dom, clone = _setupCloneElement(1) | |
546 | confirm(len(clone.childNodes) == 1 | |
547 | and clone.childNodes.length == 1 | |
548 | and clone.parentNode is None | |
549 | and clone.toxml() == '<doc attr="value"><foo/></doc>' | |
550 | , "testCloneElementDeep") | |
551 | dom.unlink() | |
552 | ||
553 | def _setupCloneElement(deep): | |
554 | dom = parseString("<doc attr='value'><foo/></doc>") | |
555 | root = dom.documentElement | |
556 | clone = root.cloneNode(deep) | |
557 | _testCloneElementCopiesAttributes( | |
558 | root, clone, "testCloneElement" + (deep and "Deep" or "Shallow")) | |
559 | # mutilate the original so shared data is detected | |
560 | root.tagName = root.nodeName = "MODIFIED" | |
561 | root.setAttribute("attr", "NEW VALUE") | |
562 | root.setAttribute("added", "VALUE") | |
563 | return dom, clone | |
564 | ||
565 | def _testCloneElementCopiesAttributes(e1, e2, test): | |
566 | attrs1 = e1.attributes | |
567 | attrs2 = e2.attributes | |
568 | keys1 = attrs1.keys() | |
569 | keys2 = attrs2.keys() | |
570 | keys1.sort() | |
571 | keys2.sort() | |
572 | confirm(keys1 == keys2, "clone of element has same attribute keys") | |
573 | for i in range(len(keys1)): | |
574 | a1 = attrs1.item(i) | |
575 | a2 = attrs2.item(i) | |
576 | confirm(a1 is not a2 | |
577 | and a1.value == a2.value | |
578 | and a1.nodeValue == a2.nodeValue | |
579 | and a1.namespaceURI == a2.namespaceURI | |
580 | and a1.localName == a2.localName | |
581 | , "clone of attribute node has proper attribute values") | |
582 | confirm(a2.ownerElement is e2, | |
583 | "clone of attribute node correctly owned") | |
584 | ||
585 | def testCloneDocumentShallow(): | |
586 | doc = parseString("<?xml version='1.0'?>\n" | |
587 | "<!-- comment -->" | |
588 | "<!DOCTYPE doc [\n" | |
589 | "<!NOTATION notation SYSTEM 'http://xml.python.org/'>\n" | |
590 | "]>\n" | |
591 | "<doc attr='value'/>") | |
592 | doc2 = doc.cloneNode(0) | |
593 | confirm(doc2 is None, | |
594 | "testCloneDocumentShallow:" | |
595 | " shallow cloning of documents makes no sense!") | |
596 | ||
597 | def testCloneDocumentDeep(): | |
598 | doc = parseString("<?xml version='1.0'?>\n" | |
599 | "<!-- comment -->" | |
600 | "<!DOCTYPE doc [\n" | |
601 | "<!NOTATION notation SYSTEM 'http://xml.python.org/'>\n" | |
602 | "]>\n" | |
603 | "<doc attr='value'/>") | |
604 | doc2 = doc.cloneNode(1) | |
605 | confirm(not (doc.isSameNode(doc2) or doc2.isSameNode(doc)), | |
606 | "testCloneDocumentDeep: document objects not distinct") | |
607 | confirm(len(doc.childNodes) == len(doc2.childNodes), | |
608 | "testCloneDocumentDeep: wrong number of Document children") | |
609 | confirm(doc2.documentElement.nodeType == Node.ELEMENT_NODE, | |
610 | "testCloneDocumentDeep: documentElement not an ELEMENT_NODE") | |
611 | confirm(doc2.documentElement.ownerDocument.isSameNode(doc2), | |
612 | "testCloneDocumentDeep: documentElement owner is not new document") | |
613 | confirm(not doc.documentElement.isSameNode(doc2.documentElement), | |
614 | "testCloneDocumentDeep: documentElement should not be shared") | |
615 | if doc.doctype is not None: | |
616 | # check the doctype iff the original DOM maintained it | |
617 | confirm(doc2.doctype.nodeType == Node.DOCUMENT_TYPE_NODE, | |
618 | "testCloneDocumentDeep: doctype not a DOCUMENT_TYPE_NODE") | |
619 | confirm(doc2.doctype.ownerDocument.isSameNode(doc2)) | |
620 | confirm(not doc.doctype.isSameNode(doc2.doctype)) | |
621 | ||
622 | def testCloneDocumentTypeDeepOk(): | |
623 | doctype = create_nonempty_doctype() | |
624 | clone = doctype.cloneNode(1) | |
625 | confirm(clone is not None | |
626 | and clone.nodeName == doctype.nodeName | |
627 | and clone.name == doctype.name | |
628 | and clone.publicId == doctype.publicId | |
629 | and clone.systemId == doctype.systemId | |
630 | and len(clone.entities) == len(doctype.entities) | |
631 | and clone.entities.item(len(clone.entities)) is None | |
632 | and len(clone.notations) == len(doctype.notations) | |
633 | and clone.notations.item(len(clone.notations)) is None | |
634 | and len(clone.childNodes) == 0) | |
635 | for i in range(len(doctype.entities)): | |
636 | se = doctype.entities.item(i) | |
637 | ce = clone.entities.item(i) | |
638 | confirm((not se.isSameNode(ce)) | |
639 | and (not ce.isSameNode(se)) | |
640 | and ce.nodeName == se.nodeName | |
641 | and ce.notationName == se.notationName | |
642 | and ce.publicId == se.publicId | |
643 | and ce.systemId == se.systemId | |
644 | and ce.encoding == se.encoding | |
645 | and ce.actualEncoding == se.actualEncoding | |
646 | and ce.version == se.version) | |
647 | for i in range(len(doctype.notations)): | |
648 | sn = doctype.notations.item(i) | |
649 | cn = clone.notations.item(i) | |
650 | confirm((not sn.isSameNode(cn)) | |
651 | and (not cn.isSameNode(sn)) | |
652 | and cn.nodeName == sn.nodeName | |
653 | and cn.publicId == sn.publicId | |
654 | and cn.systemId == sn.systemId) | |
655 | ||
656 | def testCloneDocumentTypeDeepNotOk(): | |
657 | doc = create_doc_with_doctype() | |
658 | clone = doc.doctype.cloneNode(1) | |
659 | confirm(clone is None, "testCloneDocumentTypeDeepNotOk") | |
660 | ||
661 | def testCloneDocumentTypeShallowOk(): | |
662 | doctype = create_nonempty_doctype() | |
663 | clone = doctype.cloneNode(0) | |
664 | confirm(clone is not None | |
665 | and clone.nodeName == doctype.nodeName | |
666 | and clone.name == doctype.name | |
667 | and clone.publicId == doctype.publicId | |
668 | and clone.systemId == doctype.systemId | |
669 | and len(clone.entities) == 0 | |
670 | and clone.entities.item(0) is None | |
671 | and len(clone.notations) == 0 | |
672 | and clone.notations.item(0) is None | |
673 | and len(clone.childNodes) == 0) | |
674 | ||
675 | def testCloneDocumentTypeShallowNotOk(): | |
676 | doc = create_doc_with_doctype() | |
677 | clone = doc.doctype.cloneNode(0) | |
678 | confirm(clone is None, "testCloneDocumentTypeShallowNotOk") | |
679 | ||
680 | def check_import_document(deep, testName): | |
681 | doc1 = parseString("<doc/>") | |
682 | doc2 = parseString("<doc/>") | |
683 | try: | |
684 | doc1.importNode(doc2, deep) | |
685 | except xml.dom.NotSupportedErr: | |
686 | pass | |
687 | else: | |
688 | raise Exception(testName + | |
689 | ": expected NotSupportedErr when importing a document") | |
690 | ||
691 | def testImportDocumentShallow(): | |
692 | check_import_document(0, "testImportDocumentShallow") | |
693 | ||
694 | def testImportDocumentDeep(): | |
695 | check_import_document(1, "testImportDocumentDeep") | |
696 | ||
697 | # The tests of DocumentType importing use these helpers to construct | |
698 | # the documents to work with, since not all DOM builders actually | |
699 | # create the DocumentType nodes. | |
700 | ||
701 | def create_doc_without_doctype(doctype=None): | |
702 | return getDOMImplementation().createDocument(None, "doc", doctype) | |
703 | ||
704 | def create_nonempty_doctype(): | |
705 | doctype = getDOMImplementation().createDocumentType("doc", None, None) | |
706 | doctype.entities._seq = [] | |
707 | doctype.notations._seq = [] | |
708 | notation = xml.dom.minidom.Notation("my-notation", None, | |
709 | "http://xml.python.org/notations/my") | |
710 | doctype.notations._seq.append(notation) | |
711 | entity = xml.dom.minidom.Entity("my-entity", None, | |
712 | "http://xml.python.org/entities/my", | |
713 | "my-notation") | |
714 | entity.version = "1.0" | |
715 | entity.encoding = "utf-8" | |
716 | entity.actualEncoding = "us-ascii" | |
717 | doctype.entities._seq.append(entity) | |
718 | return doctype | |
719 | ||
720 | def create_doc_with_doctype(): | |
721 | doctype = create_nonempty_doctype() | |
722 | doc = create_doc_without_doctype(doctype) | |
723 | doctype.entities.item(0).ownerDocument = doc | |
724 | doctype.notations.item(0).ownerDocument = doc | |
725 | return doc | |
726 | ||
727 | def testImportDocumentTypeShallow(): | |
728 | src = create_doc_with_doctype() | |
729 | target = create_doc_without_doctype() | |
730 | try: | |
731 | imported = target.importNode(src.doctype, 0) | |
732 | except xml.dom.NotSupportedErr: | |
733 | pass | |
734 | else: | |
735 | raise Exception( | |
736 | "testImportDocumentTypeShallow: expected NotSupportedErr") | |
737 | ||
738 | def testImportDocumentTypeDeep(): | |
739 | src = create_doc_with_doctype() | |
740 | target = create_doc_without_doctype() | |
741 | try: | |
742 | imported = target.importNode(src.doctype, 1) | |
743 | except xml.dom.NotSupportedErr: | |
744 | pass | |
745 | else: | |
746 | raise Exception( | |
747 | "testImportDocumentTypeDeep: expected NotSupportedErr") | |
748 | ||
749 | # Testing attribute clones uses a helper, and should always be deep, | |
750 | # even if the argument to cloneNode is false. | |
751 | def check_clone_attribute(deep, testName): | |
752 | doc = parseString("<doc attr='value'/>") | |
753 | attr = doc.documentElement.getAttributeNode("attr") | |
754 | assert attr is not None | |
755 | clone = attr.cloneNode(deep) | |
756 | confirm(not clone.isSameNode(attr)) | |
757 | confirm(not attr.isSameNode(clone)) | |
758 | confirm(clone.ownerElement is None, | |
759 | testName + ": ownerElement should be None") | |
760 | confirm(clone.ownerDocument.isSameNode(attr.ownerDocument), | |
761 | testName + ": ownerDocument does not match") | |
762 | confirm(clone.specified, | |
763 | testName + ": cloned attribute must have specified == True") | |
764 | ||
765 | def testCloneAttributeShallow(): | |
766 | check_clone_attribute(0, "testCloneAttributeShallow") | |
767 | ||
768 | def testCloneAttributeDeep(): | |
769 | check_clone_attribute(1, "testCloneAttributeDeep") | |
770 | ||
771 | def check_clone_pi(deep, testName): | |
772 | doc = parseString("<?target data?><doc/>") | |
773 | pi = doc.firstChild | |
774 | assert pi.nodeType == Node.PROCESSING_INSTRUCTION_NODE | |
775 | clone = pi.cloneNode(deep) | |
776 | confirm(clone.target == pi.target | |
777 | and clone.data == pi.data) | |
778 | ||
779 | def testClonePIShallow(): | |
780 | check_clone_pi(0, "testClonePIShallow") | |
781 | ||
782 | def testClonePIDeep(): | |
783 | check_clone_pi(1, "testClonePIDeep") | |
784 | ||
785 | def testNormalize(): | |
786 | doc = parseString("<doc/>") | |
787 | root = doc.documentElement | |
788 | root.appendChild(doc.createTextNode("first")) | |
789 | root.appendChild(doc.createTextNode("second")) | |
790 | confirm(len(root.childNodes) == 2 | |
791 | and root.childNodes.length == 2, "testNormalize -- preparation") | |
792 | doc.normalize() | |
793 | confirm(len(root.childNodes) == 1 | |
794 | and root.childNodes.length == 1 | |
795 | and root.firstChild is root.lastChild | |
796 | and root.firstChild.data == "firstsecond" | |
797 | , "testNormalize -- result") | |
798 | doc.unlink() | |
799 | ||
800 | doc = parseString("<doc/>") | |
801 | root = doc.documentElement | |
802 | root.appendChild(doc.createTextNode("")) | |
803 | doc.normalize() | |
804 | confirm(len(root.childNodes) == 0 | |
805 | and root.childNodes.length == 0, | |
806 | "testNormalize -- single empty node removed") | |
807 | doc.unlink() | |
808 | ||
809 | def testSiblings(): | |
810 | doc = parseString("<doc><?pi?>text?<elm/></doc>") | |
811 | root = doc.documentElement | |
812 | (pi, text, elm) = root.childNodes | |
813 | ||
814 | confirm(pi.nextSibling is text and | |
815 | pi.previousSibling is None and | |
816 | text.nextSibling is elm and | |
817 | text.previousSibling is pi and | |
818 | elm.nextSibling is None and | |
819 | elm.previousSibling is text, "testSiblings") | |
820 | ||
821 | doc.unlink() | |
822 | ||
823 | def testParents(): | |
824 | doc = parseString("<doc><elm1><elm2/><elm2><elm3/></elm2></elm1></doc>") | |
825 | root = doc.documentElement | |
826 | elm1 = root.childNodes[0] | |
827 | (elm2a, elm2b) = elm1.childNodes | |
828 | elm3 = elm2b.childNodes[0] | |
829 | ||
830 | confirm(root.parentNode is doc and | |
831 | elm1.parentNode is root and | |
832 | elm2a.parentNode is elm1 and | |
833 | elm2b.parentNode is elm1 and | |
834 | elm3.parentNode is elm2b, "testParents") | |
835 | ||
836 | doc.unlink() | |
837 | ||
838 | def testNodeListItem(): | |
839 | doc = parseString("<doc><e/><e/></doc>") | |
840 | children = doc.childNodes | |
841 | docelem = children[0] | |
842 | confirm(children[0] is children.item(0) | |
843 | and children.item(1) is None | |
844 | and docelem.childNodes.item(0) is docelem.childNodes[0] | |
845 | and docelem.childNodes.item(1) is docelem.childNodes[1] | |
846 | and docelem.childNodes.item(0).childNodes.item(0) is None, | |
847 | "test NodeList.item()") | |
848 | doc.unlink() | |
849 | ||
850 | def testSAX2DOM(): | |
851 | from xml.dom import pulldom | |
852 | ||
853 | sax2dom = pulldom.SAX2DOM() | |
854 | sax2dom.startDocument() | |
855 | sax2dom.startElement("doc", {}) | |
856 | sax2dom.characters("text") | |
857 | sax2dom.startElement("subelm", {}) | |
858 | sax2dom.characters("text") | |
859 | sax2dom.endElement("subelm") | |
860 | sax2dom.characters("text") | |
861 | sax2dom.endElement("doc") | |
862 | sax2dom.endDocument() | |
863 | ||
864 | doc = sax2dom.document | |
865 | root = doc.documentElement | |
866 | (text1, elm1, text2) = root.childNodes | |
867 | text3 = elm1.childNodes[0] | |
868 | ||
869 | confirm(text1.previousSibling is None and | |
870 | text1.nextSibling is elm1 and | |
871 | elm1.previousSibling is text1 and | |
872 | elm1.nextSibling is text2 and | |
873 | text2.previousSibling is elm1 and | |
874 | text2.nextSibling is None and | |
875 | text3.previousSibling is None and | |
876 | text3.nextSibling is None, "testSAX2DOM - siblings") | |
877 | ||
878 | confirm(root.parentNode is doc and | |
879 | text1.parentNode is root and | |
880 | elm1.parentNode is root and | |
881 | text2.parentNode is root and | |
882 | text3.parentNode is elm1, "testSAX2DOM - parents") | |
883 | ||
884 | doc.unlink() | |
885 | ||
886 | def testEncodings(): | |
887 | doc = parseString('<foo>€</foo>') | |
888 | confirm(doc.toxml() == u'<?xml version="1.0" ?><foo>\u20ac</foo>' | |
889 | and doc.toxml('utf-8') == '<?xml version="1.0" encoding="utf-8"?><foo>\xe2\x82\xac</foo>' | |
890 | and doc.toxml('iso-8859-15') == '<?xml version="1.0" encoding="iso-8859-15"?><foo>\xa4</foo>', | |
891 | "testEncodings - encoding EURO SIGN") | |
892 | doc.unlink() | |
893 | ||
894 | class UserDataHandler: | |
895 | called = 0 | |
896 | def handle(self, operation, key, data, src, dst): | |
897 | dst.setUserData(key, data + 1, self) | |
898 | src.setUserData(key, None, None) | |
899 | self.called = 1 | |
900 | ||
901 | def testUserData(): | |
902 | dom = Document() | |
903 | n = dom.createElement('e') | |
904 | confirm(n.getUserData("foo") is None) | |
905 | n.setUserData("foo", None, None) | |
906 | confirm(n.getUserData("foo") is None) | |
907 | n.setUserData("foo", 12, 12) | |
908 | n.setUserData("bar", 13, 13) | |
909 | confirm(n.getUserData("foo") == 12) | |
910 | confirm(n.getUserData("bar") == 13) | |
911 | n.setUserData("foo", None, None) | |
912 | confirm(n.getUserData("foo") is None) | |
913 | confirm(n.getUserData("bar") == 13) | |
914 | ||
915 | handler = UserDataHandler() | |
916 | n.setUserData("bar", 12, handler) | |
917 | c = n.cloneNode(1) | |
918 | confirm(handler.called | |
919 | and n.getUserData("bar") is None | |
920 | and c.getUserData("bar") == 13) | |
921 | n.unlink() | |
922 | c.unlink() | |
923 | dom.unlink() | |
924 | ||
925 | def testRenameAttribute(): | |
926 | doc = parseString("<doc a='v'/>") | |
927 | elem = doc.documentElement | |
928 | attrmap = elem.attributes | |
929 | attr = elem.attributes['a'] | |
930 | ||
931 | # Simple renaming | |
932 | attr = doc.renameNode(attr, xml.dom.EMPTY_NAMESPACE, "b") | |
933 | confirm(attr.name == "b" | |
934 | and attr.nodeName == "b" | |
935 | and attr.localName is None | |
936 | and attr.namespaceURI == xml.dom.EMPTY_NAMESPACE | |
937 | and attr.prefix is None | |
938 | and attr.value == "v" | |
939 | and elem.getAttributeNode("a") is None | |
940 | and elem.getAttributeNode("b").isSameNode(attr) | |
941 | and attrmap["b"].isSameNode(attr) | |
942 | and attr.ownerDocument.isSameNode(doc) | |
943 | and attr.ownerElement.isSameNode(elem)) | |
944 | ||
945 | # Rename to have a namespace, no prefix | |
946 | attr = doc.renameNode(attr, "http://xml.python.org/ns", "c") | |
947 | confirm(attr.name == "c" | |
948 | and attr.nodeName == "c" | |
949 | and attr.localName == "c" | |
950 | and attr.namespaceURI == "http://xml.python.org/ns" | |
951 | and attr.prefix is None | |
952 | and attr.value == "v" | |
953 | and elem.getAttributeNode("a") is None | |
954 | and elem.getAttributeNode("b") is None | |
955 | and elem.getAttributeNode("c").isSameNode(attr) | |
956 | and elem.getAttributeNodeNS( | |
957 | "http://xml.python.org/ns", "c").isSameNode(attr) | |
958 | and attrmap["c"].isSameNode(attr) | |
959 | and attrmap[("http://xml.python.org/ns", "c")].isSameNode(attr)) | |
960 | ||
961 | # Rename to have a namespace, with prefix | |
962 | attr = doc.renameNode(attr, "http://xml.python.org/ns2", "p:d") | |
963 | confirm(attr.name == "p:d" | |
964 | and attr.nodeName == "p:d" | |
965 | and attr.localName == "d" | |
966 | and attr.namespaceURI == "http://xml.python.org/ns2" | |
967 | and attr.prefix == "p" | |
968 | and attr.value == "v" | |
969 | and elem.getAttributeNode("a") is None | |
970 | and elem.getAttributeNode("b") is None | |
971 | and elem.getAttributeNode("c") is None | |
972 | and elem.getAttributeNodeNS( | |
973 | "http://xml.python.org/ns", "c") is None | |
974 | and elem.getAttributeNode("p:d").isSameNode(attr) | |
975 | and elem.getAttributeNodeNS( | |
976 | "http://xml.python.org/ns2", "d").isSameNode(attr) | |
977 | and attrmap["p:d"].isSameNode(attr) | |
978 | and attrmap[("http://xml.python.org/ns2", "d")].isSameNode(attr)) | |
979 | ||
980 | # Rename back to a simple non-NS node | |
981 | attr = doc.renameNode(attr, xml.dom.EMPTY_NAMESPACE, "e") | |
982 | confirm(attr.name == "e" | |
983 | and attr.nodeName == "e" | |
984 | and attr.localName is None | |
985 | and attr.namespaceURI == xml.dom.EMPTY_NAMESPACE | |
986 | and attr.prefix is None | |
987 | and attr.value == "v" | |
988 | and elem.getAttributeNode("a") is None | |
989 | and elem.getAttributeNode("b") is None | |
990 | and elem.getAttributeNode("c") is None | |
991 | and elem.getAttributeNode("p:d") is None | |
992 | and elem.getAttributeNodeNS( | |
993 | "http://xml.python.org/ns", "c") is None | |
994 | and elem.getAttributeNode("e").isSameNode(attr) | |
995 | and attrmap["e"].isSameNode(attr)) | |
996 | ||
997 | try: | |
998 | doc.renameNode(attr, "http://xml.python.org/ns", "xmlns") | |
999 | except xml.dom.NamespaceErr: | |
1000 | pass | |
1001 | else: | |
1002 | print "expected NamespaceErr" | |
1003 | ||
1004 | checkRenameNodeSharedConstraints(doc, attr) | |
1005 | doc.unlink() | |
1006 | ||
1007 | def testRenameElement(): | |
1008 | doc = parseString("<doc/>") | |
1009 | elem = doc.documentElement | |
1010 | ||
1011 | # Simple renaming | |
1012 | elem = doc.renameNode(elem, xml.dom.EMPTY_NAMESPACE, "a") | |
1013 | confirm(elem.tagName == "a" | |
1014 | and elem.nodeName == "a" | |
1015 | and elem.localName is None | |
1016 | and elem.namespaceURI == xml.dom.EMPTY_NAMESPACE | |
1017 | and elem.prefix is None | |
1018 | and elem.ownerDocument.isSameNode(doc)) | |
1019 | ||
1020 | # Rename to have a namespace, no prefix | |
1021 | elem = doc.renameNode(elem, "http://xml.python.org/ns", "b") | |
1022 | confirm(elem.tagName == "b" | |
1023 | and elem.nodeName == "b" | |
1024 | and elem.localName == "b" | |
1025 | and elem.namespaceURI == "http://xml.python.org/ns" | |
1026 | and elem.prefix is None | |
1027 | and elem.ownerDocument.isSameNode(doc)) | |
1028 | ||
1029 | # Rename to have a namespace, with prefix | |
1030 | elem = doc.renameNode(elem, "http://xml.python.org/ns2", "p:c") | |
1031 | confirm(elem.tagName == "p:c" | |
1032 | and elem.nodeName == "p:c" | |
1033 | and elem.localName == "c" | |
1034 | and elem.namespaceURI == "http://xml.python.org/ns2" | |
1035 | and elem.prefix == "p" | |
1036 | and elem.ownerDocument.isSameNode(doc)) | |
1037 | ||
1038 | # Rename back to a simple non-NS node | |
1039 | elem = doc.renameNode(elem, xml.dom.EMPTY_NAMESPACE, "d") | |
1040 | confirm(elem.tagName == "d" | |
1041 | and elem.nodeName == "d" | |
1042 | and elem.localName is None | |
1043 | and elem.namespaceURI == xml.dom.EMPTY_NAMESPACE | |
1044 | and elem.prefix is None | |
1045 | and elem.ownerDocument.isSameNode(doc)) | |
1046 | ||
1047 | checkRenameNodeSharedConstraints(doc, elem) | |
1048 | doc.unlink() | |
1049 | ||
1050 | def checkRenameNodeSharedConstraints(doc, node): | |
1051 | # Make sure illegal NS usage is detected: | |
1052 | try: | |
1053 | doc.renameNode(node, "http://xml.python.org/ns", "xmlns:foo") | |
1054 | except xml.dom.NamespaceErr: | |
1055 | pass | |
1056 | else: | |
1057 | print "expected NamespaceErr" | |
1058 | ||
1059 | doc2 = parseString("<doc/>") | |
1060 | try: | |
1061 | doc2.renameNode(node, xml.dom.EMPTY_NAMESPACE, "foo") | |
1062 | except xml.dom.WrongDocumentErr: | |
1063 | pass | |
1064 | else: | |
1065 | print "expected WrongDocumentErr" | |
1066 | ||
1067 | def testRenameOther(): | |
1068 | # We have to create a comment node explicitly since not all DOM | |
1069 | # builders used with minidom add comments to the DOM. | |
1070 | doc = xml.dom.minidom.getDOMImplementation().createDocument( | |
1071 | xml.dom.EMPTY_NAMESPACE, "e", None) | |
1072 | node = doc.createComment("comment") | |
1073 | try: | |
1074 | doc.renameNode(node, xml.dom.EMPTY_NAMESPACE, "foo") | |
1075 | except xml.dom.NotSupportedErr: | |
1076 | pass | |
1077 | else: | |
1078 | print "expected NotSupportedErr when renaming comment node" | |
1079 | doc.unlink() | |
1080 | ||
1081 | def checkWholeText(node, s): | |
1082 | t = node.wholeText | |
1083 | confirm(t == s, "looking for %s, found %s" % (repr(s), repr(t))) | |
1084 | ||
1085 | def testWholeText(): | |
1086 | doc = parseString("<doc>a</doc>") | |
1087 | elem = doc.documentElement | |
1088 | text = elem.childNodes[0] | |
1089 | assert text.nodeType == Node.TEXT_NODE | |
1090 | ||
1091 | checkWholeText(text, "a") | |
1092 | elem.appendChild(doc.createTextNode("b")) | |
1093 | checkWholeText(text, "ab") | |
1094 | elem.insertBefore(doc.createCDATASection("c"), text) | |
1095 | checkWholeText(text, "cab") | |
1096 | ||
1097 | # make sure we don't cross other nodes | |
1098 | splitter = doc.createComment("comment") | |
1099 | elem.appendChild(splitter) | |
1100 | text2 = doc.createTextNode("d") | |
1101 | elem.appendChild(text2) | |
1102 | checkWholeText(text, "cab") | |
1103 | checkWholeText(text2, "d") | |
1104 | ||
1105 | x = doc.createElement("x") | |
1106 | elem.replaceChild(x, splitter) | |
1107 | splitter = x | |
1108 | checkWholeText(text, "cab") | |
1109 | checkWholeText(text2, "d") | |
1110 | ||
1111 | x = doc.createProcessingInstruction("y", "z") | |
1112 | elem.replaceChild(x, splitter) | |
1113 | splitter = x | |
1114 | checkWholeText(text, "cab") | |
1115 | checkWholeText(text2, "d") | |
1116 | ||
1117 | elem.removeChild(splitter) | |
1118 | checkWholeText(text, "cabd") | |
1119 | checkWholeText(text2, "cabd") | |
1120 | ||
1121 | def testReplaceWholeText(): | |
1122 | def setup(): | |
1123 | doc = parseString("<doc>a<e/>d</doc>") | |
1124 | elem = doc.documentElement | |
1125 | text1 = elem.firstChild | |
1126 | text2 = elem.lastChild | |
1127 | splitter = text1.nextSibling | |
1128 | elem.insertBefore(doc.createTextNode("b"), splitter) | |
1129 | elem.insertBefore(doc.createCDATASection("c"), text1) | |
1130 | return doc, elem, text1, splitter, text2 | |
1131 | ||
1132 | doc, elem, text1, splitter, text2 = setup() | |
1133 | text = text1.replaceWholeText("new content") | |
1134 | checkWholeText(text, "new content") | |
1135 | checkWholeText(text2, "d") | |
1136 | confirm(len(elem.childNodes) == 3) | |
1137 | ||
1138 | doc, elem, text1, splitter, text2 = setup() | |
1139 | text = text2.replaceWholeText("new content") | |
1140 | checkWholeText(text, "new content") | |
1141 | checkWholeText(text1, "cab") | |
1142 | confirm(len(elem.childNodes) == 5) | |
1143 | ||
1144 | doc, elem, text1, splitter, text2 = setup() | |
1145 | text = text1.replaceWholeText("") | |
1146 | checkWholeText(text2, "d") | |
1147 | confirm(text is None | |
1148 | and len(elem.childNodes) == 2) | |
1149 | ||
1150 | def testSchemaType(): | |
1151 | doc = parseString( | |
1152 | "<!DOCTYPE doc [\n" | |
1153 | " <!ENTITY e1 SYSTEM 'http://xml.python.org/e1'>\n" | |
1154 | " <!ENTITY e2 SYSTEM 'http://xml.python.org/e2'>\n" | |
1155 | " <!ATTLIST doc id ID #IMPLIED \n" | |
1156 | " ref IDREF #IMPLIED \n" | |
1157 | " refs IDREFS #IMPLIED \n" | |
1158 | " enum (a|b) #IMPLIED \n" | |
1159 | " ent ENTITY #IMPLIED \n" | |
1160 | " ents ENTITIES #IMPLIED \n" | |
1161 | " nm NMTOKEN #IMPLIED \n" | |
1162 | " nms NMTOKENS #IMPLIED \n" | |
1163 | " text CDATA #IMPLIED \n" | |
1164 | " >\n" | |
1165 | "]><doc id='name' notid='name' text='splat!' enum='b'" | |
1166 | " ref='name' refs='name name' ent='e1' ents='e1 e2'" | |
1167 | " nm='123' nms='123 abc' />") | |
1168 | elem = doc.documentElement | |
1169 | # We don't want to rely on any specific loader at this point, so | |
1170 | # just make sure we can get to all the names, and that the | |
1171 | # DTD-based namespace is right. The names can vary by loader | |
1172 | # since each supports a different level of DTD information. | |
1173 | t = elem.schemaType | |
1174 | confirm(t.name is None | |
1175 | and t.namespace == xml.dom.EMPTY_NAMESPACE) | |
1176 | names = "id notid text enum ref refs ent ents nm nms".split() | |
1177 | for name in names: | |
1178 | a = elem.getAttributeNode(name) | |
1179 | t = a.schemaType | |
1180 | confirm(hasattr(t, "name") | |
1181 | and t.namespace == xml.dom.EMPTY_NAMESPACE) | |
1182 | ||
1183 | def testSetIdAttribute(): | |
1184 | doc = parseString("<doc a1='v' a2='w'/>") | |
1185 | e = doc.documentElement | |
1186 | a1 = e.getAttributeNode("a1") | |
1187 | a2 = e.getAttributeNode("a2") | |
1188 | confirm(doc.getElementById("v") is None | |
1189 | and not a1.isId | |
1190 | and not a2.isId) | |
1191 | e.setIdAttribute("a1") | |
1192 | confirm(e.isSameNode(doc.getElementById("v")) | |
1193 | and a1.isId | |
1194 | and not a2.isId) | |
1195 | e.setIdAttribute("a2") | |
1196 | confirm(e.isSameNode(doc.getElementById("v")) | |
1197 | and e.isSameNode(doc.getElementById("w")) | |
1198 | and a1.isId | |
1199 | and a2.isId) | |
1200 | # replace the a1 node; the new node should *not* be an ID | |
1201 | a3 = doc.createAttribute("a1") | |
1202 | a3.value = "v" | |
1203 | e.setAttributeNode(a3) | |
1204 | confirm(doc.getElementById("v") is None | |
1205 | and e.isSameNode(doc.getElementById("w")) | |
1206 | and not a1.isId | |
1207 | and a2.isId | |
1208 | and not a3.isId) | |
1209 | # renaming an attribute should not affect its ID-ness: | |
1210 | doc.renameNode(a2, xml.dom.EMPTY_NAMESPACE, "an") | |
1211 | confirm(e.isSameNode(doc.getElementById("w")) | |
1212 | and a2.isId) | |
1213 | ||
1214 | def testSetIdAttributeNS(): | |
1215 | NS1 = "http://xml.python.org/ns1" | |
1216 | NS2 = "http://xml.python.org/ns2" | |
1217 | doc = parseString("<doc" | |
1218 | " xmlns:ns1='" + NS1 + "'" | |
1219 | " xmlns:ns2='" + NS2 + "'" | |
1220 | " ns1:a1='v' ns2:a2='w'/>") | |
1221 | e = doc.documentElement | |
1222 | a1 = e.getAttributeNodeNS(NS1, "a1") | |
1223 | a2 = e.getAttributeNodeNS(NS2, "a2") | |
1224 | confirm(doc.getElementById("v") is None | |
1225 | and not a1.isId | |
1226 | and not a2.isId) | |
1227 | e.setIdAttributeNS(NS1, "a1") | |
1228 | confirm(e.isSameNode(doc.getElementById("v")) | |
1229 | and a1.isId | |
1230 | and not a2.isId) | |
1231 | e.setIdAttributeNS(NS2, "a2") | |
1232 | confirm(e.isSameNode(doc.getElementById("v")) | |
1233 | and e.isSameNode(doc.getElementById("w")) | |
1234 | and a1.isId | |
1235 | and a2.isId) | |
1236 | # replace the a1 node; the new node should *not* be an ID | |
1237 | a3 = doc.createAttributeNS(NS1, "a1") | |
1238 | a3.value = "v" | |
1239 | e.setAttributeNode(a3) | |
1240 | confirm(e.isSameNode(doc.getElementById("w"))) | |
1241 | confirm(not a1.isId) | |
1242 | confirm(a2.isId) | |
1243 | confirm(not a3.isId) | |
1244 | confirm(doc.getElementById("v") is None) | |
1245 | # renaming an attribute should not affect its ID-ness: | |
1246 | doc.renameNode(a2, xml.dom.EMPTY_NAMESPACE, "an") | |
1247 | confirm(e.isSameNode(doc.getElementById("w")) | |
1248 | and a2.isId) | |
1249 | ||
1250 | def testSetIdAttributeNode(): | |
1251 | NS1 = "http://xml.python.org/ns1" | |
1252 | NS2 = "http://xml.python.org/ns2" | |
1253 | doc = parseString("<doc" | |
1254 | " xmlns:ns1='" + NS1 + "'" | |
1255 | " xmlns:ns2='" + NS2 + "'" | |
1256 | " ns1:a1='v' ns2:a2='w'/>") | |
1257 | e = doc.documentElement | |
1258 | a1 = e.getAttributeNodeNS(NS1, "a1") | |
1259 | a2 = e.getAttributeNodeNS(NS2, "a2") | |
1260 | confirm(doc.getElementById("v") is None | |
1261 | and not a1.isId | |
1262 | and not a2.isId) | |
1263 | e.setIdAttributeNode(a1) | |
1264 | confirm(e.isSameNode(doc.getElementById("v")) | |
1265 | and a1.isId | |
1266 | and not a2.isId) | |
1267 | e.setIdAttributeNode(a2) | |
1268 | confirm(e.isSameNode(doc.getElementById("v")) | |
1269 | and e.isSameNode(doc.getElementById("w")) | |
1270 | and a1.isId | |
1271 | and a2.isId) | |
1272 | # replace the a1 node; the new node should *not* be an ID | |
1273 | a3 = doc.createAttributeNS(NS1, "a1") | |
1274 | a3.value = "v" | |
1275 | e.setAttributeNode(a3) | |
1276 | confirm(e.isSameNode(doc.getElementById("w"))) | |
1277 | confirm(not a1.isId) | |
1278 | confirm(a2.isId) | |
1279 | confirm(not a3.isId) | |
1280 | confirm(doc.getElementById("v") is None) | |
1281 | # renaming an attribute should not affect its ID-ness: | |
1282 | doc.renameNode(a2, xml.dom.EMPTY_NAMESPACE, "an") | |
1283 | confirm(e.isSameNode(doc.getElementById("w")) | |
1284 | and a2.isId) | |
1285 | ||
1286 | def testPickledDocument(): | |
1287 | doc = parseString("<?xml version='1.0' encoding='us-ascii'?>\n" | |
1288 | "<!DOCTYPE doc PUBLIC 'http://xml.python.org/public'" | |
1289 | " 'http://xml.python.org/system' [\n" | |
1290 | " <!ELEMENT e EMPTY>\n" | |
1291 | " <!ENTITY ent SYSTEM 'http://xml.python.org/entity'>\n" | |
1292 | "]><doc attr='value'> text\n" | |
1293 | "<?pi sample?> <!-- comment --> <e/> </doc>") | |
1294 | s = pickle.dumps(doc) | |
1295 | doc2 = pickle.loads(s) | |
1296 | stack = [(doc, doc2)] | |
1297 | while stack: | |
1298 | n1, n2 = stack.pop() | |
1299 | confirm(n1.nodeType == n2.nodeType | |
1300 | and len(n1.childNodes) == len(n2.childNodes) | |
1301 | and n1.nodeName == n2.nodeName | |
1302 | and not n1.isSameNode(n2) | |
1303 | and not n2.isSameNode(n1)) | |
1304 | if n1.nodeType == Node.DOCUMENT_TYPE_NODE: | |
1305 | len(n1.entities) | |
1306 | len(n2.entities) | |
1307 | len(n1.notations) | |
1308 | len(n2.notations) | |
1309 | confirm(len(n1.entities) == len(n2.entities) | |
1310 | and len(n1.notations) == len(n2.notations)) | |
1311 | for i in range(len(n1.notations)): | |
1312 | no1 = n1.notations.item(i) | |
1313 | no2 = n1.notations.item(i) | |
1314 | confirm(no1.name == no2.name | |
1315 | and no1.publicId == no2.publicId | |
1316 | and no1.systemId == no2.systemId) | |
1317 | statck.append((no1, no2)) | |
1318 | for i in range(len(n1.entities)): | |
1319 | e1 = n1.entities.item(i) | |
1320 | e2 = n2.entities.item(i) | |
1321 | confirm(e1.notationName == e2.notationName | |
1322 | and e1.publicId == e2.publicId | |
1323 | and e1.systemId == e2.systemId) | |
1324 | stack.append((e1, e2)) | |
1325 | if n1.nodeType != Node.DOCUMENT_NODE: | |
1326 | confirm(n1.ownerDocument.isSameNode(doc) | |
1327 | and n2.ownerDocument.isSameNode(doc2)) | |
1328 | for i in range(len(n1.childNodes)): | |
1329 | stack.append((n1.childNodes[i], n2.childNodes[i])) | |
1330 | ||
1331 | ||
1332 | # --- MAIN PROGRAM | |
1333 | ||
1334 | names = globals().keys() | |
1335 | names.sort() | |
1336 | ||
1337 | failed = [] | |
1338 | ||
1339 | try: | |
1340 | Node.allnodes | |
1341 | except AttributeError: | |
1342 | # We don't actually have the minidom from the standard library, | |
1343 | # but are picking up the PyXML version from site-packages. | |
1344 | def check_allnodes(): | |
1345 | pass | |
1346 | else: | |
1347 | def check_allnodes(): | |
1348 | confirm(len(Node.allnodes) == 0, | |
1349 | "assertion: len(Node.allnodes) == 0") | |
1350 | if len(Node.allnodes): | |
1351 | print "Garbage left over:" | |
1352 | if verbose: | |
1353 | print Node.allnodes.items()[0:10] | |
1354 | else: | |
1355 | # Don't print specific nodes if repeatable results | |
1356 | # are needed | |
1357 | print len(Node.allnodes) | |
1358 | Node.allnodes = {} | |
1359 | ||
1360 | for name in names: | |
1361 | if name.startswith("test"): | |
1362 | func = globals()[name] | |
1363 | try: | |
1364 | func() | |
1365 | check_allnodes() | |
1366 | except: | |
1367 | failed.append(name) | |
1368 | print "Test Failed: ", name | |
1369 | sys.stdout.flush() | |
1370 | traceback.print_exception(*sys.exc_info()) | |
1371 | print repr(sys.exc_info()[1]) | |
1372 | Node.allnodes = {} | |
1373 | ||
1374 | if failed: | |
1375 | print "\n\n\n**** Check for failures in these tests:" | |
1376 | for name in failed: | |
1377 | print " " + name |