qzl
26 lines
704 B
Python
26 lines
704 B
Python
from __future__ import annotations
|
|
|
|
import pytest
|
|
|
|
from v1.agent.crewai_flow import AgentFlow
|
|
|
|
|
|
class TestCrewAIFlow:
|
|
@pytest.mark.asyncio
|
|
async def test_flow_stages_run_in_order(self):
|
|
flow = AgentFlow()
|
|
await flow.run()
|
|
assert flow.state.stage_trace == ["intent", "execution", "reporting"]
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_flow_state_initialized(self):
|
|
flow = AgentFlow()
|
|
assert flow.state.stage_trace == []
|
|
assert flow.state.current_stage is None
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_flow_updates_current_stage(self):
|
|
flow = AgentFlow()
|
|
await flow.run()
|
|
assert flow.state.current_stage == "reporting"
|