Add tests

Author: ddrscott

pyproject.toml

@@ -12,5 +12,17 @@ dependencies = [
     "click>=8.1.8",
     "rich>=13.9.4",
 ]
+
 [project.scripts]
-wiz = "wiz:cli"
+wiz = "wiz:cli"
+
+[project.optional-dependencies]
+dev = [
+    "pytest>=7.3.1",
+    "pytest-mock>=3.10.0",
+]
+
+[tool.pytest]
+testpaths = ["tests"]
+python_files = "test_*.py"
+python_functions = "test_*"

tests/__init__.py

@@ -0,0 +1 @@
+# Package marker for tests

tests/conftest.py

@@ -0,0 +1,8 @@
+# Common pytest fixtures go here
+import pytest
+from click.testing import CliRunner
+
+@pytest.fixture
+def runner():
+    """Create a CLI runner for testing commands."""
+    return CliRunner()

tests/test_cli.py

@@ -0,0 +1,73 @@
+import pytest
+from unittest.mock import patch, MagicMock, mock_open
+from click.testing import CliRunner
+from wiz import cli, files, apply
+
+def test_files_command(runner):
+    """Test the 'files' command."""
+    with patch('wiz.get_file_table') as mock_get_table:
+        # Mock the return value of get_file_table
+        mock_table = MagicMock()
+        mock_get_table.return_value = (mock_table, ['file1.py', 'file2.py'])
+
+        result = runner.invoke(files)
+
+        assert result.exit_code == 0
+        # The function was called once
+        mock_get_table.assert_called_once()
+
+def test_apply_command_stdin(runner):
+    """Test the 'apply' command with stdin input."""
+    stdin_content = (
+        "[FILE test.py]\n"
+        "print('hello world')\n"
+        "[/FILE]\n"
+    )
+
+    with patch('wiz.process_file_blocks') as mock_process:
+        # Configure the mock to return a single file result
+        mock_process.return_value = [('test.py', "print('hello world')", 1)]
+
+        # Mock file operations
+        with patch('builtins.open', mock_open()):
+            with patch('os.makedirs'):
+                # Invoke the apply command with stdin input
+                result = runner.invoke(apply, ['-'], input=stdin_content)
+
+                assert result.exit_code == 0
+                # Verify process_file_blocks was called
+                mock_process.assert_called_once()
+
+def test_apply_command_file(runner):
+    """Test the 'apply' command with file input."""
+    file_content = (
+        "[FILE test.py]\n"
+        "print('hello world')\n"
+        "[/FILE]\n"
+    )
+
+    with patch('wiz.process_file_blocks') as mock_process:
+        # Configure the mock to return a single file result
+        mock_process.return_value = [('test.py', "print('hello world')", 1)]
+
+        # Mock file operations
+        with patch('builtins.open', mock_open(read_data=file_content)):
+            with patch('os.makedirs'):
+                # Invoke the apply command with file input
+                result = runner.invoke(apply, ['.response.md'])
+
+                assert result.exit_code == 0
+                # Verify process_file_blocks was called
+                mock_process.assert_called_once()
+
+def test_prompt_command(runner):
+    """Test the 'prompt' command."""
+    with patch('wiz.reply') as mock_reply:
+        mock_reply.return_value = "Response content"
+        with patch('builtins.open', mock_open()):
+            result = runner.invoke(cli, ['prompt', 'How can I improve this code?'])
+
+            assert result.exit_code == 0
+            # Verify reply was called with the question
+            mock_reply.assert_called_once()
+            assert 'How can I improve this code?' in mock_reply.call_args[0][0]

tests/test_file_filtering.py

@@ -0,0 +1,65 @@
+import pytest
+from unittest.mock import patch, mock_open
+from wiz import project_files
+
+@pytest.fixture
+def mock_glob():
+    """Mock glob to return predefined file list."""
+    with patch('wiz.glob') as mock:
+        mock.return_value = [
+            'main.py',
+            'utils.py',
+            'test_file.py',
+            '.hidden_file.py',
+            'node_modules/some_file.js',
+            'image.png',
+            'document.pdf',
+            'ignored_pattern.txt'
+        ]
+        yield mock
+
+def test_project_files_basic(mock_glob):
+    """Test basic file filtering without exclude pattern."""
+    with patch('os.path.exists', return_value=False):  # No .gitignore
+        result = project_files()
+        # Should exclude binary files, hidden files, and node_modules
+        assert 'main.py' in result
+        assert 'utils.py' in result
+        assert 'test_file.py' in result
+        assert '.hidden_file.py' not in result
+        assert 'node_modules/some_file.js' not in result
+        assert 'image.png' not in result
+        assert 'document.pdf' not in result
+
+def test_project_files_with_exclude(mock_glob):
+    """Test file filtering with exclude pattern."""
+    with patch('os.path.exists', return_value=False):  # No .gitignore
+        result = project_files(exclude_pattern=r'test_.*\.py$')
+        assert 'main.py' in result
+        assert 'utils.py' in result
+        assert 'test_file.py' not in result
+
+def test_gitignore_patterns():
+    """Test respecting .gitignore patterns."""
+    mock_gitignore_content = "ignored_pattern*\n*.log"
+
+    with patch('wiz.glob', return_value=['file.py', 'ignored_pattern.txt', 'debug.log']):
+        with patch('os.path.exists', return_value=True):
+            with patch('builtins.open', mock_open(read_data=mock_gitignore_content)):
+                with patch('fnmatch.fnmatch') as mock_fnmatch:
+                    # Set up mock behavior
+                    def match_side_effect(filename, pattern):
+                        if pattern == "ignored_pattern*" and filename.startswith("ignored_pattern"):
+                            return True
+                        if pattern == "*.log" and filename.endswith(".log"):
+                            return True
+                        return False
+
+                    mock_fnmatch.side_effect = match_side_effect
+
+                    result = project_files()
+
+                    # Only file.py should be included
+                    assert 'file.py' in result
+                    assert 'ignored_pattern.txt' not in result
+                    assert 'debug.log' not in result

tests/test_parse.py

@@ -0,0 +1,54 @@
+import pytest
+from wiz import process_file_blocks, TAG
+
+def test_process_file_blocks_basic():
+    """Test basic file block processing."""
+    input_lines = [
+        f"[{TAG} test.py]",
+        "print('hello world')",
+        f"[/{TAG}]"
+    ]
+    result = process_file_blocks(input_lines)
+    assert len(result) == 1
+    assert result[0][0] == "test.py"  # file path
+    assert result[0][1] == "print('hello world')"  # content
+    assert result[0][2] == 1  # line number
+
+def test_process_file_blocks_with_code_fence():
+    """Test file block processing with code fences."""
+    input_lines = [
+        f"[{TAG} test.py]",
+        "```python",
+        "print('hello world')",
+        "```",
+        f"[/{TAG}]"
+    ]
+    result = process_file_blocks(input_lines)
+    assert len(result) == 1
+    assert result[0][0] == "test.py"
+    assert result[0][1] == "print('hello world')"
+    assert result[0][2] == 1
+
+def test_process_file_blocks_multiple():
+    """Test processing multiple file blocks."""
+    input_lines = [
+        f"[{TAG} file1.py]",
+        "print('file1')",
+        f"[/{TAG}]",
+        "Some text in between",
+        f"[{TAG} file2.py]",
+        "print('file2')",
+        f"[/{TAG}]"
+    ]
+    result = process_file_blocks(input_lines)
+    assert len(result) == 2
+    assert result[0][0] == "file1.py"
+    assert result[0][1] == "print('file1')"
+    assert result[1][0] == "file2.py"
+    assert result[1][1] == "print('file2')"
+
+def test_process_file_blocks_empty():
+    """Test processing empty input."""
+    input_lines = []
+    result = process_file_blocks(input_lines)
+    assert len(result) == 0