fz-mirrors/AstrBot
1
0
Fork 0
mirror of https://github.com/AstrBotDevs/AstrBot synced 2026-07-15 17:30:13 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix(test): use MagicMock instead of AsyncMock for sync ABP methods

Browse Source
This commit is contained in:
LIghtJUNction
2026-03-23 21:39:52 +08:00
parent f62fc4d8a0
commit 0b25f4eba1
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
tests/unit/test_internal_runtime.py
View File

@@ -89,7 +89,7 @@ class TestAstrbotOrchestrator:
mock_star = MagicMock()
mock_star.call_tool = AsyncMock()
with patch.object(orchestrator.abp, 'register_star', new_callable=AsyncMock) as mock_register:
with patch.object(orchestrator.abp, 'register_star', new_callable=MagicMock) as mock_register:
await orchestrator.register_star("test_star", mock_star)
mock_register.assert_called_once_with("test_star", mock_star)
@@ -109,7 +109,7 @@ class TestAstrbotOrchestrator:
mock_star = MagicMock()
orchestrator._stars["test_star"] = mock_star
with patch.object(orchestrator.abp, 'unregister_star', new_callable=AsyncMock) as mock_unregister:
with patch.object(orchestrator.abp, 'unregister_star', new_callable=MagicMock) as mock_unregister:
await orchestrator.unregister_star("test_star")
mock_unregister.assert_called_once_with("test_star")