231 lines
7.1 KiB
Python
231 lines
7.1 KiB
Python
"""Tests for CLI commands."""
|
|
|
|
import pytest
|
|
import tempfile
|
|
import os
|
|
from click.testing import CliRunner
|
|
|
|
from src.cli import main
|
|
|
|
|
|
class TestCLIPromptCommands:
|
|
"""Test cases for prompt CLI commands."""
|
|
|
|
def setup_method(self):
|
|
"""Set up test fixtures."""
|
|
self.runner = CliRunner()
|
|
self.temp_dir = tempfile.mkdtemp()
|
|
|
|
def teardown_method(self):
|
|
"""Clean up test fixtures."""
|
|
import shutil
|
|
shutil.rmtree(self.temp_dir, ignore_errors=True)
|
|
|
|
def test_prompt_create(self):
|
|
"""Test creating a prompt."""
|
|
result = self.runner.invoke(main, [
|
|
"prompt", "create", "test-prompt",
|
|
"--template", "Hello {{ name }}",
|
|
"--description", "A test prompt",
|
|
"--dir", self.temp_dir
|
|
])
|
|
assert result.exit_code == 0
|
|
assert "Created prompt: test-prompt" in result.output
|
|
|
|
def test_prompt_create_duplicate(self):
|
|
"""Test creating duplicate prompt."""
|
|
self.runner.invoke(main, [
|
|
"prompt", "create", "test-prompt",
|
|
"--template", "Hello {{ name }}",
|
|
"--dir", self.temp_dir
|
|
])
|
|
result = self.runner.invoke(main, [
|
|
"prompt", "create", "test-prompt",
|
|
"--template", "Hello {{ name }}",
|
|
"--dir", self.temp_dir
|
|
])
|
|
assert result.exit_code == 0
|
|
assert "already exists" in result.output
|
|
|
|
def test_prompt_list(self):
|
|
"""Test listing prompts."""
|
|
self.runner.invoke(main, [
|
|
"prompt", "create", "test-prompt",
|
|
"--template", "Hello {{ name }}",
|
|
"--dir", self.temp_dir
|
|
])
|
|
result = self.runner.invoke(main, [
|
|
"prompt", "list",
|
|
"--dir", self.temp_dir
|
|
])
|
|
assert result.exit_code == 0
|
|
assert "test-prompt" in result.output
|
|
|
|
def test_prompt_list_empty(self):
|
|
"""Test listing prompts when empty."""
|
|
result = self.runner.invoke(main, [
|
|
"prompt", "list",
|
|
"--dir", self.temp_dir
|
|
])
|
|
assert result.exit_code == 0
|
|
|
|
def test_prompt_delete(self):
|
|
"""Test deleting a prompt."""
|
|
self.runner.invoke(main, [
|
|
"prompt", "create", "test-prompt",
|
|
"--template", "Hello {{ name }}",
|
|
"--dir", self.temp_dir
|
|
])
|
|
result = self.runner.invoke(main, [
|
|
"prompt", "delete", "test-prompt",
|
|
"--force",
|
|
"--dir", self.temp_dir
|
|
])
|
|
assert result.exit_code == 0
|
|
assert "Deleted prompt" in result.output
|
|
|
|
def test_prompt_delete_not_found(self):
|
|
"""Test deleting non-existent prompt."""
|
|
result = self.runner.invoke(main, [
|
|
"prompt", "delete", "non-existent",
|
|
"--force",
|
|
"--dir", self.temp_dir
|
|
])
|
|
assert result.exit_code != 0 or "not found" in result.output
|
|
|
|
|
|
class TestCLITagCommands:
|
|
"""Test cases for tag CLI commands."""
|
|
|
|
def setup_method(self):
|
|
"""Set up test fixtures."""
|
|
self.runner = CliRunner()
|
|
self.temp_dir = tempfile.mkdtemp()
|
|
|
|
def teardown_method(self):
|
|
"""Clean up test fixtures."""
|
|
import shutil
|
|
shutil.rmtree(self.temp_dir, ignore_errors=True)
|
|
|
|
def test_tag_add(self):
|
|
"""Test adding a tag to prompt."""
|
|
self.runner.invoke(main, [
|
|
"prompt", "create", "test-prompt",
|
|
"--template", "Hello {{ name }}",
|
|
"--dir", self.temp_dir
|
|
])
|
|
result = self.runner.invoke(main, [
|
|
"tag", "add", "test-prompt", "test-tag",
|
|
"--dir", self.temp_dir
|
|
])
|
|
assert result.exit_code == 0
|
|
assert "Added tag" in result.output
|
|
|
|
def test_tag_list(self):
|
|
"""Test listing tags."""
|
|
self.runner.invoke(main, [
|
|
"prompt", "create", "test-prompt",
|
|
"--template", "Hello {{ name }}",
|
|
"--dir", self.temp_dir
|
|
])
|
|
self.runner.invoke(main, [
|
|
"tag", "add", "test-prompt", "test-tag",
|
|
"--dir", self.temp_dir
|
|
])
|
|
result = self.runner.invoke(main, [
|
|
"tag", "list",
|
|
"--dir", self.temp_dir
|
|
])
|
|
assert result.exit_code == 0
|
|
assert "test-tag" in result.output
|
|
|
|
def test_tag_remove(self):
|
|
"""Test removing a tag from prompt."""
|
|
self.runner.invoke(main, [
|
|
"prompt", "create", "test-prompt",
|
|
"--template", "Hello {{ name }}",
|
|
"--dir", self.temp_dir
|
|
])
|
|
self.runner.invoke(main, [
|
|
"tag", "add", "test-prompt", "test-tag",
|
|
"--dir", self.temp_dir
|
|
])
|
|
result = self.runner.invoke(main, [
|
|
"tag", "remove", "test-prompt", "test-tag",
|
|
"--dir", self.temp_dir
|
|
])
|
|
assert result.exit_code == 0
|
|
assert "Removed tag" in result.output
|
|
|
|
|
|
class TestCLISearchCommands:
|
|
"""Test cases for search CLI commands."""
|
|
|
|
def setup_method(self):
|
|
"""Set up test fixtures."""
|
|
self.runner = CliRunner()
|
|
self.temp_dir = tempfile.mkdtemp()
|
|
|
|
def teardown_method(self):
|
|
"""Clean up test fixtures."""
|
|
import shutil
|
|
shutil.rmtree(self.temp_dir, ignore_errors=True)
|
|
|
|
def test_search_by_name(self):
|
|
"""Test searching by name."""
|
|
self.runner.invoke(main, [
|
|
"prompt", "create", "my-test-prompt",
|
|
"--template", "Hello {{ name }}",
|
|
"--dir", self.temp_dir
|
|
])
|
|
result = self.runner.invoke(main, [
|
|
"search", "my-test-prompt",
|
|
"--name", "my-test",
|
|
"--dir", self.temp_dir
|
|
])
|
|
assert result.exit_code == 0
|
|
assert "my-test-prompt" in result.output
|
|
|
|
def test_search_by_tag(self):
|
|
"""Test searching by tag."""
|
|
self.runner.invoke(main, [
|
|
"prompt", "create", "test-prompt",
|
|
"--template", "Hello {{ name }}",
|
|
"--tag", "documentation",
|
|
"--dir", self.temp_dir
|
|
])
|
|
result = self.runner.invoke(main, [
|
|
"search",
|
|
"--tag", "documentation",
|
|
"--dir", self.temp_dir
|
|
])
|
|
assert result.exit_code == 0
|
|
assert "test-prompt" in result.output
|
|
|
|
|
|
class TestCLIConfigCommands:
|
|
"""Test cases for config CLI commands."""
|
|
|
|
def setup_method(self):
|
|
"""Set up test fixtures."""
|
|
self.runner = CliRunner()
|
|
self.temp_dir = tempfile.mkdtemp()
|
|
|
|
def teardown_method(self):
|
|
"""Clean up test fixtures."""
|
|
import shutil
|
|
shutil.rmtree(self.temp_dir, ignore_errors=True)
|
|
|
|
def test_config_show(self):
|
|
"""Test showing configuration."""
|
|
result = self.runner.invoke(main, ["config", "show"])
|
|
assert result.exit_code == 0
|
|
assert "ollama_url" in result.output or "Configuration" in result.output
|
|
|
|
def test_config_set(self):
|
|
"""Test setting configuration value."""
|
|
result = self.runner.invoke(main, [
|
|
"config", "set", "default_model", "test-model"
|
|
])
|
|
assert result.exit_code == 0 or "Set" in result.output
|