| import json |
| import re |
| import sys |
| |
| from mozlog.structured.formatters.base import BaseFormatter |
| |
| |
| LONE_SURROGATE_RE = re.compile(u"[\uD800-\uDFFF]") |
| |
| |
| def surrogate_replacement_ucs4(match): |
| return "U+" + hex(ord(match.group()))[2:] |
| |
| |
| class SurrogateReplacementUcs2(object): |
| def __init__(self): |
| self.skip = False |
| |
| def __call__(self, match): |
| char = match.group() |
| |
| if self.skip: |
| self.skip = False |
| return char |
| |
| is_low = 0xD800 <= ord(char) <= 0xDBFF |
| |
| escape = True |
| if is_low: |
| next_idx = match.end() |
| if next_idx < len(match.string): |
| next_char = match.string[next_idx] |
| if 0xDC00 <= ord(next_char) <= 0xDFFF: |
| escape = False |
| |
| if not escape: |
| self.skip = True |
| return char |
| |
| return "U+" + hex(ord(match.group()))[2:] |
| |
| |
| if sys.maxunicode == 0x10FFFF: |
| surrogate_replacement = surrogate_replacement_ucs4 |
| else: |
| surrogate_replacement = SurrogateReplacementUcs2() |
| |
| |
| def replace_lone_surrogate(data): |
| return LONE_SURROGATE_RE.subn(surrogate_replacement, data)[0] |
| |
| |
| class WptreportFormatter(BaseFormatter): |
| """Formatter that produces results in the format that wptreport expects.""" |
| |
| def __init__(self): |
| self.raw_results = {} |
| self.results = {} |
| |
| def suite_start(self, data): |
| if 'run_info' in data: |
| self.results['run_info'] = data['run_info'] |
| self.results['time_start'] = data['time'] |
| |
| def suite_end(self, data): |
| self.results['time_end'] = data['time'] |
| self.results["results"] = [] |
| for test_name in self.raw_results: |
| result = {"test": test_name} |
| result.update(self.raw_results[test_name]) |
| self.results["results"].append(result) |
| return json.dumps(self.results) |
| |
| def find_or_create_test(self, data): |
| test_name = data["test"] |
| if test_name not in self.raw_results: |
| self.raw_results[test_name] = { |
| "subtests": [], |
| "status": "", |
| "message": None |
| } |
| return self.raw_results[test_name] |
| |
| def test_start(self, data): |
| test = self.find_or_create_test(data) |
| test["start_time"] = data["time"] |
| |
| def create_subtest(self, data): |
| test = self.find_or_create_test(data) |
| subtest_name = replace_lone_surrogate(data["subtest"]) |
| |
| subtest = { |
| "name": subtest_name, |
| "status": "", |
| "message": None |
| } |
| test["subtests"].append(subtest) |
| |
| return subtest |
| |
| def test_status(self, data): |
| subtest = self.create_subtest(data) |
| subtest["status"] = data["status"] |
| if "expected" in data: |
| subtest["expected"] = data["expected"] |
| if "message" in data: |
| subtest["message"] = replace_lone_surrogate(data["message"]) |
| |
| def test_end(self, data): |
| test = self.find_or_create_test(data) |
| start_time = test.pop("start_time") |
| test["duration"] = data["time"] - start_time |
| test["status"] = data["status"] |
| if "expected" in data: |
| test["expected"] = data["expected"] |
| if "message" in data: |
| test["message"] = replace_lone_surrogate(data["message"]) |
| |
| def assertion_count(self, data): |
| test = self.find_or_create_test(data) |
| test["asserts"] = { |
| "count": data["count"], |
| "min": data["min_expected"], |
| "max": data["max_expected"] |
| } |
