完成视频复刻和优化

core/backend/apps/assets/tests.py

@@ -890,3 +890,40 @@ class ReviewScopeTests(TestCase):
         with patch("apps.assets.review.assets_client.is_enabled", return_value=False):
             out = review.poll_team_reviews(team)
         self.assertEqual(set(out.keys()), set(keep.values()))
+
+    def test_poll_team_reviews_auto_submits_unsubmitted(self):
+        from apps.assets import review
+
+        _, team = _mk_team("usub", "TeamUSub")
+        empty = Asset.objects.create(
+            team=team, name="待审角色", asset_type="image", source="ai_generated",
+            category=Asset.Category.PERSON, review_status="",
+        )
+        with patch("apps.assets.review.submit_asset_for_review", return_value=True) as sub, patch(
+            "apps.assets.review.ensure_review_drain_loop"
+        ):
+            review.poll_team_reviews(team)
+        sub.assert_called()
+        self.assertEqual(sub.call_args.args[0].id, empty.id)
+
+    def test_drain_platform_reviews_submits_then_polls(self):
+        from apps.assets import review
+
+        _, team = _mk_team("drain", "TeamDrain")
+        Asset.objects.create(
+            team=team, name="未送", asset_type="image", source="ai_generated",
+            category=Asset.Category.PERSON, review_status="",
+        )
+        proc = Asset.objects.create(
+            team=team, name="审中", asset_type="image", source="ai_generated",
+            category=Asset.Category.PERSON, review_status="processing",
+        )
+        with patch("apps.assets.review.submit_asset_for_review", return_value=True) as sub, patch(
+            "apps.assets.review.poll_asset_review", return_value="active"
+        ) as polled:
+            out = review.drain_platform_reviews()
+        self.assertEqual(out["submitted"], 1)
+        self.assertEqual(out["polled"], 1)
+        self.assertEqual(out["statuses"][str(proc.id)], "active")
+        sub.assert_called_once()
+        polled.assert_called_once()