import unittest import xml.etree.ElementTree import os.path import sys from TestUtil import PcapComparator, ID2TExecution class RegressionTest(unittest.TestCase): REGRESSION_DIRECTORY = "../test/regression_files" REGRESSION_DIRECTORY_ID2T_RELATIVE = "test/regression_files" ID2T_RELATIVE_TO_LOCAL_PREFIX = "../" META_FILE = "fileinfo.xml" def test_regression(self): config_location = self.REGRESSION_DIRECTORY + os.sep + self.META_FILE xml_root = xml.etree.ElementTree.parse(config_location).getroot() comparator = PcapComparator() for test in xml_root.getchildren(): self.assertXMLTagHasAttribute(test, "seed", "s needs a seed-attribute") self.assertXMLTagHasAttribute(test, "outfile", "s needs a outfile-attribute") self.assertXMLTagHasAttribute(test, "infile", "s needs a infile-attribute") self.assertXMLTagHasAttribute(test, "name", "s needs a name-attribute") params = [] for param in test.getchildren(): self.assertEqual(param.tag, "param", "-children must be s") self.assertIsNotNone(param.get("key"), " needs a key-attribute") self.assertIsNotNone(param.get("value"), " needs a value-attribute") params.append("%s=%s" % (param.get("key"), param.get("value"))) infile = os.path.join(self.REGRESSION_DIRECTORY_ID2T_RELATIVE, test.get("infile")) outfile = os.path.join(self.REGRESSION_DIRECTORY, test.get("outfile")) execution = ID2TExecution(infile, seed=test.get("seed")) self.print_warning(execution.get_run_command(params)) execution.run(params) new_file = self.ID2T_RELATIVE_TO_LOCAL_PREFIX + os.sep + execution.get_pcap_filename() old_file = outfile try: comparator.compare_files(new_file, old_file) except AssertionError as e: execution.cleanup() raise AssertionError("Test %s failed" % test.get("name")) from e self.print_warning("Regression-test %s passed" % test.get("name")) def assertXMLTagHasAttribute(self, tag, attribute, msg=None): self.assertIsNotNone(tag.get(attribute), msg) def print_warning(self, *text): print(*text, file=sys.stderr)