Enable tests also locally

Author: rbx

CLAUDE.md

@@ -38,10 +38,25 @@ Pearbot is an AI-powered code review tool that analyzes Pull Requests on GitHub
 # Install dependencies
 pip install .
 
+# Install test dependencies
+pip install .[test]
+
 # Install Ollama model (default: llama3.1)
 ollama pull llama3.1
 ```
 
+### Testing
+```bash
+# Run tests locally
+pytest
+
+# Run tests with verbose output
+pytest -v
+
+# Run specific test file
+pytest tests/test_basic.py
+```
+
 ### Running the Application
 
 ```bash

pyproject.toml

@@ -16,6 +16,11 @@ dependencies = [
     "pyyaml==6.0.1",
 ]
 
+[project.optional-dependencies]
+test = [
+    "pytest>=7.0.0",
+]
+
 [project.urls]
 repository = "https://github.com/rbx/pearbot"
 

pytest.ini

@@ -0,0 +1,6 @@
+[tool:pytest]
+testpaths = tests
+python_files = test_*.py
+python_classes = Test*
+python_functions = test_*
+addopts = -v

src/agents.py

@@ -1,7 +1,7 @@
 import ollama
 import yaml
 
-from model import post_request_generate
+from .model import post_request_generate
 
 class Agent:
     def __init__(self, role="code_reviewer", use_post_request=False, prompt_style="default"):

src/pearbot.py

@@ -5,11 +5,11 @@
 
 load_dotenv()
 
-from storage import get_or_create_session
-from agents import Agent
-from review_github import GitHubReviewer
-from review_local import analyze_diff
-from ollama_utils import get_available_models
+from .storage import get_or_create_session
+from .agents import Agent
+from .review_github import GitHubReviewer
+from .review_local import analyze_diff
+from .ollama_utils import get_available_models
 
 def main():
     parser = argparse.ArgumentParser(description="Pearbot Code Review")

src/review_github.py

@@ -10,8 +10,8 @@
 from github import Github, GithubException
 from flask import Flask, request, abort
 
-from ollama_utils import validate_models
-from utils import remove_reasoning
+from .ollama_utils import validate_models
+from .utils import remove_reasoning
 
 class GitHubReviewer:
     def __init__(self, code_review_agent, feedback_improver_agent, initial_review_models, final_review_model, skip_reasoning: bool):

src/review_local.py

@@ -4,8 +4,8 @@
 
 from colorama import Fore, Style
 
-from ollama_utils import validate_models
-from utils import remove_reasoning
+from .ollama_utils import validate_models
+from .utils import remove_reasoning
 
 def extract_commit_info(diff_content):
     commit_range = None

tests/__init__.py

@@ -0,0 +1,56 @@
+import os
+import subprocess
+import sys
+import pytest
+from unittest.mock import patch
+
+def test_module_import():
+    """Test that pearbot module can be imported successfully."""
+    try:
+        from src import pearbot
+        assert True, "pearbot module imported successfully"
+    except ImportError as e:
+        pytest.fail(f"Failed to import pearbot module: {e}")
+
+def test_help_command():
+    """Test that --help command works."""
+    # Set dummy environment variables like in CI
+    env = os.environ.copy()
+    env.update({
+        'GITHUB_APP_ID': 'dummy_id',
+        'GITHUB_PRIVATE_KEY': 'dummy_key',
+        'GITHUB_APP_WEBHOOK_SECRET': 'dummy_secret'
+    })
+
+    result = subprocess.run(
+        [sys.executable, '-m', 'src.pearbot', '--help'],
+        capture_output=True,
+        text=True,
+        env=env
+    )
+
+    assert result.returncode == 0, f"Help command failed with: {result.stderr}"
+    assert "Pearbot Code Review" in result.stdout, "Help text should contain 'Pearbot Code Review'"
+    assert "--server" in result.stdout, "Help should show --server option"
+    assert "--diff" in result.stdout, "Help should show --diff option"
+
+def test_server_startup():
+    """Test that server can start (with timeout like in CI)."""
+    # Set dummy environment variables like in CI
+    env = os.environ.copy()
+    env.update({
+        'GITHUB_APP_ID': 'dummy_id',
+        'GITHUB_PRIVATE_KEY': 'dummy_key',
+        'GITHUB_APP_WEBHOOK_SECRET': 'dummy_secret'
+    })
+
+    # Use timeout like in CI - expect it to timeout (exit code 124) or succeed (exit code 0)
+    result = subprocess.run(
+        ['timeout', '5s', sys.executable, '-m', 'src.pearbot', '--server'],
+        capture_output=True,
+        text=True,
+        env=env
+    )
+
+    # Exit code 124 means timeout (expected), 0 means success, anything else is failure
+    assert result.returncode in [0, 124], f"Server startup failed with exit code {result.returncode}: {result.stderr}"