mirror of
https://github.com/Significant-Gravitas/AutoGPT.git
synced 2026-04-08 03:00:28 -04:00
Adding tests
This commit is contained in:
16
rnd/autogpt_server/autogpt_server/server/realtime_test.py
Normal file
16
rnd/autogpt_server/autogpt_server/server/realtime_test.py
Normal file
@@ -0,0 +1,16 @@
|
||||
import pytest
|
||||
import time
|
||||
|
||||
from autogpt_server.server.realtime import start_realtime, start_realtime_connector
|
||||
from autogpt_server.data.execution import ExecutionQueue, Execution
|
||||
|
||||
@pytest.mark.asyncio
|
||||
async def test_realtime():
|
||||
api_queue = ExecutionQueue()
|
||||
exec_queue = ExecutionQueue()
|
||||
|
||||
start_realtime(api_queue, exec_queue)
|
||||
api_queue.add(Execution(run_id="1", node_id="2", data={}))
|
||||
time.sleep(0.1)
|
||||
msg: Execution = api_queue.get()
|
||||
assert msg.node_id == "hi"
|
||||
Reference in New Issue
Block a user