from lxml import etree

from io import StringIO

def test_parse():

tree = etree.parse(StringIO('<foo><bar></bar></foo>')) # $ decodeFormat=XML decodeInput=StringIO(..) decodeOutput=etree.parse(..) xmlVuln='XXE' getAPathArgument=StringIO(..)

r = tree.xpath('/foo/bar') # $ getXPath='/foo/bar'

def test_XPath_class():

root = etree.XML("<root><a>TEXT</a></root>") # $ decodeFormat=XML decodeInput="<root><a>TEXT</a></root>" decodeOutput=etree.XML(..) xmlVuln='XXE'

find_text = etree.XPath("path") # $ constructedXPath="path"

text = find_text(root)[0]

def test_ETXpath_class():

root = etree.XML("<root><a>TEXT</a></root>") # $ decodeFormat=XML decodeInput="<root><a>TEXT</a></root>" decodeOutput=etree.XML(..) xmlVuln='XXE'

find_text = etree.ETXPath("path") # $ constructedXPath="path"

text = find_text(root)[0]

def test_XPathEvaluator_class():

root = etree.XML("<root><a>TEXT</a></root>") # $ decodeFormat=XML decodeInput="<root><a>TEXT</a></root>" decodeOutput=etree.XML(..) xmlVuln='XXE'

search_root = etree.XPathEvaluator(root)

text = search_root("path")[0] # $ getXPath="path"