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

fix(backend): use fixed dates in credit_test to avoid month boundary flakiness

Browse Source 
The test was failing because datetime.now().replace(month=1) in February
still produced a January 2026 date, which matched the "35 days ago"
updatedAt also landing in January 2026 - causing the refill to be skipped.

Co-Authored-By: Claude Opus 4.5 <noreply@anthropic.com>
This commit is contained in:
Nicholas Tindle
2026-02-04 21:43:54 -06:00
parent fe264545a9
commit ecdf2f8c09
1 changed files with 10 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

18
autogpt_platform/backend/backend/data/credit_test.py
View File

@@ -124,16 +124,18 @@ async def test_block_credit_top_up(server: SpinTestServer):
@pytest.mark.asyncio(loop_scope="session")
async def test_block_credit_reset(server: SpinTestServer):
"""Test that BetaUserCredit provides monthly refills correctly."""
await disable_test_user_transactions()
# Save original time_now function for restoration
original_time_now = user_credit.time_now
try:
# Test month 1 behavior
month1 = datetime.now(timezone.utc).replace(month=1, day=1)
# Test month 1 behavior - use a fixed year to avoid date boundary issues
month1 = datetime(2025, 1, 15, 12, 0, 0, tzinfo=timezone.utc)
user_credit.time_now = lambda: month1
# Reset transactions AFTER setting time_now so the old date calculation
# uses a date that's definitely in a previous month relative to month1
await disable_test_user_transactions()
# First call in month 1 should trigger refill
balance = await user_credit.get_credits(DEFAULT_USER_ID)
assert balance == REFILL_VALUE # Should get 1000 credits
@@ -159,8 +161,8 @@ async def test_block_credit_reset(server: SpinTestServer):
},
)
# Now test month 2 behavior
month2 = datetime.now(timezone.utc).replace(month=2, day=1)
# Now test month 2 behavior - use fixed date
month2 = datetime(2025, 2, 15, 12, 0, 0, tzinfo=timezone.utc)
user_credit.time_now = lambda: month2
# In month 2, since balance (1100) > refill (1000), no refill should happen
@@ -185,8 +187,8 @@ async def test_block_credit_reset(server: SpinTestServer):
}
)
# Move to month 3
month3 = datetime.now(timezone.utc).replace(month=3, day=1)
# Move to month 3 - use fixed date
month3 = datetime(2025, 3, 15, 12, 0, 0, tzinfo=timezone.utc)
user_credit.time_now = lambda: month3
# Should get refilled since balance (400) < refill value (1000)