更新

backend/tests/test_worker_scale_controls.py

@@ -109,7 +109,7 @@ class WorkerScaleControlTests(unittest.IsolatedAsyncioTestCase):
         )
         worker._load_user_agent = AsyncMock(return_value="test-agent")
         worker._build_im_session_from_storage = AsyncMock(return_value=session)
-        worker._require_sec_user_id = AsyncMock(return_value="sec-user")
+        worker._best_effort_sec_user_id = AsyncMock(return_value="sec-user")
         worker._persist_im_session = AsyncMock()
         worker._run_im_direct_service = AsyncMock()
 
@@ -124,7 +124,10 @@ class WorkerScaleControlTests(unittest.IsolatedAsyncioTestCase):
         self.assertTrue(started)
         self.assertEqual(reason, "")
         validate.assert_not_awaited()
-        worker._require_sec_user_id.assert_awaited_once()
+        worker._best_effort_sec_user_id.assert_awaited_once_with(
+            refresh_if_missing=True,
+            refresh_if_stale=True,
+        )
         worker._run_im_direct_service.assert_awaited_once_with(session)
 
     async def test_disabled_follow_welcome_uses_cached_lightweight_config(self):
@@ -171,20 +174,18 @@ class WorkerScaleControlTests(unittest.IsolatedAsyncioTestCase):
         worker.get_db.assert_not_awaited()
         worker._require_sec_user_id.assert_not_awaited()
 
-    async def test_missing_cached_sec_user_id_stops_hosting(self):
+    async def test_disabled_follow_welcome_ignores_missing_cached_sec_user_id(self):
         worker = DouyinWorker(account_id=505)
         worker._im_service = SimpleNamespace(session=object())
         worker._follow_config_loaded = True
         worker._refresh_follow_welcome_config = AsyncMock(
             return_value=(False, "", "")
         )
-        worker._require_sec_user_id = AsyncMock(return_value="")
+        worker._best_effort_sec_user_id = AsyncMock(return_value="")
 
         await worker.follow_welcome_tick()
 
-        worker._require_sec_user_id.assert_awaited_once_with(
-            "托管运行中"
-        )
+        worker._best_effort_sec_user_id.assert_not_awaited()
 
 
 if __name__ == "__main__":