wehub-resource-sync
a0c8464e58
Build Package / build (ubuntu-latest) (push) Failing after 1s
CodeQL / Analyze (python) (push) Failing after 1s
Core Typecheck / core-typecheck (push) Failing after 1s
Linting / lint (push) Failing after 1s
llama-dev tests / test-llama-dev (push) Failing after 1s
Publish Sub-Package to PyPI if Needed / publish_subpackage_if_needed (push) Has been skipped
Sync Docs to Developer Hub / sync-docs (push) Failing after 0s
Build Package / build (windows-latest) (push) Has been cancelled
122 lines
4.2 KiB
Python
122 lines
4.2 KiB
Python
from unittest import mock
|
|
|
|
from click.testing import CliRunner
|
|
from llama_dev.cli import cli
|
|
|
|
|
|
def test_cmd_exec_no_package_no_all_flag():
|
|
runner = CliRunner()
|
|
result = runner.invoke(cli, ["pkg", "exec", "--cmd", "echo hello"])
|
|
assert result.exit_code != 0
|
|
assert "Either specify a package name or use the --all flag" in result.output
|
|
|
|
|
|
@mock.patch("llama_dev.pkg.cmd_exec.is_llama_index_package")
|
|
def test_cmd_exec_invalid_package(mock_is_llama_index):
|
|
mock_is_llama_index.return_value = False
|
|
|
|
runner = CliRunner()
|
|
result = runner.invoke(cli, ["pkg", "exec", "invalid-pkg", "--cmd", "echo hello"])
|
|
|
|
assert result.exit_code != 0
|
|
print(result.output)
|
|
assert "not a path to a LlamaIndex package" in result.output
|
|
|
|
|
|
@mock.patch("llama_dev.pkg.cmd_exec.find_all_packages")
|
|
@mock.patch("llama_dev.pkg.cmd_exec.subprocess.run")
|
|
def test_cmd_exec_all_flag(mock_subprocess, mock_find_all, data_path):
|
|
mock_find_all.return_value = [data_path / "fake/pkg1", data_path / "fake/pkg2"]
|
|
mock_subprocess.return_value = mock.Mock(
|
|
returncode=0, stdout="Command output", stderr=""
|
|
)
|
|
|
|
runner = CliRunner()
|
|
result = runner.invoke(cli, ["pkg", "exec", "--all", "--cmd", "echo hello"])
|
|
|
|
assert result.exit_code == 0
|
|
assert mock_subprocess.call_count == 2
|
|
assert "Command succeeded" in result.output
|
|
|
|
|
|
@mock.patch("llama_dev.pkg.cmd_exec.is_llama_index_package")
|
|
@mock.patch("llama_dev.pkg.cmd_exec.subprocess.run")
|
|
def test_cmd_exec_single_package_success(mock_subprocess, mock_is_llama_index):
|
|
mock_is_llama_index.return_value = True
|
|
mock_subprocess.return_value = mock.Mock(
|
|
returncode=0, stdout="Command output", stderr=""
|
|
)
|
|
|
|
runner = CliRunner()
|
|
result = runner.invoke(cli, ["pkg", "exec", "valid-pkg", "--cmd", "echo hello"])
|
|
|
|
assert result.exit_code == 0
|
|
assert mock_subprocess.call_count == 1
|
|
assert "Command succeeded" in result.output
|
|
|
|
|
|
@mock.patch("llama_dev.pkg.cmd_exec.is_llama_index_package")
|
|
@mock.patch("llama_dev.pkg.cmd_exec.subprocess.run")
|
|
def test_cmd_exec_failure_without_fail_fast(mock_subprocess, mock_is_llama_index):
|
|
mock_is_llama_index.return_value = True
|
|
mock_subprocess.return_value = mock.Mock(
|
|
returncode=1, stdout="", stderr="Error message"
|
|
)
|
|
|
|
runner = CliRunner()
|
|
result = runner.invoke(cli, ["pkg", "exec", "valid-pkg", "--cmd", "echo hello"])
|
|
|
|
assert result.exit_code == 0 # Command should continue despite failure
|
|
assert "Command 'echo hello' failed" in result.output
|
|
|
|
|
|
@mock.patch("llama_dev.pkg.cmd_exec.is_llama_index_package")
|
|
@mock.patch("llama_dev.pkg.cmd_exec.subprocess.run")
|
|
def test_cmd_exec_failure_with_fail_fast(mock_subprocess, mock_is_llama_index):
|
|
mock_is_llama_index.return_value = True
|
|
mock_subprocess.return_value = mock.Mock(
|
|
returncode=1, stdout="", stderr="Error message"
|
|
)
|
|
|
|
runner = CliRunner()
|
|
result = runner.invoke(
|
|
cli, ["pkg", "exec", "valid-pkg", "--cmd", "echo hello", "--fail-fast"]
|
|
)
|
|
|
|
assert result.exit_code != 0 # Command should fail at the first error
|
|
assert "Command 'echo hello' failed" in result.output
|
|
|
|
|
|
@mock.patch("llama_dev.pkg.cmd_exec.is_llama_index_package")
|
|
@mock.patch("llama_dev.pkg.cmd_exec.subprocess.run")
|
|
def test_cmd_exec_multiple_packages(mock_subprocess, mock_is_llama_index):
|
|
mock_is_llama_index.return_value = True
|
|
mock_subprocess.return_value = mock.Mock(
|
|
returncode=0, stdout="Command output", stderr=""
|
|
)
|
|
|
|
runner = CliRunner()
|
|
result = runner.invoke(cli, ["pkg", "exec", "pkg1", "pkg2", "--cmd", "echo hello"])
|
|
|
|
assert result.exit_code == 0
|
|
assert mock_subprocess.call_count == 2
|
|
assert "Command succeeded" in result.output
|
|
|
|
|
|
@mock.patch("llama_dev.pkg.cmd_exec.is_llama_index_package")
|
|
@mock.patch("llama_dev.pkg.cmd_exec.subprocess.run")
|
|
def test_cmd_exec_silent(mock_subprocess, mock_is_llama_index):
|
|
mock_is_llama_index.return_value = True
|
|
mock_subprocess.return_value = mock.Mock(
|
|
returncode=0, stdout="Command output", stderr=""
|
|
)
|
|
|
|
runner = CliRunner()
|
|
result = runner.invoke(
|
|
cli, ["pkg", "exec", "valid-pkg", "--silent", "--cmd", "echo hello"]
|
|
)
|
|
|
|
assert result.exit_code == 0
|
|
assert mock_subprocess.call_count == 1
|
|
assert result.output == ""
|