Add health reporting, container probe, and JNN channel support

2026-09-10 10:38:21 -04:00
parent 02a68f029a
commit 8529fca032
11 changed files with 349 additions and 13 deletions
@@ -0,0 +1,38 @@
+import json
+import tempfile
+import threading
+import unittest
+from functools import partial
+from http.client import HTTPConnection
+from http.server import ThreadingHTTPServer
+from pathlib import Path
+from unittest.mock import patch
+from epg.server import Handler
+
+
+class ServerTests(unittest.TestCase):
+    def test_health_codes_and_static_files(self):
+        with tempfile.TemporaryDirectory() as directory:
+            Path(directory, 'tvj.m3u').write_text('#EXTM3U\n')
+            server = ThreadingHTTPServer(('127.0.0.1', 0), partial(Handler, directory=directory))
+            thread = threading.Thread(target=server.serve_forever, daemon=True)
+            thread.start()
+            try:
+                for status, expected in [('healthy', 200), ('degraded', 200), ('unhealthy', 503)]:
+                    with patch('epg.server.evaluate', return_value={'status': status}):
+                        connection = HTTPConnection(*server.server_address, timeout=5)
+                        connection.request('GET', '/health')
+                        response = connection.getresponse()
+                        self.assertEqual(response.status, expected)
+                        self.assertEqual(json.loads(response.read())['status'], status)
+                        connection.close()
+                connection = HTTPConnection(*server.server_address, timeout=5)
+                connection.request('GET', '/tvj.m3u')
+                response = connection.getresponse()
+                self.assertEqual(response.status, 200)
+                self.assertEqual(response.read(), b'#EXTM3U\n')
+                connection.close()
+            finally:
+                server.shutdown()
+                server.server_close()
+                thread.join()