chore: 迁移服务器前同步本地修改

tests/unit/test_provider_manager.py

@@ -30,8 +30,7 @@ async def test_add_provider_happy_path_writes_register_and_db():
     pm.provider_mapper["openai"].create_provider = AsyncMock(return_value=fake_provider)
 
     postgres = MagicMock()
-    postgres.add_provider_db = MagicMock()
-    postgres.add_provider_db.remote = AsyncMock(return_value=None)
+    postgres.add_provider_db = AsyncMock(return_value=None)
 
     await pm.add_provider(
         provider_type="openai",
@@ -44,8 +43,8 @@ async def test_add_provider_happy_path_writes_register_and_db():
 
     assert "my-openai" in pm.provider_register
     assert pm.provider_register["my-openai"] is fake_provider
-    postgres.add_provider_db.remote.assert_awaited_once()
-    kwargs = postgres.add_provider_db.remote.await_args.kwargs
+    postgres.add_provider_db.assert_awaited_once()
+    kwargs = postgres.add_provider_db.await_args.kwargs
     assert kwargs["provider_title"] == "my-openai"
     assert kwargs["provider_apikey"] == "sk-xxx"
     assert kwargs["provider_models"] == ["gpt-4o"]
@@ -55,8 +54,7 @@ async def test_add_provider_happy_path_writes_register_and_db():
 async def test_add_provider_unknown_type_returns_none(caplog):
     pm = ProviderManager(postgres=None)
     postgres = MagicMock()
-    postgres.add_provider_db = MagicMock()
-    postgres.add_provider_db.remote = AsyncMock()
+    postgres.add_provider_db = AsyncMock()
 
     result = await pm.add_provider(
         provider_type="not_supported",
@@ -69,7 +67,7 @@ async def test_add_provider_unknown_type_returns_none(caplog):
 
     assert result is None
     assert "x" not in pm.provider_register
-    postgres.add_provider_db.remote.assert_not_awaited()
+    postgres.add_provider_db.assert_not_awaited()
 
 
 @pytest.mark.asyncio
@@ -85,8 +83,7 @@ async def test_add_provider_network_error_raises_retryable():
     )
 
     postgres = MagicMock()
-    postgres.add_provider_db = MagicMock()
-    postgres.add_provider_db.remote = AsyncMock()
+    postgres.add_provider_db = AsyncMock()
 
     with pytest.raises(RetryableError):
         await pm.add_provider(