casa6

class Xunit:

    xml_escape_table = {

        "&": "&",

        '"': """,

        "'": "'",

        ">": ">",

        "<": "&lt;",

    }

​

    results = []

    fail_total = 0

    def append_result (self,testname, runtime, returncode, testerr):

        result =    {

          "testname": testname,

          "runtime" : runtime,

          "returncode": returncode,

          "testerr": testerr,

        }

        self.results.append(result)

​

​

    def xml_escape(self, text):

        return "".join(self.xml_escape_table.get(c,c) for c in text)

​

    def test_result_to_xml (self,result):

        self.fail_total = self.fail_total + len(result['testerr'])

        testxml = '<testcase classname="' + result['testname'].replace(".py", "") + '"' \

              + ' name="Failed tests" time="' + result['runtime'] + '">'

        if ( result['returncode'] != 0) :

            testxml = testxml + '<failure>' + str(result['testerr']) + '</failure>'

            if self.fail_total == 0:

                self.fail_total=self.fail_total + 1

        testxml = testxml + '</testcase>\n'

        return testxml

​

    def generateXml(self, testname):

        xmlResults = list(map(lambda result: self.test_result_to_xml(result), self.results))

​

        testHeader = '<?xml version="1.0" encoding="UTF-8"?>' + "\n" \

                 + '<testsuite name="UnitTests" tests="' \

                 + str(len(self.results)) + '" errors="'+ str(self.fail_total) + '"' \

                 + ' failures="' + str(self.fail_total) + '" skip="0">\n'

        print("Results: " + str(self.results))

        testFooter ="\n</testsuite>"

​

        # Write xUnit.xml

        xUnit = open("xUnit-"+testname+".xml", "w+")

        xUnit.write(testHeader + ''.join(xmlResults) + testFooter)

        xUnit.close()