fix sanitize_severity() maping "ok" -> "Ok" (not all caps)

gnocpush/utils.py

@@ -15,7 +15,7 @@ def sanitize_severity(severity):
         case 'major' | 'warning': s = 'Major'
         case 'minor' | 'info': s = 'Minor'
         case 'unknown': s = 'Unknown'
-        case 'ok': s = 'Ok'
+        case 'ok': s = 'OK'
 
     log.debug(f'severity: {severity} -> {s}')
 

pyproject.toml

@@ -6,6 +6,7 @@ dependencies = [
   "flask",
   "globalnoc_alertmon_agent",
   "prometheus-flask-exporter==0.23.0",
+  "pytest",
   "python-dateutil",
   "requests",
   "waitress",
@@ -24,3 +25,10 @@ ignore = ['E221', 'E251']
 
 [tool.setuptools.packages.find]
 exclude = ["charts"]
+
+[tool.pytest.ini_options]
+minversion = "6.0"
+addopts = "-q"
+testpaths = [
+  "tests",
+]

tests/test_severity.py

@@ -0,0 +1,33 @@
+#!/usr/bin/env python3
+
+from gnocpush.utils import sanitize_severity
+
+
+def test_critical():
+    words = ["critical", "CRITICAL", "cRiTiCaL", "Critical"]
+    for word in words:
+        assert sanitize_severity(word) == 'Critical'
+
+
+def test_major():
+    words = ["major", "MAJOR", "mAjOr", "Major"]
+    for word in words:
+        assert sanitize_severity(word) == 'Major'
+
+
+def test_minor():
+    words = ["minor", "MINOR", "mInOr", "Minor"]
+    for word in words:
+        assert sanitize_severity(word) == 'Minor'
+
+
+def test_unknown():
+    words = ["unknown", "UNKNOWN", "uNkNoWn", "Unknown", "foo", "bar", "baz"]
+    for word in words:
+        assert sanitize_severity(word) == 'Unknown'
+
+
+def test_ok():
+    words = ["ok", "OK", "oK", "Ok"]
+    for word in words:
+        assert sanitize_severity(word) == 'OK'