wehub-resource-sync
ba5ae13cc1
Build uv cache / build-cache (3.10) (push) Has been cancelled
Build uv cache / build-cache (3.11) (push) Has been cancelled
Build uv cache / build-cache (3.12) (push) Has been cancelled
CodeQL Advanced / Analyze (actions) (push) Has been cancelled
CodeQL Advanced / Analyze (python) (push) Has been cancelled
Check Documentation Broken Links / Check broken links (push) Has been cancelled
Vulnerability Scan / pip-audit (push) Has been cancelled
Build uv cache / build-cache (3.13) (push) Has been cancelled
61 lines
1.9 KiB
Python
61 lines
1.9 KiB
Python
"""Tests for ``crewai replay`` and the trained-agents file plumbing."""
|
|
|
|
import subprocess
|
|
from unittest import mock
|
|
|
|
from click.testing import CliRunner
|
|
import pytest
|
|
|
|
from crewai_cli import replay_from_task
|
|
from crewai_cli.cli import replay
|
|
|
|
|
|
@pytest.fixture
|
|
def runner() -> CliRunner:
|
|
return CliRunner()
|
|
|
|
|
|
@mock.patch("crewai_cli.cli.replay_task_command")
|
|
def test_replay_passes_filename(replay_task_command_mock: mock.Mock, runner: CliRunner) -> None:
|
|
result = runner.invoke(replay, ["-t", "abc123", "-f", "my_custom.pkl"])
|
|
|
|
replay_task_command_mock.assert_called_once_with(
|
|
"abc123", trained_agents_file="my_custom.pkl"
|
|
)
|
|
assert result.exit_code == 0
|
|
|
|
|
|
@mock.patch("crewai_cli.cli.replay_task_command")
|
|
def test_replay_without_filename_passes_none(
|
|
replay_task_command_mock: mock.Mock, runner: CliRunner
|
|
) -> None:
|
|
result = runner.invoke(replay, ["-t", "abc123"])
|
|
|
|
replay_task_command_mock.assert_called_once_with(
|
|
"abc123", trained_agents_file=None
|
|
)
|
|
assert result.exit_code == 0
|
|
|
|
|
|
@mock.patch("crewai_cli.replay_from_task.subprocess.run")
|
|
def test_replay_task_command_sets_env_var(mock_subprocess_run: mock.Mock) -> None:
|
|
mock_subprocess_run.return_value = subprocess.CompletedProcess(
|
|
args=["uv", "run", "replay", "abc123"], returncode=0
|
|
)
|
|
replay_from_task.replay_task_command("abc123", trained_agents_file="my_custom.pkl")
|
|
|
|
_, kwargs = mock_subprocess_run.call_args
|
|
assert kwargs["env"]["CREWAI_TRAINED_AGENTS_FILE"] == "my_custom.pkl"
|
|
|
|
|
|
@mock.patch("crewai_cli.replay_from_task.subprocess.run")
|
|
def test_replay_task_command_omits_env_var_without_filename(
|
|
mock_subprocess_run: mock.Mock,
|
|
) -> None:
|
|
mock_subprocess_run.return_value = subprocess.CompletedProcess(
|
|
args=["uv", "run", "replay", "abc123"], returncode=0
|
|
)
|
|
replay_from_task.replay_task_command("abc123")
|
|
|
|
_, kwargs = mock_subprocess_run.call_args
|
|
assert "CREWAI_TRAINED_AGENTS_FILE" not in kwargs["env"] |