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(tests): Ensure user ID is not None on schedule in tests (#7749)

Browse Source 
* ensure user id is never None on schedule

* remove comment
This commit is contained in:
Aarushi
2024-08-08 10:11:57 +01:00
committed by GitHub
parent 78e96f8a1a
commit 9c74d76a3a
2 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
rnd/autogpt_server/autogpt_server/data/schedule.py
View File

@@ -64,6 +64,7 @@ async def add_schedule(schedule: ExecutionSchedule) -> ExecutionSchedule:
obj = await AgentGraphExecutionSchedule.prisma().create(
data={
"id": schedule.id,
"userId": schedule.user_id,
"agentGraphId": schedule.graph_id,
"agentGraphVersion": schedule.graph_version,
"schedule": schedule.schedule,

5
rnd/autogpt_server/test/executor/test_scheduler.py
View File

@@ -6,7 +6,6 @@ from autogpt_server.usecases.sample import create_test_graph, create_test_user
from autogpt_server.util.service import get_service_client
@pytest.mark.skip(reason="flakey test, needs to be investigated")
@pytest.mark.asyncio(scope="session")
async def test_agent_schedule(server):
await db.connect()
@@ -31,6 +30,6 @@ async def test_agent_schedule(server):
assert len(schedules) == 1
assert schedules[schedule_id] == "0 0 * * *"
scheduler.update_schedule(schedule_id, is_enabled=False)
schedules = scheduler.get_execution_schedules(test_graph.id)
scheduler.update_schedule(schedule_id, is_enabled=False, user_id=test_user.id)
schedules = scheduler.get_execution_schedules(test_graph.id, user_id=test_user.id)
assert len(schedules) == 0