35 lines
1.2 KiB
Python
35 lines
1.2 KiB
Python
import pytest
|
|
|
|
from backend.util.test import SpinTestServer
|
|
|
|
|
|
@pytest.fixture(scope="session")
|
|
async def server():
|
|
async with SpinTestServer() as server:
|
|
yield server
|
|
|
|
|
|
@pytest.fixture(scope="session", autouse=True)
|
|
async def graph_cleanup(server):
|
|
created_graph_ids = []
|
|
original_create_graph = server.agent_server.create_graph
|
|
|
|
async def create_graph_wrapper(*args, **kwargs):
|
|
created_graph = await original_create_graph(*args, **kwargs)
|
|
# Extract user_id correctly
|
|
user_id = kwargs.get("user_id", args[2] if len(args) > 2 else None)
|
|
created_graph_ids.append((created_graph.id, user_id))
|
|
return created_graph
|
|
|
|
try:
|
|
server.agent_server.create_graph = create_graph_wrapper
|
|
yield # This runs the test function
|
|
finally:
|
|
server.agent_server.create_graph = original_create_graph
|
|
|
|
# Delete the created graphs and assert they were deleted
|
|
for graph_id, user_id in created_graph_ids:
|
|
resp = await server.agent_server.delete_graph(graph_id, user_id)
|
|
num_deleted = resp["version_counts"]
|
|
assert num_deleted > 0, f"Graph {graph_id} was not deleted."
|