79 lines
3.2 KiB
Python
79 lines
3.2 KiB
Python
|
"""Exercise kvb-check-ansible's bench: the synthetic carrier fact and a real run of the bench playbook."""
|
||
|
|
|
||
|
|
from __future__ import annotations
|
||
|
|
|
||
|
|
import json
|
||
|
|
import os
|
||
|
|
import shutil
|
||
|
|
import subprocess
|
||
|
|
import sys
|
||
|
|
import tempfile
|
||
|
|
import unittest
|
||
|
|
from pathlib import Path
|
||
|
|
|
||
|
|
from ansible_support import BENCH, SCRIPT, LogCase, ansible_bench, ansible_log
|
||
|
|
|
||
|
|
|
||
|
|
class TestBenchCarrier(LogCase):
|
||
|
|
def test_writes_exactly_the_asked_size(self):
|
||
|
|
out = self.base / "carrier.json"
|
||
|
|
result = self.run_script("bench-carrier", "--kb", "3", "--out", str(out))
|
||
|
|
self.assertEqual(result.returncode, 0, result.stderr)
|
||
|
|
self.assertEqual(out.stat().st_size, 3 * 1024)
|
||
|
|
fact = json.loads(out.read_text(encoding="ascii"))
|
||
|
|
self.assertEqual(result.stdout, f"{out}\t3072 bytes\t{len(fact) - 1} entries\n")
|
||
|
|
self.assertIn("pad", fact)
|
||
|
|
for size in (1024, 1500, 400 * 1024):
|
||
|
|
with self.subTest(size=size):
|
||
|
|
self.assertEqual(len(json.dumps(ansible_bench.carrier(size))), size)
|
||
|
|
self.assertGreater(len(ansible_bench.carrier(400 * 1024)), 100)
|
||
|
|
|
||
|
|
def test_bench_inventory_pins_no_interpreter_path(self):
|
||
|
|
self.assertNotRegex((BENCH / "hosts.yml").read_text(encoding="utf-8"), r"interpreter:\s*/")
|
||
|
|
|
||
|
|
|
||
|
|
@unittest.skipUnless(shutil.which("ansible-playbook"), "ansible-playbook not installed")
|
||
|
|
class TestBench(unittest.TestCase):
|
||
|
|
@classmethod
|
||
|
|
def setUpClass(cls):
|
||
|
|
tmp = tempfile.TemporaryDirectory()
|
||
|
|
cls.addClassCleanup(tmp.cleanup)
|
||
|
|
base = Path(tmp.name)
|
||
|
|
carrier = base / "carrier.json"
|
||
|
|
carrier.write_text(json.dumps(ansible_bench.carrier(8 * 1024)), encoding="ascii")
|
||
|
|
cls.log = base / "bench.log"
|
||
|
|
env = {
|
||
|
|
"PATH": os.environ["PATH"],
|
||
|
|
"HOME": str(base),
|
||
|
|
"ANSIBLE_LOCAL_TEMP": str(base / "tmp"),
|
||
|
|
"ANSIBLE_NOCOLOR": "1",
|
||
|
|
"ANSIBLE_CALLBACKS_ENABLED": "ansible.posix.profile_tasks",
|
||
|
|
}
|
||
|
|
cls.result = subprocess.run(
|
||
|
|
[
|
||
|
|
"ansible-playbook", "-i", str(BENCH / "hosts.yml"), str(BENCH / "bench.yml"),
|
||
|
|
"-e", "bench_count=3", "-e", f"bench_carrier_file={carrier}",
|
||
|
|
],
|
||
|
|
cwd=base, env=env, capture_output=True, text=True, check=False,
|
||
|
|
)
|
||
|
|
cls.log.write_text(cls.result.stdout, encoding="utf-8")
|
||
|
|
|
||
|
|
def test_runs_one_skipped_and_one_ran_task_per_item(self):
|
||
|
|
self.assertEqual(self.result.returncode, 0, self.result.stdout + self.result.stderr)
|
||
|
|
self.assertEqual(self.result.stdout.count("skipping: [bench]"), 3)
|
||
|
|
self.assertEqual(self.result.stdout.count("included: "), 3)
|
||
|
|
|
||
|
|
def test_profile_tasks_output_ranks_by_kind(self):
|
||
|
|
if ansible_log.ELAPSED.search(self.result.stdout) is None:
|
||
|
|
self.skipTest("ansible.posix.profile_tasks not installed")
|
||
|
|
result = subprocess.run(
|
||
|
|
[sys.executable, str(SCRIPT), "kinds", "--log", str(self.log), "--json"],
|
||
|
|
capture_output=True, text=True, check=False,
|
||
|
|
)
|
||
|
|
self.assertEqual(result.returncode, 0, result.stderr)
|
||
|
|
counts = {row["kind"]: row["count"] for row in json.loads(result.stdout)["kinds"]}
|
||
|
|
self.assertEqual(counts, {"ran": 4, "skipped": 3, "included": 1})
|
||
|
|
|
||
|
|
|
||
|
|
if __name__ == "__main__":
|
||
|
|
unittest.main()
|