From 009158ee4d5ccaf27130bb88cfc3688074299f3f Mon Sep 17 00:00:00 2001 From: Oliver Neumann Date: Mon, 2 Jun 2025 09:00:00 +0000 Subject: [PATCH] Add unit tests --- tests/test_slo_tracker.py | 20 ++++++++++++++++++++ 1 file changed, 20 insertions(+) create mode 100644 tests/test_slo_tracker.py diff --git a/tests/test_slo_tracker.py b/tests/test_slo_tracker.py new file mode 100644 index 0000000..bee16bf --- /dev/null +++ b/tests/test_slo_tracker.py @@ -0,0 +1,20 @@ +from unittest.mock import MagicMock, patch +from slo_tracker import evaluate_slo + + +def make_prom(value): + prom = MagicMock() + prom.custom_query.return_value = [{"value": [0, str(value)]}] + return prom + + +def test_slo_ok(): + slo = {"name": "test", "query": "up", "target": 0.99} + result = evaluate_slo(make_prom(0.999), slo, "7d") + assert result["ok"] is True + + +def test_slo_burning(): + slo = {"name": "test", "query": "up", "target": 0.99} + result = evaluate_slo(make_prom(0.985), slo, "7d") + assert result["ok"] is False