fff

sentence_api/tests/test_api.py

@@ -1165,3 +1165,91 @@ def test_admin_gap_repair_requires_transcriber(tmp_path):
 
     response = client.post(f"/api/v1/admin/videos/{VIDEO_HASH}/repair-gaps")
     assert response.status_code == 503
+
+
+def test_admin_can_repair_missing_and_english_translations(tmp_path, monkeypatch):
+    client = make_client(tmp_path)
+    repository = client.app.state.video_repository
+    _ready_video(repository)
+    repository.save_sentence_translations(
+        VIDEO_HASH,
+        {0: "你好。", 1: "Nice to meet you."},
+    )
+
+    requested = []
+
+    def translate_sentences(sentences, **kwargs):
+        requested.append([index for index, _text in sentences])
+        existing = dict(kwargs.get("existing") or {})
+        result = {}
+        for index, _text in sentences:
+            current = existing.get(index, "")
+            if current and any("\u4e00" <= char <= "\u9fff" for char in current):
+                result[index] = current
+            else:
+                result[index] = f"补译{index}"
+        return result
+
+    class StubAITutor:
+        available = True
+
+    stub = StubAITutor()
+    stub.translate_sentences = translate_sentences
+    monkeypatch.setattr(client.app.state, "ai_tutor", stub)
+    client.app.state.subtitle_translations.ai_tutor = stub
+    client.app.state.subtitle_translations.post_wait_seconds = 0.2
+
+    snapshot = client.get(f"/api/v1/admin/videos/{VIDEO_HASH}/subtitle-translation")
+    assert snapshot.status_code == 200
+    assert snapshot.json()["missing_count"] == 1
+
+    response = client.post(
+        f"/api/v1/admin/videos/{VIDEO_HASH}/subtitle-translation/repair"
+    )
+    assert response.status_code == 200, response.text
+    payload = response.json()
+    if not payload.get("complete"):
+        finished = wait_for_subtitle_translation(client, VIDEO_HASH)
+        payload = client.get(f"/api/v1/admin/videos/{VIDEO_HASH}/subtitle-translation").json()
+        payload["attempted_count"] = response.json()["attempted_count"]
+        payload["complete"] = finished.json()["complete"]
+
+    assert payload["attempted_count"] == 1
+    assert payload["complete"] is True
+    assert payload["missing_count"] == 0
+    document = repository.get_document(VIDEO_HASH)
+    assert [sentence.zh_text for sentence in document.sentences] == ["你好。", "补译1"]
+    assert requested
+
+
+def test_admin_translation_repair_keeps_complete_chinese(tmp_path, monkeypatch):
+    client = make_client(tmp_path)
+    repository = client.app.state.video_repository
+    _ready_video(repository)
+    repository.save_sentence_translations(
+        VIDEO_HASH,
+        {0: "你好。", 1: "见到你很高兴。"},
+    )
+    called = {"count": 0}
+
+    def translate_sentences(sentences, **kwargs):
+        called["count"] += 1
+        return {index: f"新{index}" for index, _text in sentences}
+
+    class StubAITutor:
+        available = True
+
+    stub = StubAITutor()
+    stub.translate_sentences = translate_sentences
+    monkeypatch.setattr(client.app.state, "ai_tutor", stub)
+    client.app.state.subtitle_translations.ai_tutor = stub
+
+    response = client.post(
+        f"/api/v1/admin/videos/{VIDEO_HASH}/subtitle-translation/repair"
+    )
+    assert response.status_code == 200, response.text
+    assert response.json()["attempted_count"] == 0
+    assert response.json()["missing_count"] == 0
+    assert called["count"] == 0
+    document = repository.get_document(VIDEO_HASH)
+    assert [sentence.zh_text for sentence in document.sentences] == ["你好。", "见到你很高兴。"]