Annotation of embedaddon/libxml2/python/tests/reader3.py, revision 1.1.1.1
1.1 misho 1: #!/usr/bin/python -u
2: #
3: # this tests the entities substitutions with the XmlTextReader interface
4: #
5: import sys
6: import StringIO
7: import libxml2
8:
9: docstr="""<?xml version='1.0'?>
10: <!DOCTYPE doc [
11: <!ENTITY tst "<p>test</p>">
12: ]>
13: <doc>&tst;</doc>"""
14:
15: # Memory debug specific
16: libxml2.debugMemory(1)
17:
18: #
19: # First test, normal don't substitute entities.
20: #
21: f = StringIO.StringIO(docstr)
22: input = libxml2.inputBuffer(f)
23: reader = input.newTextReader("test_noent")
24: ret = reader.Read()
25: if ret != 1:
26: print "Error reading to root"
27: sys.exit(1)
28: if reader.Name() == "doc" or reader.NodeType() == 10:
29: ret = reader.Read()
30: if ret != 1:
31: print "Error reading to root"
32: sys.exit(1)
33: if reader.Name() != "doc" or reader.NodeType() != 1:
34: print "test_normal: Error reading the root element"
35: sys.exit(1)
36: ret = reader.Read()
37: if ret != 1:
38: print "test_normal: Error reading to the entity"
39: sys.exit(1)
40: if reader.Name() != "tst" or reader.NodeType() != 5:
41: print "test_normal: Error reading the entity"
42: sys.exit(1)
43: ret = reader.Read()
44: if ret != 1:
45: print "test_normal: Error reading to the end of root"
46: sys.exit(1)
47: if reader.Name() != "doc" or reader.NodeType() != 15:
48: print "test_normal: Error reading the end of the root element"
49: sys.exit(1)
50: ret = reader.Read()
51: if ret != 0:
52: print "test_normal: Error detecting the end"
53: sys.exit(1)
54:
55: #
56: # Second test, completely substitute the entities.
57: #
58: f = StringIO.StringIO(docstr)
59: input = libxml2.inputBuffer(f)
60: reader = input.newTextReader("test_noent")
61: reader.SetParserProp(libxml2.PARSER_SUBST_ENTITIES, 1)
62: ret = reader.Read()
63: if ret != 1:
64: print "Error reading to root"
65: sys.exit(1)
66: if reader.Name() == "doc" or reader.NodeType() == 10:
67: ret = reader.Read()
68: if ret != 1:
69: print "Error reading to root"
70: sys.exit(1)
71: if reader.Name() != "doc" or reader.NodeType() != 1:
72: print "test_noent: Error reading the root element"
73: sys.exit(1)
74: ret = reader.Read()
75: if ret != 1:
76: print "test_noent: Error reading to the entity content"
77: sys.exit(1)
78: if reader.Name() != "p" or reader.NodeType() != 1:
79: print "test_noent: Error reading the p element from entity"
80: sys.exit(1)
81: ret = reader.Read()
82: if ret != 1:
83: print "test_noent: Error reading to the text node"
84: sys.exit(1)
85: if reader.NodeType() != 3 or reader.Value() != "test":
86: print "test_noent: Error reading the text node"
87: sys.exit(1)
88: ret = reader.Read()
89: if ret != 1:
90: print "test_noent: Error reading to the end of p element"
91: sys.exit(1)
92: if reader.Name() != "p" or reader.NodeType() != 15:
93: print "test_noent: Error reading the end of the p element"
94: sys.exit(1)
95: ret = reader.Read()
96: if ret != 1:
97: print "test_noent: Error reading to the end of root"
98: sys.exit(1)
99: if reader.Name() != "doc" or reader.NodeType() != 15:
100: print "test_noent: Error reading the end of the root element"
101: sys.exit(1)
102: ret = reader.Read()
103: if ret != 0:
104: print "test_noent: Error detecting the end"
105: sys.exit(1)
106:
107: #
108: # third test, crazy stuff about empty element in external parsed entities
109: #
110: s = """<!DOCTYPE struct [
111: <!ENTITY simplestruct2.ent SYSTEM "simplestruct2.ent">
112: ]>
113: <struct>&simplestruct2.ent;</struct>
114: """
115: expect="""10 struct 0 0
116: 1 struct 0 0
117: 1 descr 1 1
118: 15 struct 0 0
119: """
120: res=""
121: simplestruct2_ent="""<descr/>"""
122:
123: def myResolver(URL, ID, ctxt):
124: if URL == "simplestruct2.ent":
125: return(StringIO.StringIO(simplestruct2_ent))
126: return None
127:
128: libxml2.setEntityLoader(myResolver)
129:
130: input = libxml2.inputBuffer(StringIO.StringIO(s))
131: reader = input.newTextReader("test3")
132: reader.SetParserProp(libxml2.PARSER_SUBST_ENTITIES,1)
133: while reader.Read() == 1:
134: res = res + "%s %s %d %d\n" % (reader.NodeType(),reader.Name(),
135: reader.Depth(),reader.IsEmptyElement())
136:
137: if res != expect:
138: print "test3 failed: unexpected output"
139: print res
140: sys.exit(1)
141:
142: #
143: # cleanup
144: #
145: del f
146: del input
147: del reader
148:
149: # Memory debug specific
150: libxml2.cleanupParser()
151: if libxml2.debugMemory(1) == 0:
152: print "OK"
153: else:
154: print "Memory leak %d bytes" % (libxml2.debugMemory(1))
155: libxml2.dumpMemory()
FreeBSD-CVSweb <freebsd-cvsweb@FreeBSD.org>