Add some unittests

Author: martialblog

.github/workflows/unittest.yml

@@ -18,3 +18,6 @@ jobs:
     - name: Lint
       run: |
         make lint
+    - name: Test
+      run: |
+        make coverage

Makefile

@@ -1,4 +1,9 @@
-.PHONY: lint
+.PHONY: lint test coverage
 
 lint:
 	python -m pylint check_sensorProbe2plus.py
+test:
+	python -m unittest -v test_check_sensorProbe2plus.py
+coverage:
+	python -m coverage run -m unittest -b test_check_sensorProbe2plus.py
+	python -m coverage report -m --include check_sensorProbe2plus.py

requirements-dev.txt

@@ -1 +1,2 @@
 pylint==3.3.7
+coverage==7.9.0

test_check_sensorProbe2plus.py

@@ -0,0 +1,36 @@
+#!/usr/bin/env python3
+
+import unittest
+import unittest.mock as mock
+import sys
+
+sys.path.append('..')
+
+from check_sensorProbe2plus import convert_state_to_nagios
+from check_sensorProbe2plus import print_status_message
+
+from check_sensorProbe2plus import NagiosState
+
+class UtilTesting(unittest.TestCase):
+
+    def test_state(self):
+        actual = convert_state_to_nagios(1)
+        expected = NagiosState.UNKNOWN
+
+        self.assertEqual(actual, expected)
+
+    @mock.patch('builtins.print')
+    def test_status_message(self, mock_print):
+        sensor_states = {
+            "OK": [],
+            "WARNING": [],
+            "CRITICAL": [],
+            "UNKNOWN": [],
+        }
+
+        print_status_message(sensor_states, "perfdata")
+
+        calls = [mock.call('OK sensorProbe2plus: Sensor reports that everything is fine|p e r f d a t a ')]
+
+        mock_print.assert_has_calls(calls)
+