GithubBackup/MoviePilot
1
0
Fork 0
mirror of https://github.com/jxxghp/MoviePilot.git synced 2026-09-07 00:16:57 +08:00
Code Issues Packages Projects Releases Wiki Activity

test: inject agent protocol config snapshots

Browse Source
This commit is contained in:
jxxghp
2026-08-22 13:30:20 +08:00
parent 4fbe19a49c
commit 4e9eea13fd
1 changed files with 26 additions and 13 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tests/test_agent_protocol_lifecycle.py
+26 -13
View File
@@ -8,7 +8,6 @@ from fastapi.security import HTTPAuthorizationCredentials
from app import schemas from app import schemas
from app.api.endpoints import anthropic, openai from app.api.endpoints import anthropic, openai
from app.runtime.config import settings
_API_TOKEN = "test-agent-protocol-token" _API_TOKEN = "test-agent-protocol-token"
@@ -73,10 +72,18 @@ def test_streaming_protocols_reject_config_disable_before_manager_lookup() -> No
), ),
) )
with patch.object(settings, "AI_AGENT_ENABLE", False), patch.object( runtime_config = SimpleNamespace(
settings, ai_agent_enable=False,
"API_TOKEN", api_token=_API_TOKEN,
_API_TOKEN, )
with patch.object(
openai,
"get_api_runtime_config_snapshot",
return_value=runtime_config,
), patch.object(
anthropic,
"get_api_runtime_config_snapshot",
return_value=runtime_config,
), patch.object( ), patch.object(
openai, openai,
"get_running_agent_manager", "get_running_agent_manager",
@@ -111,10 +118,13 @@ def test_openai_stream_rejects_shutdown_race_and_cleans_request_session() -> Non
) )
return await _collect(response) return await _collect(response)
with patch.object(settings, "AI_AGENT_ENABLE", True), patch.object( with patch.object(
settings, openai,
"API_TOKEN", "get_api_runtime_config_snapshot",
_API_TOKEN, return_value=SimpleNamespace(
ai_agent_enable=True,
api_token=_API_TOKEN,
),
), patch.object( ), patch.object(
openai, openai,
"get_running_agent_manager", "get_running_agent_manager",
@@ -145,10 +155,13 @@ def test_anthropic_stream_rejects_shutdown_race_and_cleans_request_session() ->
) )
return await _collect(response) return await _collect(response)
with patch.object(settings, "AI_AGENT_ENABLE", True), patch.object( with patch.object(
settings, anthropic,
"API_TOKEN", "get_api_runtime_config_snapshot",
_API_TOKEN, return_value=SimpleNamespace(
ai_agent_enable=True,
api_token=_API_TOKEN,
),
), patch.object( ), patch.object(
anthropic, anthropic,
"get_running_agent_manager", "get_running_agent_manager",