mirror of
https://github.com/Significant-Gravitas/AutoGPT.git
synced 2026-02-06 21:05:13 -05:00
ef7cfbb860
Restructuring the Repo to make it clear the difference between classic autogpt and the autogpt platform: * Move the "classic" projects `autogpt`, `forge`, `frontend`, and `benchmark` into a `classic` folder * Also rename `autogpt` to `original_autogpt` for absolute clarity * Rename `rnd/` to `autogpt_platform/` * `rnd/autogpt_builder` -> `autogpt_platform/frontend` * `rnd/autogpt_server` -> `autogpt_platform/backend` * Adjust any paths accordingly
40 lines
1.3 KiB
Python
40 lines
1.3 KiB
Python
import pytest
|
|
|
|
from backend.data import db, graph
|
|
from backend.executor import ExecutionScheduler
|
|
from backend.usecases.sample import create_test_graph, create_test_user
|
|
from backend.util.service import get_service_client
|
|
from backend.util.settings import Config
|
|
from backend.util.test import SpinTestServer
|
|
|
|
|
|
@pytest.mark.asyncio(scope="session")
|
|
async def test_agent_schedule(server: SpinTestServer):
|
|
await db.connect()
|
|
test_user = await create_test_user()
|
|
test_graph = await graph.create_graph(create_test_graph(), user_id=test_user.id)
|
|
|
|
scheduler = get_service_client(
|
|
ExecutionScheduler, Config().execution_scheduler_port
|
|
)
|
|
|
|
schedules = scheduler.get_execution_schedules(test_graph.id, test_user.id)
|
|
assert len(schedules) == 0
|
|
|
|
schedule_id = scheduler.add_execution_schedule(
|
|
graph_id=test_graph.id,
|
|
user_id=test_user.id,
|
|
graph_version=1,
|
|
cron="0 0 * * *",
|
|
input_data={"input": "data"},
|
|
)
|
|
assert schedule_id
|
|
|
|
schedules = scheduler.get_execution_schedules(test_graph.id, test_user.id)
|
|
assert len(schedules) == 1
|
|
assert schedules[schedule_id] == "0 0 * * *"
|
|
|
|
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
|