forked from mirrors/gecko-dev
e8a78d6118
There's some unittest-related functions that we heavily lean on that are deprecated: https://docs.python.org/3/library/unittest.html#deprecated-aliases This is a big find-and-replace that was restricted based on files that matched the pattern `*test*.py` and that weren't in any of the paths listed in `tools/rewriting/ThirdPartyPaths.txt`. Differential Revision: https://phabricator.services.mozilla.com/D138608
68 lines
2.6 KiB
Python
68 lines
2.6 KiB
Python
from __future__ import absolute_import
|
|
import unittest
|
|
from mozharness.mozilla.structuredlog import StructuredOutputParser
|
|
from mozharness.base.log import INFO, WARNING
|
|
from mozharness.mozilla.automation import TBPL_SUCCESS, TBPL_WARNING
|
|
from mozharness.mozilla.mozbase import MozbaseMixin
|
|
from mozlog.handlers.statushandler import RunSummary
|
|
|
|
success_summary = RunSummary(
|
|
unexpected_statuses={},
|
|
expected_statuses={"PASS": 3, "OK": 1, "FAIL": 1},
|
|
known_intermittent_statuses={"FAIL": 1},
|
|
log_level_counts={"info": 5},
|
|
action_counts={"test_status": 4, "test_end": 1, "suite_end": 1},
|
|
)
|
|
|
|
failure_summary = RunSummary(
|
|
unexpected_statuses={"FAIL": 2},
|
|
expected_statuses={"PASS": 2, "OK": 1},
|
|
known_intermittent_statuses={},
|
|
log_level_counts={"warning": 2, "info": 3},
|
|
action_counts={"test_status": 3, "test_end": 2, "suite_end": 1},
|
|
)
|
|
|
|
|
|
class TestParser(MozbaseMixin, StructuredOutputParser):
|
|
def __init__(self, *args, **kwargs):
|
|
super(TestParser, self).__init__(*args, **kwargs)
|
|
self.config = {}
|
|
|
|
|
|
class TestStructuredOutputParser(unittest.TestCase):
|
|
def setUp(self):
|
|
self.parser = TestParser()
|
|
|
|
def test_evaluate_parser_success(self):
|
|
self.parser.handler.expected_statuses = {"PASS": 3, "OK": 1, "FAIL": 1}
|
|
self.parser.handler.log_level_counts = {"info": 5}
|
|
self.parser.handler.action_counts = {
|
|
"test_status": 4,
|
|
"test_end": 1,
|
|
"suite_end": 1,
|
|
}
|
|
self.parser.handler.known_intermittent_statuses = {"FAIL": 1}
|
|
result = self.parser.evaluate_parser(
|
|
return_code=TBPL_SUCCESS, success_codes=[TBPL_SUCCESS]
|
|
)
|
|
tbpl_status, worst_log_level, joined_summary = result
|
|
self.assertEqual(tbpl_status, TBPL_SUCCESS)
|
|
self.assertEqual(worst_log_level, INFO)
|
|
self.assertEqual(joined_summary, success_summary)
|
|
|
|
def test_evaluate_parser_failure(self):
|
|
self.parser.handler.unexpected_statuses = {"FAIL": 2}
|
|
self.parser.handler.expected_statuses = {"PASS": 2, "OK": 1}
|
|
self.parser.handler.log_level_counts = {"warning": 2, "info": 3}
|
|
self.parser.handler.action_counts = {
|
|
"test_status": 3,
|
|
"test_end": 2,
|
|
"suite_end": 1,
|
|
}
|
|
result = self.parser.evaluate_parser(
|
|
return_code=TBPL_SUCCESS, success_codes=[TBPL_SUCCESS]
|
|
)
|
|
tbpl_status, worst_log_level, joined_summary = result
|
|
self.assertEqual(tbpl_status, TBPL_WARNING)
|
|
self.assertEqual(worst_log_level, WARNING)
|
|
self.assertEqual(joined_summary, failure_summary)
|