deploy standalone ASR pipeline on tools-pve

2026-09-18 02:32:14 -07:00
parent f30f72b295
commit b23f00f41a
38 changed files with 182773 additions and 0 deletions
@@ -0,0 +1,52 @@
+from __future__ import annotations
+
+import importlib
+import json
+import os
+import tempfile
+import unittest
+from pathlib import Path
+
+
+class ServiceTests(unittest.TestCase):
+    @classmethod
+    def setUpClass(cls):
+        cls.tmp = tempfile.TemporaryDirectory()
+        os.environ["ASR_DATA_DIR"] = str(Path(cls.tmp.name) / "data")
+        os.environ["ASR_STATE_DIR"] = str(Path(cls.tmp.name) / "state")
+        from asrservice import config, db, worker
+        cls.config, cls.db, cls.worker = config, db, worker
+
+    @classmethod
+    def tearDownClass(cls):
+        cls.tmp.cleanup()
+
+    def test_job_events_follow_updates(self):
+        row = self.db.create_job("a" * 24, "x.m4a", "x.m4a", source="ssh")
+        self.assertEqual(row["status"], "queued")
+        self.db.update_job(row["id"], "transcribing", "started", attempts=1)
+        current = self.db.get_job(row["id"])
+        self.assertEqual(current["attempts"], 1)
+        self.assertEqual([e["status"] for e in self.db.events(row["id"])],
+                         ["queued", "transcribing"])
+
+    def test_transcript_contract(self):
+        text = self.worker.transcript_from_response({
+            "segments": [
+                {"speaker": "speaker_0", "text": "你好。"},
+                {"speaker": "speaker_0", "text": "继续。"},
+                {"speaker": "speaker_1", "text": "收到。"},
+            ],
+            "voiceprint": {"session_id": "s1", "matches": []},
+        })
+        self.assertIn("voiceprint_session: s1", text)
+        self.assertIn("【speaker_0】你好。继续。", text)
+        self.assertEqual(text.count("【speaker_0】"), 1)
+
+    def test_transcript_rejects_empty_segments(self):
+        with self.assertRaises(RuntimeError):
+            self.worker.transcript_from_response({"segments": []})
+
+
+if __name__ == "__main__":
+    unittest.main()