amorfo77/parsedmarc
1
0
Fork 0
mirror of https://github.com/domainaware/parsedmarc.git synced 2026-02-17 07:03:58 +00:00
Code Issues Projects Releases Wiki Activity

Update tests.py

Browse Source 
Fix code style
This commit is contained in:
Sean Whalen
2024-07-16 15:14:40 -04:00
committed by GitHub
parent 410663dbca
commit e9f0cdef1f
1 changed files with 8 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
tests.py
View File

@@ -15,12 +15,14 @@ def minify_xml(xml_string):
tree = etree.fromstring(xml_string.encode('utf-8'), parser)
return etree.tostring(tree, pretty_print=False).decode('utf-8')
def compare_xml(xml1, xml2):
parser = etree.XMLParser(remove_blank_text=True)
tree1 = etree.fromstring(xml1.encode('utf-8'), parser)
tree2 = etree.fromstring(xml2.encode('utf-8'), parser)
return etree.tostring(tree1) == etree.tostring(tree2)
class Test(unittest.TestCase):
def testBase64Decoding(self):
"""Test base64 decoding"""
@@ -44,7 +46,8 @@ class Test(unittest.TestCase):
print()
xmlnice = open("samples/extract_report/nice-input.xml").read()
print(xmlnice)
xmlchanged = minify_xml(open("samples/extract_report/changed-input.xml").read())
xmlchanged = minify_xml(open(
"samples/extract_report/changed-input.xml").read())
print(xmlchanged)
self.assertTrue(compare_xml(xmlnice, xmlnice))
self.assertTrue(compare_xml(xmlchanged, xmlchanged))
@@ -91,10 +94,12 @@ class Test(unittest.TestCase):
print("Testing {0}: " .format(file), end="")
xmlout = parsedmarc.extract_report(file)
print(xmlout)
xmlin = minify_xml(open("samples/extract_report/nice-input.xml").read())
xmlin = minify_xml(open(
"samples/extract_report/nice-input.xml").read())
print(xmlin)
self.assertTrue(compare_xml(xmlout, xmlin))
xmlin = minify_xml(open("samples/extract_report/changed-input.xml").read())
xmlin = minify_xml(open(
"samples/extract_report/changed-input.xml").read())
print(xmlin)
self.assertFalse(compare_xml(xmlout, xmlin))
print("Passed!")