|
| 1 | +from unittest.runner import TextTestResult |
| 2 | +from .points import _parse_points, _name_test |
| 3 | +import atexit |
| 4 | +import json |
| 5 | +import traceback |
| 6 | + |
| 7 | +results = [] |
| 8 | + |
| 9 | + |
| 10 | +class TMCResult(TextTestResult): |
| 11 | + |
| 12 | + def __init__(self, stream, descriptions, verbosity): |
| 13 | + super(TMCResult, self).__init__(stream, descriptions, verbosity) |
| 14 | + |
| 15 | + def startTest(self, test): |
| 16 | + super(TMCResult, self).startTest(test) |
| 17 | + |
| 18 | + def addSuccess(self, test): |
| 19 | + super(TMCResult, self).addSuccess(test) |
| 20 | + self.addResult(test, 'passed') |
| 21 | + |
| 22 | + def addFailure(self, test, err): |
| 23 | + super(TMCResult, self).addFailure(test, err) |
| 24 | + self.addResult(test, 'failed', err) |
| 25 | + |
| 26 | + def addError(self, test, err): |
| 27 | + super(TMCResult, self).addError(test, err) |
| 28 | + self.addResult(test, 'errored', err) |
| 29 | + |
| 30 | + def addResult(self, test, status, err=None): |
| 31 | + points = _parse_points(test) |
| 32 | + message = "" |
| 33 | + backtrace = [] |
| 34 | + if err is not None: |
| 35 | + message = str(err[1]) |
| 36 | + backtrace = traceback.format_tb(err[2]) |
| 37 | + |
| 38 | + details = { |
| 39 | + 'name': _name_test(test), |
| 40 | + 'status': status, |
| 41 | + 'message': message, |
| 42 | + 'passed': status == 'passed', |
| 43 | + 'points': points, |
| 44 | + 'backtrace': backtrace |
| 45 | + } |
| 46 | + results.append(details) |
| 47 | + |
| 48 | + # TODO: Do not do this if not using TMCTestRunner |
| 49 | + @atexit.register |
| 50 | + def write_output(): |
| 51 | + with open('.tmc_test_results.json', 'w', encoding='utf8') as f: |
| 52 | + json.dump(results, f, ensure_ascii=False) |
0 commit comments