mirror of
https://github.com/MacRimi/ProxMenux.git
synced 2026-11-02 10:52:22 +00:00
feat(post-install): add optional zram profile and metrics
This commit is contained in:
Gabriel Gonzalez Pineda
@@ -0,0 +1,110 @@
|
||||
"""Tests for reading zram metrics from injectable proc and sys paths."""
|
||||
|
||||
import importlib
|
||||
import importlib.util
|
||||
import sys
|
||||
import tempfile
|
||||
import unittest
|
||||
from pathlib import Path
|
||||
|
||||
|
||||
SCRIPTS_DIR = Path(__file__).resolve().parents[1]
|
||||
if str(SCRIPTS_DIR) not in sys.path:
|
||||
sys.path.insert(0, str(SCRIPTS_DIR))
|
||||
|
||||
class ZramMetricsTests(unittest.TestCase):
|
||||
def test_reader_module_is_available(self):
|
||||
module_spec = importlib.util.find_spec("zram_metrics")
|
||||
self.assertIsNotNone(module_spec)
|
||||
module = importlib.import_module("zram_metrics")
|
||||
self.assertTrue(callable(getattr(module, "read_zram_metrics", None)))
|
||||
|
||||
def setUp(self):
|
||||
self.temporary = tempfile.TemporaryDirectory()
|
||||
self.addCleanup(self.temporary.cleanup)
|
||||
self.root = Path(self.temporary.name)
|
||||
self.block_root = self.root / "sys" / "block"
|
||||
self.block_root.mkdir(parents=True)
|
||||
self.swaps_path = self.root / "proc" / "swaps"
|
||||
self.swaps_path.parent.mkdir(parents=True)
|
||||
self.swaps_path.write_text(
|
||||
"Filename\tType\tSize\tUsed\tPriority\n", encoding="utf-8"
|
||||
)
|
||||
|
||||
def add_device(self, name="zram0", mm_stat="2097152 1048576 1179648 0"):
|
||||
device = self.block_root / name
|
||||
device.mkdir()
|
||||
(device / "disksize").write_text("1073741824\n", encoding="utf-8")
|
||||
(device / "mm_stat").write_text(mm_stat + "\n", encoding="utf-8")
|
||||
(device / "comp_algorithm").write_text(
|
||||
"lzo lzo-rle [zstd] lz4\n", encoding="utf-8"
|
||||
)
|
||||
return device
|
||||
|
||||
def read_metrics(self):
|
||||
from zram_metrics import read_zram_metrics
|
||||
|
||||
return read_zram_metrics(self.block_root, self.swaps_path)
|
||||
|
||||
def test_no_zram_devices_reports_unavailable(self):
|
||||
(self.block_root / "sda").mkdir()
|
||||
|
||||
self.assertEqual(self.read_metrics(), {"available": False, "devices": []})
|
||||
|
||||
def test_reads_capacity_swap_usage_and_compression_counters(self):
|
||||
self.add_device()
|
||||
self.swaps_path.write_text(
|
||||
"Filename\tType\tSize\tUsed\tPriority\n"
|
||||
"/dev/zram0\tpartition\t1048576\t1024\t100\n",
|
||||
encoding="utf-8",
|
||||
)
|
||||
|
||||
self.assertEqual(
|
||||
self.read_metrics(),
|
||||
{
|
||||
"available": True,
|
||||
"devices": [
|
||||
{
|
||||
"name": "zram0",
|
||||
"active": True,
|
||||
"capacity_bytes": 1073741824,
|
||||
"swap_used_bytes": 1048576,
|
||||
"original_bytes": 2097152,
|
||||
"compressed_bytes": 1048576,
|
||||
"memory_used_bytes": 1179648,
|
||||
"algorithm": "zstd",
|
||||
"compression_ratio": 2.0,
|
||||
"saved_bytes": 917504,
|
||||
}
|
||||
],
|
||||
},
|
||||
)
|
||||
|
||||
def test_zero_compressed_bytes_has_null_ratio(self):
|
||||
self.add_device(mm_stat="2097152 0 0 0")
|
||||
|
||||
result = self.read_metrics()
|
||||
|
||||
self.assertTrue(result["available"])
|
||||
self.assertEqual(len(result["devices"]), 1)
|
||||
self.assertFalse(result["devices"][0]["active"])
|
||||
self.assertEqual(result["devices"][0]["swap_used_bytes"], 0)
|
||||
self.assertIsNone(result["devices"][0]["compression_ratio"])
|
||||
|
||||
def test_malformed_or_short_mm_stat_omits_device(self):
|
||||
for mm_stat in ("not-a-number 1048576 1179648", "2097152 1048576"):
|
||||
with self.subTest(mm_stat=mm_stat):
|
||||
device = self.block_root / "zram0"
|
||||
if device.exists():
|
||||
(device / "mm_stat").write_text(mm_stat + "\n", encoding="utf-8")
|
||||
else:
|
||||
self.add_device(mm_stat=mm_stat)
|
||||
|
||||
result = self.read_metrics()
|
||||
|
||||
self.assertFalse(result["available"])
|
||||
self.assertEqual(result["devices"], [])
|
||||
|
||||
|
||||
if __name__ == "__main__":
|
||||
unittest.main()
|
||||
@@ -0,0 +1,166 @@
|
||||
"""Contract tests for the opt-in ZRAM profile's pure Bash helpers."""
|
||||
|
||||
import subprocess
|
||||
import tempfile
|
||||
import unittest
|
||||
from pathlib import Path
|
||||
|
||||
|
||||
HELPERS = Path(__file__).resolve().parents[3] / "scripts/post_install/zram_profile_helpers.sh"
|
||||
SWAPS_HEADER = "Filename\tType\tSize\tUsed\tPriority\n"
|
||||
|
||||
|
||||
class HelperAvailabilityTests(unittest.TestCase):
|
||||
def test_helper_file_is_available(self):
|
||||
self.assertTrue(HELPERS.is_file(), f"ZRAM helper is missing: {HELPERS}")
|
||||
|
||||
|
||||
def call_helper(function, *arguments):
|
||||
return subprocess.run(
|
||||
[
|
||||
"bash",
|
||||
"-c",
|
||||
'source "$1" || exit; shift; "$@"',
|
||||
"bash",
|
||||
str(HELPERS),
|
||||
function,
|
||||
*(str(argument) for argument in arguments),
|
||||
],
|
||||
capture_output=True,
|
||||
text=True,
|
||||
check=False,
|
||||
)
|
||||
|
||||
|
||||
class ZramGeneratorConfigTests(unittest.TestCase):
|
||||
def assert_config(self, supports_algorithm_params, expected_algorithm):
|
||||
result = call_helper("zram_render_generator_config", supports_algorithm_params)
|
||||
self.assertEqual(result.returncode, 0, result.stderr)
|
||||
settings = {
|
||||
key.strip(): value.strip()
|
||||
for line in result.stdout.splitlines()
|
||||
if "=" in line
|
||||
for key, value in [line.split("=", 1)]
|
||||
}
|
||||
self.assertIn("[zram0]", result.stdout.splitlines())
|
||||
self.assertEqual(settings.get("zram-size", "").strip(), "ram / 10")
|
||||
self.assertEqual(settings.get("swap-priority", "").strip(), "100")
|
||||
self.assertEqual(settings.get("compression-algorithm", "").strip(), expected_algorithm)
|
||||
self.assertNotIn("writeback", result.stdout.lower())
|
||||
|
||||
def test_generator_1_1_2_2_uses_plain_zstd(self):
|
||||
self.assert_config("false", "zstd")
|
||||
|
||||
def test_generator_1_2_1_2_uses_zstd_level_nine(self):
|
||||
self.assert_config("true", "zstd(level=9)")
|
||||
|
||||
|
||||
class NonZramSwapDetectionTests(unittest.TestCase):
|
||||
def setUp(self):
|
||||
self.temporary = tempfile.TemporaryDirectory()
|
||||
self.addCleanup(self.temporary.cleanup)
|
||||
directory = Path(self.temporary.name)
|
||||
self.swaps_file = directory / "swaps"
|
||||
self.fstab_file = directory / "fstab"
|
||||
|
||||
def detect(self, swaps, fstab):
|
||||
self.swaps_file.write_text(swaps, encoding="utf-8")
|
||||
self.fstab_file.write_text(fstab, encoding="utf-8")
|
||||
return call_helper("zram_has_non_zram_swap", self.swaps_file, self.fstab_file)
|
||||
|
||||
def test_active_disk_swap_is_detected(self):
|
||||
result = self.detect(
|
||||
SWAPS_HEADER + "/dev/nvme0n1p3\tpartition\t1048576\t0\t-2\n",
|
||||
"",
|
||||
)
|
||||
self.assertEqual(result.returncode, 0, result.stderr)
|
||||
|
||||
def test_persistent_swapfile_is_detected(self):
|
||||
result = self.detect(SWAPS_HEADER, "/swapfile none swap sw 0 0\n")
|
||||
self.assertEqual(result.returncode, 0, result.stderr)
|
||||
|
||||
def test_commented_disk_swap_is_ignored(self):
|
||||
result = self.detect(SWAPS_HEADER, " # /swapfile none swap sw 0 0\n")
|
||||
self.assertEqual(result.returncode, 1, result.stderr)
|
||||
|
||||
def test_active_and_persistent_zram_are_ignored(self):
|
||||
result = self.detect(
|
||||
SWAPS_HEADER + "/dev/zram0\tpartition\t1048576\t0\t100\n",
|
||||
"/dev/zram0 none swap defaults 0 0\n",
|
||||
)
|
||||
self.assertEqual(result.returncode, 1, result.stderr)
|
||||
|
||||
def test_regular_fstab_entry_does_not_count_as_swap(self):
|
||||
result = self.detect(SWAPS_HEADER, "/dev/nvme0n1p2 / ext4 defaults 0 1\n")
|
||||
self.assertEqual(result.returncode, 1, result.stderr)
|
||||
|
||||
|
||||
class ForeignZramConfigTests(unittest.TestCase):
|
||||
def setUp(self):
|
||||
self.temporary = tempfile.TemporaryDirectory()
|
||||
self.addCleanup(self.temporary.cleanup)
|
||||
self.root = Path(self.temporary.name)
|
||||
self.etc = self.root / "etc/systemd"
|
||||
(self.etc / "zram-generator.conf.d").mkdir(parents=True)
|
||||
self.owned = self.etc / "zram-generator.conf.d/90-proxmenux.conf"
|
||||
|
||||
def foreign_configs(self):
|
||||
return call_helper(
|
||||
"zram_foreign_generator_config", self.owned, self.root
|
||||
).stdout.splitlines()
|
||||
|
||||
def test_empty_packaged_default_does_not_block_profile(self):
|
||||
(self.etc / "zram-generator.conf").write_text(
|
||||
"# package default\n[zram0]\n; no active directives\n",
|
||||
encoding="utf-8",
|
||||
)
|
||||
|
||||
self.assertEqual(self.foreign_configs(), [])
|
||||
|
||||
def test_active_foreign_directive_is_reported(self):
|
||||
foreign = self.etc / "zram-generator.conf"
|
||||
foreign.write_text("[zram1]\nzram-size = ram / 4\n", encoding="utf-8")
|
||||
|
||||
self.assertEqual(self.foreign_configs(), [str(foreign)])
|
||||
|
||||
def test_owned_drop_in_is_not_reported(self):
|
||||
self.owned.write_text(
|
||||
"# ProxMenux ZRAM profile\n[zram0]\nzram-size = ram / 10\n",
|
||||
encoding="utf-8",
|
||||
)
|
||||
|
||||
self.assertEqual(self.foreign_configs(), [])
|
||||
|
||||
|
||||
class SystemdSwapDetectionTests(unittest.TestCase):
|
||||
def setUp(self):
|
||||
self.temporary = tempfile.TemporaryDirectory()
|
||||
self.addCleanup(self.temporary.cleanup)
|
||||
self.root = Path(self.temporary.name)
|
||||
self.system_dir = self.root / "etc/systemd/system"
|
||||
self.system_dir.mkdir(parents=True)
|
||||
|
||||
def test_administrator_defined_zram_and_other_swap_units_are_reported(self):
|
||||
(self.system_dir / "dev-zram0.swap").write_text("", encoding="utf-8")
|
||||
(self.system_dir / "custom.swap").write_text("", encoding="utf-8")
|
||||
|
||||
result = call_helper("zram_list_defined_systemd_swap_units", self.root)
|
||||
|
||||
self.assertEqual(result.returncode, 0, result.stderr)
|
||||
self.assertEqual(
|
||||
set(result.stdout.splitlines()),
|
||||
{
|
||||
f"systemd: {self.system_dir / 'dev-zram0.swap'}",
|
||||
f"systemd: {self.system_dir / 'custom.swap'}",
|
||||
},
|
||||
)
|
||||
|
||||
def test_no_non_zram_systemd_swap_has_empty_output(self):
|
||||
result = call_helper("zram_list_defined_systemd_swap_units", self.root)
|
||||
|
||||
self.assertEqual(result.returncode, 0, result.stderr)
|
||||
self.assertEqual(result.stdout, "")
|
||||
|
||||
|
||||
if __name__ == "__main__":
|
||||
unittest.main()
|
||||
Reference in New Issue
Block a user