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

test(server): Add type to server fixture usages

Browse Source
This commit is contained in:
Reinier van der Leer
2024-08-29 15:42:15 +02:00
parent 5da58aa284
commit 087d3a3760
3 changed files with 8 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
rnd/autogpt_server/test/data/graph.py
View File

@@ -5,10 +5,11 @@ import pytest
from autogpt_server.blocks.basic import InputBlock, ValueBlock
from autogpt_server.data.graph import Graph, Link, Node
from autogpt_server.server.model import CreateGraph
from autogpt_server.util.test import SpinTestServer
@pytest.mark.asyncio(scope="session")
async def test_graph_creation(server):
async def test_graph_creation(server: SpinTestServer):
value_block = ValueBlock().id
input_block = InputBlock().id

8
rnd/autogpt_server/test/executor/test_manager.py
View File

@@ -7,7 +7,7 @@ from autogpt_server.data import execution, graph
from autogpt_server.executor import ExecutionManager
from autogpt_server.server import AgentServer
from autogpt_server.usecases.sample import create_test_graph, create_test_user
from autogpt_server.util.test import wait_execution
from autogpt_server.util.test import SpinTestServer, wait_execution
async def execute_graph(
@@ -85,7 +85,7 @@ async def assert_sample_graph_executions(
@pytest.mark.asyncio(scope="session")
async def test_agent_execution(server):
async def test_agent_execution(server: SpinTestServer):
test_graph = create_test_graph()
test_user = await create_test_user()
await graph.create_graph(test_graph, user_id=test_user.id)
@@ -104,7 +104,7 @@ async def test_agent_execution(server):
@pytest.mark.asyncio(scope="session")
async def test_input_pin_always_waited(server):
async def test_input_pin_always_waited(server: SpinTestServer):
"""
This test is asserting that the input pin should always be waited for the execution,
even when default value on that pin is defined, the value has to be ignored.
@@ -167,7 +167,7 @@ async def test_input_pin_always_waited(server):
@pytest.mark.asyncio(scope="session")
async def test_static_input_link_on_graph(server):
async def test_static_input_link_on_graph(server: SpinTestServer):
"""
This test is asserting the behaviour of static input link, e.g: reusable input link.

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

@@ -4,10 +4,11 @@ from autogpt_server.data import db, graph
from autogpt_server.executor import ExecutionScheduler
from autogpt_server.usecases.sample import create_test_graph, create_test_user
from autogpt_server.util.service import get_service_client
from autogpt_server.util.test import SpinTestServer
@pytest.mark.asyncio(scope="session")
async def test_agent_schedule(server):
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)