mirror of
https://github.com/trustgraph-ai/trustgraph.git
synced 2026-07-01 17:39:39 +02:00
55 lines
1.6 KiB
Python
55 lines
1.6 KiB
Python
|
"""
|
||
|
|
Unit tests for trustgraph.bootstrap.initialisers.DefaultFlowStart
|
||
|
|
|
||
|
|
Verifies the list/start timeouts are configurable and that the
|
||
|
|
configured values actually reach the flow-client request calls.
|
||
|
|
"""
|
||
|
|
|
||
|
|
from unittest.mock import AsyncMock, MagicMock
|
||
|
|
|
||
|
|
import pytest
|
||
|
|
|
||
|
|
from trustgraph.bootstrap.initialisers.default_flow_start import (
|
||
|
|
DefaultFlowStart,
|
||
|
|
)
|
||
|
|
|
||
|
|
|
||
|
|
def test_default_timeouts():
|
||
|
|
init = DefaultFlowStart(blueprint="bp")
|
||
|
|
assert init.list_timeout == 10
|
||
|
|
assert init.start_timeout == 30
|
||
|
|
|
||
|
|
|
||
|
|
def test_timeout_overrides_are_stored():
|
||
|
|
init = DefaultFlowStart(blueprint="bp", list_timeout=5, start_timeout=99)
|
||
|
|
assert init.list_timeout == 5
|
||
|
|
assert init.start_timeout == 99
|
||
|
|
|
||
|
|
|
||
|
|
@pytest.mark.asyncio
|
||
|
|
async def test_run_forwards_configured_timeouts():
|
||
|
|
init = DefaultFlowStart(blueprint="bp", list_timeout=5, start_timeout=99)
|
||
|
|
|
||
|
|
# Flow client: list-flows returns no error + empty flow list,
|
||
|
|
# start-flow returns no error.
|
||
|
|
flow = MagicMock()
|
||
|
|
flow.start = AsyncMock()
|
||
|
|
flow.stop = AsyncMock()
|
||
|
|
flow.request = AsyncMock(side_effect=[
|
||
|
|
MagicMock(error=None, flow_ids=[]), # list-flows response
|
||
|
|
MagicMock(error=None), # start-flow response
|
||
|
|
])
|
||
|
|
|
||
|
|
# Context: workspace "default" exists, hands back our mock flow client.
|
||
|
|
ctx = MagicMock()
|
||
|
|
ctx.logger = MagicMock()
|
||
|
|
ctx.config.keys = AsyncMock(return_value=["default"])
|
||
|
|
ctx.make_flow_client = MagicMock(return_value=flow)
|
||
|
|
|
||
|
|
await init.run(ctx, None, "v1")
|
||
|
|
|
||
|
|
calls = flow.request.call_args_list
|
||
|
|
assert len(calls) == 2
|
||
|
|
assert calls[0].kwargs["timeout"] == 5
|
||
|
|
assert calls[1].kwargs["timeout"] == 99
|