mirror of
https://github.com/microsoft/onefuzz.git
synced 2025-06-21 13:51:19 +00:00
60b304a220
* handle messages that are too big to fit in a queue message * tests Co-authored-by: stas <statis@microsoft.com>
62 lines
2.1 KiB
Python
Executable File
62 lines
2.1 KiB
Python
Executable File
#!/usr/bin/env python
|
|
#
|
|
# Copyright (c) Microsoft Corporation.
|
|
# Licensed under the MIT License.
|
|
|
|
|
|
import json
|
|
import unittest
|
|
from pathlib import Path
|
|
|
|
from onefuzztypes.models import Report
|
|
|
|
from __app__.onefuzzlib.reports import fix_report_size, parse_report_or_regression
|
|
|
|
|
|
class TestReportParse(unittest.TestCase):
|
|
def test_sample(self) -> None:
|
|
report_path = Path(__file__).parent / "data" / "report.json"
|
|
with open(report_path, "r") as handle:
|
|
data = json.load(handle)
|
|
|
|
invalid = {"unused_field_1": 3}
|
|
report = parse_report_or_regression(json.dumps(data))
|
|
self.assertIsInstance(report, Report)
|
|
|
|
with self.assertLogs(level="ERROR"):
|
|
self.assertIsNone(
|
|
parse_report_or_regression('"invalid"', expect_reports=True)
|
|
)
|
|
|
|
with self.assertLogs(level="WARNING") as logs:
|
|
self.assertIsNone(
|
|
parse_report_or_regression(json.dumps(invalid), expect_reports=True)
|
|
)
|
|
self.assertTrue(any(["unable to parse report" in x for x in logs.output]))
|
|
|
|
def test_report_no_resize(self) -> None:
|
|
report_path = Path(__file__).parent / "data" / "report.json"
|
|
with open(report_path, "r") as handle:
|
|
content = handle.read()
|
|
data = json.loads(content)
|
|
report = Report.parse_obj(data)
|
|
fixed_report = fix_report_size(content, report)
|
|
self.assertEqual(report, fixed_report)
|
|
|
|
def test_report_resize(self) -> None:
|
|
report_path = Path(__file__).parent / "data" / "report-long.json"
|
|
with open(report_path, "r") as handle:
|
|
content = handle.read()
|
|
data = json.loads(content)
|
|
report = Report.parse_obj(data)
|
|
fixed_report = fix_report_size(
|
|
content, report, acceptable_report_length_kb=10, keep_num_entries=10
|
|
)
|
|
self.assertEqual(len(fixed_report.call_stack), 11) # extra item is "..."
|
|
report.call_stack = report.call_stack[0:10] + ["..."]
|
|
self.assertEqual(report, fixed_report)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
unittest.main()
|