97 lines
3.6 KiB
Python
97 lines
3.6 KiB
Python
|
"""Exercise kvb-check-ansible's reporting subcommands: name, kind and occurrence tables, JSON and usage errors."""
|
||
|
|
|
||
|
|
from __future__ import annotations
|
||
|
|
|
||
|
|
import unittest
|
||
|
|
|
||
|
|
from ansible_support import WALL_KINDS, LogCase, ansible_tables
|
||
|
|
|
||
|
|
|
||
|
|
class TestNameTables(LogCase):
|
||
|
|
def test_gaps_sum_per_name_biggest_first_with_total(self):
|
||
|
|
self.assertEqual(
|
||
|
|
self.lines("gaps", "--log", str(self.wall)),
|
||
|
|
[
|
||
|
|
"5.0\t2\tsetup : install",
|
||
|
|
"1.2\t1\tsetup : restart",
|
||
|
|
"0.8\t1\tsetup : flush",
|
||
|
|
"0.3\t1\tsetup : include",
|
||
|
|
"0.2\t1\tsetup : check",
|
||
|
|
"7.5\t6\tTOTAL",
|
||
|
|
],
|
||
|
|
)
|
||
|
|
|
||
|
|
def test_gaps_json_and_top_keep_the_full_total(self):
|
||
|
|
self.assertEqual(
|
||
|
|
self.report("gaps", "--log", str(self.wall), "--top", "2"),
|
||
|
|
{
|
||
|
|
"source": "wall",
|
||
|
|
"rows": [
|
||
|
|
{"name": "setup : install", "sum": 5.0, "count": 2},
|
||
|
|
{"name": "setup : restart", "sum": 1.2, "count": 1},
|
||
|
|
],
|
||
|
|
"total": {"sum": 7.5, "count": 6},
|
||
|
|
},
|
||
|
|
)
|
||
|
|
|
||
|
|
def test_kind_sums_one_kind_per_name(self):
|
||
|
|
self.assertEqual(
|
||
|
|
self.lines("kind", "--log", str(self.wall), "--kind", "ran"),
|
||
|
|
["5.0\t2\tsetup : install", "1.2\t1\tsetup : restart", "6.2\t3\tTOTAL"],
|
||
|
|
)
|
||
|
|
self.assertEqual(self.lines("kind", "--log", str(self.wall), "--kind", "other"), ["0.8\t1\tsetup : flush", "0.8\t1\tTOTAL"])
|
||
|
|
|
||
|
|
|
||
|
|
class TestKindTable(LogCase):
|
||
|
|
def test_kinds_detect_ran_before_skipped_and_fill_the_buckets(self):
|
||
|
|
self.assertEqual(self.lines("kinds", "--log", str(self.wall)), WALL_KINDS)
|
||
|
|
found = self.report("kinds", "--log", str(self.wall))
|
||
|
|
self.assertEqual(found["source"], "wall")
|
||
|
|
self.assertEqual(
|
||
|
|
found["kinds"][0],
|
||
|
|
{"kind": "ran", "count": 3, "sum": 6.2, "mean": 2.0667,
|
||
|
|
"buckets": {"<0.25": 0, "<0.5": 0, "<1": 0, "<2": 2, ">=2": 1}},
|
||
|
|
)
|
||
|
|
|
||
|
|
def test_bucket_edges(self):
|
||
|
|
for seconds, label in ((0.0, "<0.25"), (0.2499, "<0.25"), (0.25, "<0.5"), (0.5, "<1"), (1.0, "<2"), (1.999, "<2"), (2.0, ">=2")):
|
||
|
|
with self.subTest(seconds=seconds):
|
||
|
|
self.assertEqual(ansible_tables.bucket(seconds), label)
|
||
|
|
|
||
|
|
|
||
|
|
class TestOccurrences(LogCase):
|
||
|
|
def test_task_lists_every_occurrence_of_the_needle(self):
|
||
|
|
self.assertEqual(
|
||
|
|
self.lines("task", "--log", str(self.wall), "--needle", "install"),
|
||
|
|
["2\t3.500\tran\tsetup : install", "6\t1.500\tran\tsetup : install"],
|
||
|
|
)
|
||
|
|
found = self.report("task", "--log", str(self.wall), "--needle", "install", "--top", "1")
|
||
|
|
self.assertEqual(
|
||
|
|
found,
|
||
|
|
{
|
||
|
|
"source": "wall",
|
||
|
|
"needle": "install",
|
||
|
|
"count": 2,
|
||
|
|
"sum": 5.0,
|
||
|
|
"occurrences": [{"line": 2, "seconds": 3.5, "kind": "ran", "name": "setup : install"}],
|
||
|
|
},
|
||
|
|
)
|
||
|
|
|
||
|
|
|
||
|
|
class TestUsage(LogCase):
|
||
|
|
def test_usage_errors_exit_2(self):
|
||
|
|
for args in (
|
||
|
|
("gaps",),
|
||
|
|
("gaps", "--log", str(self.wall), "--from", "5", "--to", "4"),
|
||
|
|
("gaps", "--log", str(self.wall), "--top", "0"),
|
||
|
|
("kind", "--log", str(self.wall), "--kind", "slow"),
|
||
|
|
("task", "--log", str(self.wall)),
|
||
|
|
("profile-report", "--dir", str(self.base)),
|
||
|
|
("bench-carrier", "--kb", "0", "--out", str(self.base / "c.json")),
|
||
|
|
):
|
||
|
|
with self.subTest(args=args):
|
||
|
|
self.assertEqual(self.run_script(*args).returncode, 2)
|
||
|
|
|
||
|
|
|
||
|
|
if __name__ == "__main__":
|
||
|
|
unittest.main()
|