Refactor EPG into shared engine with configurable TVJ channels

tests/test_engine.py

@@ -0,0 +1,65 @@
+import tempfile
+import unittest
+from pathlib import Path
+from unittest.mock import patch
+import xml.etree.ElementTree as ET
+
+from epg.config import BUILTINS, load
+from epg.engine import run
+from epg.providers import univtec
+
+
+def feed():
+    return [{"id": c.channel_id, "epgObject": {"events": [
+        {"title": "Match & highlights", "start": "2026-09-10T12:00:00Z",
+         "end": "2026-09-10T13:00:00Z", "custom": {
+             "longDescription": "<p>Live &amp; replay</p>", "genre": ["Sports"],
+             "seasonNumber": 2, "episodeNumber": 3, "rating": "G"}},
+        {"title": "No Program", "start": "2026-09-10T13:00:00Z", "end": "2026-09-10T14:00:00Z"},
+        {"title": "Broken", "start": 123, "end": None},
+    ]}} for c in BUILTINS.values()]
+
+
+class EngineTests(unittest.TestCase):
+    def test_selection_validation(self):
+        self.assertEqual([c.id for c in load({})[0]], [BUILTINS['tvj'].id])
+        self.assertEqual(len(load({'CHANNELS': 'tvjsn,tvjsn'})[0]), 1)
+        for selection in ('', 'unknown'):
+            with self.assertRaises(ValueError):
+                load({'CHANNELS': selection})
+
+    def test_outputs_and_shared_fetch(self):
+        for selection in ('tvj', 'tvjsn', 'tvj,tvjsn'):
+            with self.subTest(selection=selection), tempfile.TemporaryDirectory() as directory:
+                output = Path(directory) / 'tvj.xml'
+                with patch.object(univtec, 'fetch', return_value=feed()) as fetch:
+                    run({'CHANNELS': selection, 'OUTPUT_FILE': str(output)})
+                fetch.assert_called_once()
+                root = ET.parse(output).getroot()
+                ids = [BUILTINS[k].id for k in selection.split(',')]
+                self.assertEqual([c.get('id') for c in root.findall('channel')], ids)
+                self.assertEqual([p.get('channel') for p in root.findall('programme')], ids)
+                playlist = output.with_suffix('.m3u').read_text()
+                for key in selection.split(','):
+                    self.assertIn(f'tvg-id="{BUILTINS[key].id}"', playlist)
+                    self.assertIn(BUILTINS[key].stream_url, playlist)
+                self.assertEqual(root.findtext('programme/desc'), 'Live & replay')
+                self.assertEqual(root.findtext('programme/episode-num'), '1.2.')
+                self.assertEqual(root.findtext('programme/category'), 'Sports')
+
+    def test_failed_update_preserves_files(self):
+        for response in ([], feed()[:1]):
+            with tempfile.TemporaryDirectory() as directory:
+                xml = Path(directory) / 'tvj.xml'
+                playlist = Path(directory) / 'tvj.m3u'
+                xml.write_text('previous guide')
+                playlist.write_text('previous playlist')
+                with patch.object(univtec, 'fetch', return_value=response):
+                    with self.assertRaises(ValueError):
+                        run({'CHANNELS': 'tvj,tvjsn', 'OUTPUT_FILE': str(xml)})
+                self.assertEqual(xml.read_text(), 'previous guide')
+                self.assertEqual(playlist.read_text(), 'previous playlist')
+
+
+if __name__ == '__main__':
+    unittest.main()