platform/reworkd_platform/tests/test_dependancies.py

import pytest

from reworkd_platform.web.api.agent import dependancies


@pytest.mark.anyio
@pytest.mark.parametrize(
    "validator, step",
    [
        (dependancies.agent_summarize_validator, "summarize"),
        (dependancies.agent_chat_validator, "chat"),
        (dependancies.agent_analyze_validator, "analyze"),
        (dependancies.agent_create_validator, "create"),
        (dependancies.agent_execute_validator, "execute"),
    ],
)
async def test_agent_validate(mocker, validator, step):
    run_id = "asim"
    crud = mocker.Mock()
    body = mocker.Mock()
    body.run_id = run_id

    crud.create_task = mocker.AsyncMock()

    await validator(body, crud)
    crud.create_task.assert_called_once_with(run_id, step)