mirror of
				https://gitlab.gnome.org/GNOME/libxml2.git
				synced 2025-10-30 10:45:36 +03:00 
			
		
		
		
	
		
			
				
	
	
		
			161 lines
		
	
	
		
			4.1 KiB
		
	
	
	
		
			Python
		
	
	
		
			Executable File
		
	
	
	
	
			
		
		
	
	
			161 lines
		
	
	
		
			4.1 KiB
		
	
	
	
		
			Python
		
	
	
		
			Executable File
		
	
	
	
	
| #!/usr/bin/python -u
 | |
| #
 | |
| # this tests the entities substitutions with the XmlTextReader interface
 | |
| #
 | |
| import sys
 | |
| import libxml2
 | |
| try:
 | |
|     import StringIO
 | |
|     str_io = StringIO.StringIO
 | |
| except:
 | |
|     import io
 | |
|     str_io = io.StringIO
 | |
| 
 | |
| docstr="""<?xml version='1.0'?>
 | |
| <!DOCTYPE doc [
 | |
| <!ENTITY tst "<p>test</p>">
 | |
| ]>
 | |
| <doc>&tst;</doc>"""
 | |
| 
 | |
| # Memory debug specific
 | |
| libxml2.debugMemory(1)
 | |
| 
 | |
| #
 | |
| # First test, normal don't substitute entities.
 | |
| #
 | |
| f = str_io(docstr)
 | |
| input = libxml2.inputBuffer(f)
 | |
| reader = input.newTextReader("test_noent")
 | |
| ret = reader.Read()
 | |
| if ret != 1:
 | |
|     print("Error reading to root")
 | |
|     sys.exit(1)
 | |
| if reader.Name() == "doc" or reader.NodeType() == 10:
 | |
|     ret = reader.Read()
 | |
| if ret != 1:
 | |
|     print("Error reading to root")
 | |
|     sys.exit(1)
 | |
| if reader.Name() != "doc" or reader.NodeType() != 1:
 | |
|     print("test_normal: Error reading the root element")
 | |
|     sys.exit(1)
 | |
| ret = reader.Read()
 | |
| if ret != 1:
 | |
|     print("test_normal: Error reading to the entity")
 | |
|     sys.exit(1)
 | |
| if reader.Name() != "tst" or reader.NodeType() != 5:
 | |
|     print("test_normal: Error reading the entity")
 | |
|     sys.exit(1)
 | |
| ret = reader.Read()
 | |
| if ret != 1:
 | |
|     print("test_normal: Error reading to the end of root")
 | |
|     sys.exit(1)
 | |
| if reader.Name() != "doc" or reader.NodeType() != 15:
 | |
|     print("test_normal: Error reading the end of the root element")
 | |
|     sys.exit(1)
 | |
| ret = reader.Read()
 | |
| if ret != 0:
 | |
|     print("test_normal: Error detecting the end")
 | |
|     sys.exit(1)
 | |
| 
 | |
| #
 | |
| # Second test, completely substitute the entities.
 | |
| #
 | |
| f = str_io(docstr)
 | |
| input = libxml2.inputBuffer(f)
 | |
| reader = input.newTextReader("test_noent")
 | |
| reader.SetParserProp(libxml2.PARSER_SUBST_ENTITIES, 1)
 | |
| ret = reader.Read()
 | |
| if ret != 1:
 | |
|     print("Error reading to root")
 | |
|     sys.exit(1)
 | |
| if reader.Name() == "doc" or reader.NodeType() == 10:
 | |
|     ret = reader.Read()
 | |
| if ret != 1:
 | |
|     print("Error reading to root")
 | |
|     sys.exit(1)
 | |
| if reader.Name() != "doc" or reader.NodeType() != 1:
 | |
|     print("test_noent: Error reading the root element")
 | |
|     sys.exit(1)
 | |
| ret = reader.Read()
 | |
| if ret != 1:
 | |
|     print("test_noent: Error reading to the entity content")
 | |
|     sys.exit(1)
 | |
| if reader.Name() != "p" or reader.NodeType() != 1:
 | |
|     print("test_noent: Error reading the p element from entity")
 | |
|     sys.exit(1)
 | |
| ret = reader.Read()
 | |
| if ret != 1:
 | |
|     print("test_noent: Error reading to the text node")
 | |
|     sys.exit(1)
 | |
| if reader.NodeType() != 3 or reader.Value() != "test":
 | |
|     print("test_noent: Error reading the text node")
 | |
|     sys.exit(1)
 | |
| ret = reader.Read()
 | |
| if ret != 1:
 | |
|     print("test_noent: Error reading to the end of p element")
 | |
|     sys.exit(1)
 | |
| if reader.Name() != "p" or reader.NodeType() != 15:
 | |
|     print("test_noent: Error reading the end of the p element")
 | |
|     sys.exit(1)
 | |
| ret = reader.Read()
 | |
| if ret != 1:
 | |
|     print("test_noent: Error reading to the end of root")
 | |
|     sys.exit(1)
 | |
| if reader.Name() != "doc" or reader.NodeType() != 15:
 | |
|     print("test_noent: Error reading the end of the root element")
 | |
|     sys.exit(1)
 | |
| ret = reader.Read()
 | |
| if ret != 0:
 | |
|     print("test_noent: Error detecting the end")
 | |
|     sys.exit(1)
 | |
| 
 | |
| #
 | |
| # third test, crazy stuff about empty element in external parsed entities
 | |
| #
 | |
| s = """<!DOCTYPE struct [
 | |
| <!ENTITY simplestruct2.ent SYSTEM "simplestruct2.ent">
 | |
| ]>
 | |
| <struct>&simplestruct2.ent;</struct>
 | |
| """
 | |
| expect="""10 struct 0 0
 | |
| 1 struct 0 0
 | |
| 1 descr 1 1
 | |
| 15 struct 0 0
 | |
| """
 | |
| res=""
 | |
| simplestruct2_ent="""<descr/>"""
 | |
| 
 | |
| def myResolver(URL, ID, ctxt):
 | |
|     if URL == "simplestruct2.ent":
 | |
|         return(str_io(simplestruct2_ent))
 | |
|     return None
 | |
| 
 | |
| libxml2.setEntityLoader(myResolver)
 | |
| 
 | |
| input = libxml2.inputBuffer(str_io(s))
 | |
| reader = input.newTextReader("test3")
 | |
| reader.SetParserProp(libxml2.PARSER_SUBST_ENTITIES,1)
 | |
| while reader.Read() == 1:
 | |
|     res = res + "%s %s %d %d\n" % (reader.NodeType(),reader.Name(),
 | |
|                                    reader.Depth(),reader.IsEmptyElement())
 | |
| 
 | |
| if res != expect:
 | |
|     print("test3 failed: unexpected output")
 | |
|     print(res)
 | |
|     sys.exit(1)
 | |
| 
 | |
| #
 | |
| # cleanup
 | |
| #
 | |
| del f
 | |
| del input
 | |
| del reader
 | |
| 
 | |
| # Memory debug specific
 | |
| libxml2.cleanupParser()
 | |
| if libxml2.debugMemory(1) == 0:
 | |
|     print("OK")
 | |
| else:
 | |
|     print("Memory leak %d bytes" % (libxml2.debugMemory(1)))
 | |
|     libxml2.dumpMemory()
 |