github/AutoGPT
1
1
Fork 0
mirror of https://github.com/Significant-Gravitas/AutoGPT.git synced 2026-04-08 03:00:28 -04:00
Code Issues Packages Projects Releases Wiki Activity

fix(backend/copilot): mock get_user_credit_model instead of removed credit_db in tests

Browse Source
This commit is contained in:
Zamil Majdy
2026-03-15 13:47:41 +07:00
parent 6cc2c7a50d
commit abd9dac288
1 changed files with 6 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
autogpt_platform/backend/backend/copilot/tools/helpers_test.py
View File

@@ -46,7 +46,7 @@ def _patch_credit_db(
get_credits_return: int = 100,
spend_credits_side_effect: Any = None,
):
"""Patch credit_db to return a mock credit accessor."""
"""Patch get_user_credit_model to return a mock credit model."""
mock_credit = MagicMock()
mock_credit.get_credits = AsyncMock(return_value=get_credits_return)
if spend_credits_side_effect is not None:
@@ -54,7 +54,11 @@ def _patch_credit_db(
else:
mock_credit.spend_credits = AsyncMock()
return (
patch("backend.copilot.tools.helpers.credit_db", return_value=mock_credit),
patch(
"backend.copilot.tools.helpers.get_user_credit_model",
new_callable=AsyncMock,
return_value=mock_credit,
),
mock_credit,
)