anvil-dev-framework 0.1.7 → 0.1.9

package/project/tests/test-hooks.sh

@@ -0,0 +1,259 @@
+#!/bin/bash
+#
+# Test Harness for Hook Scripts
+#
+# Tests Anvil hook scripts (pre_tool_use.py, session_start.py, etc.)
+# Uses heredocs for JSON fixtures to avoid shell escaping issues.
+#
+# Usage:
+#   ./test-hooks.sh                    # Run all tests
+#   ./test-hooks.sh pre_tool_use       # Test specific hook
+#   ./test-hooks.sh --verbose          # Run with verbose output
+#
+
+set -e
+
+# Load test library
+SCRIPT_DIR="$(cd "$(dirname "$0")" && pwd)"
+source "$SCRIPT_DIR/test-lib.sh"
+
+# Path to hooks
+HOOKS_DIR="$PROJECT_ROOT/project/hooks"
+
+# Check hooks directory exists
+if [ ! -d "$HOOKS_DIR" ]; then
+    echo "Error: hooks directory not found at $HOOKS_DIR"
+    exit 1
+fi
+
+# ============================================================================
+# JSON Fixtures for pre_tool_use.py
+# ============================================================================
+
+read -r -d '' FIX_SAFE_BASH << 'EOF' || true
+{
+  "tool_name": "Bash",
+  "tool_input": {
+    "command": "ls -la"
+  }
+}
+EOF
+
+read -r -d '' FIX_DANGEROUS_RM << 'EOF' || true
+{
+  "tool_name": "Bash",
+  "tool_input": {
+    "command": "rm -rf /"
+  }
+}
+EOF
+
+read -r -d '' FIX_DANGEROUS_RM_HOME << 'EOF' || true
+{
+  "tool_name": "Bash",
+  "tool_input": {
+    "command": "rm -rf ~"
+  }
+}
+EOF
+
+read -r -d '' FIX_ENV_READ << 'EOF' || true
+{
+  "tool_name": "Read",
+  "tool_input": {
+    "file_path": "/project/.env"
+  }
+}
+EOF
+
+read -r -d '' FIX_ENV_SAMPLE << 'EOF' || true
+{
+  "tool_name": "Read",
+  "tool_input": {
+    "file_path": "/project/.env.sample"
+  }
+}
+EOF
+
+read -r -d '' FIX_EDIT_FILE << 'EOF' || true
+{
+  "tool_name": "Edit",
+  "tool_input": {
+    "file_path": "/project/src/app.ts",
+    "old_string": "foo",
+    "new_string": "bar"
+  }
+}
+EOF
+
+read -r -d '' FIX_ASK_USER << 'EOF' || true
+{
+  "tool_name": "AskUserQuestion",
+  "tool_input": {
+    "question": "Which option do you prefer?"
+  }
+}
+EOF
+
+# ============================================================================
+# JSON Fixtures for session_start.py
+# ============================================================================
+
+read -r -d '' FIX_SESSION_START << 'EOF' || true
+{
+  "session_id": "test-session-12345",
+  "cwd": "/Users/test/project"
+}
+EOF
+
+# ============================================================================
+# Test Functions - pre_tool_use.py
+# ============================================================================
+
+test_pre_tool_use_safe_bash() {
+    local hook="$HOOKS_DIR/pre_tool_use.py"
+    [ -f "$hook" ] || { echo "Hook not found: $hook"; return 1; }
+
+    # Safe command should pass (exit 0)
+    local result
+    result=$(echo "$FIX_SAFE_BASH" | python3 "$hook" 2>&1) || true
+    local exit_code=$?
+
+    # Exit code 0 means allowed
+    [ "$exit_code" -eq 0 ] || return 1
+}
+
+test_pre_tool_use_blocks_rm_rf() {
+    local hook="$HOOKS_DIR/pre_tool_use.py"
+    [ -f "$hook" ] || { echo "Hook not found: $hook"; return 1; }
+
+    # Dangerous rm -rf should be blocked (exit 2)
+    set +e
+    echo "$FIX_DANGEROUS_RM" | python3 "$hook" 2>&1
+    local exit_code=$?
+    set -e
+
+    # Exit code 2 means blocked
+    [ "$exit_code" -eq 2 ] || { echo "Expected exit 2, got $exit_code"; return 1; }
+}
+
+test_pre_tool_use_blocks_rm_home() {
+    local hook="$HOOKS_DIR/pre_tool_use.py"
+    [ -f "$hook" ] || { echo "Hook not found: $hook"; return 1; }
+
+    # rm -rf ~ should be blocked
+    set +e
+    echo "$FIX_DANGEROUS_RM_HOME" | python3 "$hook" 2>&1
+    local exit_code=$?
+    set -e
+
+    [ "$exit_code" -eq 2 ] || { echo "Expected exit 2, got $exit_code"; return 1; }
+}
+
+test_pre_tool_use_blocks_env_read() {
+    local hook="$HOOKS_DIR/pre_tool_use.py"
+    [ -f "$hook" ] || { echo "Hook not found: $hook"; return 1; }
+
+    # Reading .env should be blocked
+    set +e
+    echo "$FIX_ENV_READ" | python3 "$hook" 2>&1
+    local exit_code=$?
+    set -e
+
+    [ "$exit_code" -eq 2 ] || { echo "Expected exit 2, got $exit_code"; return 1; }
+}
+
+test_pre_tool_use_allows_env_sample() {
+    local hook="$HOOKS_DIR/pre_tool_use.py"
+    [ -f "$hook" ] || { echo "Hook not found: $hook"; return 1; }
+
+    # Reading .env.sample should be allowed
+    local result
+    result=$(echo "$FIX_ENV_SAMPLE" | python3 "$hook" 2>&1) || true
+    local exit_code=$?
+
+    [ "$exit_code" -eq 0 ] || { echo "Expected exit 0, got $exit_code"; return 1; }
+}
+
+test_pre_tool_use_allows_edit() {
+    local hook="$HOOKS_DIR/pre_tool_use.py"
+    [ -f "$hook" ] || { echo "Hook not found: $hook"; return 1; }
+
+    # Normal file edit should be allowed
+    local result
+    result=$(echo "$FIX_EDIT_FILE" | python3 "$hook" 2>&1) || true
+    local exit_code=$?
+
+    [ "$exit_code" -eq 0 ] || { echo "Expected exit 0, got $exit_code"; return 1; }
+}
+
+test_pre_tool_use_performance() {
+    local hook="$HOOKS_DIR/pre_tool_use.py"
+    [ -f "$hook" ] || { echo "Hook not found: $hook"; return 1; }
+
+    local start end elapsed
+    start=$(python3 -c "import time; print(int(time.time() * 1000))")
+    echo "$FIX_SAFE_BASH" | python3 "$hook" > /dev/null 2>&1
+    end=$(python3 -c "import time; print(int(time.time() * 1000))")
+
+    elapsed=$((end - start))
+
+    # Should complete in under 100ms
+    if [ "$elapsed" -lt 100 ]; then
+        return 0
+    else
+        echo "   Performance: ${elapsed}ms (target: <100ms)"
+        return 1
+    fi
+}
+
+# ============================================================================
+# Test Functions - session_start.py
+# ============================================================================
+
+test_session_start_runs() {
+    local hook="$HOOKS_DIR/session_start.py"
+    [ -f "$hook" ] || { echo "Hook not found: $hook"; return 1; }
+
+    # Should run without error
+    local result
+    result=$(echo "$FIX_SESSION_START" | python3 "$hook" 2>&1) || true
+    local exit_code=$?
+
+    # Exit code 0 means success
+    [ "$exit_code" -eq 0 ] || { echo "Expected exit 0, got $exit_code"; return 1; }
+}
+
+# ============================================================================
+# Run Tests
+# ============================================================================
+
+echo "Hook Test Harness"
+echo "================="
+echo "Hooks Dir: $HOOKS_DIR"
+echo ""
+
+# Parse arguments
+HOOK_FILTER="${1:-}"
+
+if [ -z "$HOOK_FILTER" ] || [ "$HOOK_FILTER" = "--verbose" ] || [ "$HOOK_FILTER" = "pre_tool_use" ]; then
+    echo "Testing: pre_tool_use.py"
+    echo "------------------------"
+    run_test "safe_bash_allowed" test_pre_tool_use_safe_bash
+    run_test "blocks_rm_rf_root" test_pre_tool_use_blocks_rm_rf
+    run_test "blocks_rm_rf_home" test_pre_tool_use_blocks_rm_home
+    run_test "blocks_env_read" test_pre_tool_use_blocks_env_read
+    run_test "allows_env_sample" test_pre_tool_use_allows_env_sample
+    run_test "allows_normal_edit" test_pre_tool_use_allows_edit
+    run_test "performance" test_pre_tool_use_performance
+    echo ""
+fi
+
+if [ -z "$HOOK_FILTER" ] || [ "$HOOK_FILTER" = "--verbose" ] || [ "$HOOK_FILTER" = "session_start" ]; then
+    echo "Testing: session_start.py"
+    echo "-------------------------"
+    run_test "session_start_runs" test_session_start_runs
+    echo ""
+fi
+
+print_summary

package/project/tests/test-lib.sh

@@ -0,0 +1,248 @@
+#!/bin/bash
+#
+# Anvil Test Library - Shared utilities for test harnesses
+#
+# Usage: source this file in test scripts
+#   source "$(dirname "$0")/test-lib.sh"
+#
+# Provides:
+#   - run_test: Execute a test with pass/fail output
+#   - assert_contains: Check if output contains expected pattern
+#   - assert_equals: Check if values are equal
+#   - assert_exit_code: Check command exit code
+#   - setup_test_env: Create isolated test environment
+#   - cleanup_test_env: Clean up test environment
+#
+
+set -e
+
+# Colors for output
+RED='\033[0;31m'
+GREEN='\033[0;32m'
+YELLOW='\033[1;33m'
+NC='\033[0m' # No Color
+
+# Test counters
+TESTS_RUN=0
+TESTS_PASSED=0
+TESTS_FAILED=0
+
+# Get script directory for relative paths
+TEST_LIB_DIR="$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)"
+PROJECT_ROOT="$(cd "$TEST_LIB_DIR/../.." && pwd)"
+
+#
+# run_test - Execute a test function with pass/fail reporting
+#
+# Arguments:
+#   $1 - Test name (displayed in output)
+#   $2 - Test function or command to run
+#
+# Example:
+#   run_test "basic_json_input" test_basic_json
+#
+run_test() {
+    local name="$1"
+    local test_fn="$2"
+
+    TESTS_RUN=$((TESTS_RUN + 1))
+
+    # Run test and capture result
+    if eval "$test_fn" 2>&1; then
+        echo -e "${GREEN}✅ $name${NC}"
+        TESTS_PASSED=$((TESTS_PASSED + 1))
+        return 0
+    else
+        echo -e "${RED}❌ $name${NC}"
+        TESTS_FAILED=$((TESTS_FAILED + 1))
+        return 1
+    fi
+}
+
+#
+# assert_contains - Check if output contains expected pattern
+#
+# Arguments:
+#   $1 - Actual output
+#   $2 - Expected pattern (grep regex, use -E for extended regex)
+#   $3 - (optional) Failure message
+#
+# Returns: 0 if contains, 1 if not
+#
+# Note: For extended regex, pass -E as first arg:
+#   assert_contains "$output" "pattern" "message"       # Basic regex
+#   assert_contains "$output" "pat|tern" "message"      # Extended (auto-detected)
+#
+assert_contains() {
+    local actual="$1"
+    local expected="$2"
+    local message="${3:-Expected pattern not found}"
+
+    # Use extended regex if pattern contains | or ()
+    local grep_flags="-q"
+    if echo "$expected" | grep -qE '\||\(|\)|\+|\?'; then
+        grep_flags="-qE"
+    fi
+
+    if echo "$actual" | grep $grep_flags "$expected"; then
+        return 0
+    else
+        echo -e "   ${YELLOW}$message${NC}"
+        echo "   Expected pattern: $expected"
+        echo "   Actual output: $actual"
+        return 1
+    fi
+}
+
+#
+# assert_not_contains - Check if output does NOT contain pattern
+#
+# Arguments:
+#   $1 - Actual output
+#   $2 - Pattern that should NOT be present
+#   $3 - (optional) Failure message
+#
+assert_not_contains() {
+    local actual="$1"
+    local pattern="$2"
+    local message="${3:-Unexpected pattern found}"
+
+    if echo "$actual" | grep -q "$pattern"; then
+        echo -e "   ${YELLOW}$message${NC}"
+        echo "   Unexpected pattern: $pattern"
+        echo "   Actual output: $actual"
+        return 1
+    else
+        return 0
+    fi
+}
+
+#
+# assert_equals - Check if two values are equal
+#
+# Arguments:
+#   $1 - Actual value
+#   $2 - Expected value
+#   $3 - (optional) Failure message
+#
+assert_equals() {
+    local actual="$1"
+    local expected="$2"
+    local message="${3:-Values not equal}"
+
+    if [ "$actual" = "$expected" ]; then
+        return 0
+    else
+        echo -e "   ${YELLOW}$message${NC}"
+        echo "   Expected: $expected"
+        echo "   Actual: $actual"
+        return 1
+    fi
+}
+
+#
+# assert_exit_code - Check command exit code
+#
+# Arguments:
+#   $1 - Expected exit code
+#   $2... - Command to run
+#
+# Example:
+#   assert_exit_code 0 echo "hello"
+#   assert_exit_code 2 python3 script.py  # Expect exit 2
+#
+assert_exit_code() {
+    local expected="$1"
+    shift
+    local cmd="$@"
+
+    set +e
+    eval "$cmd" >/dev/null 2>&1
+    local actual=$?
+    set -e
+
+    if [ "$actual" -eq "$expected" ]; then
+        return 0
+    else
+        echo -e "   ${YELLOW}Wrong exit code${NC}"
+        echo "   Expected: $expected"
+        echo "   Actual: $actual"
+        echo "   Command: $cmd"
+        return 1
+    fi
+}
+
+#
+# setup_test_env - Create isolated test environment
+#
+# Creates a temporary directory and sets up mock environment variables
+#
+# Returns: Path to temp directory (also sets TEST_TEMP_DIR)
+#
+setup_test_env() {
+    TEST_TEMP_DIR=$(mktemp -d)
+
+    # Mock environment variables
+    export CLAUDE_SESSION_ID="test-session-$(date +%s)"
+    export CLAUDE_MODEL="claude-opus-4-5-20251101"
+
+    # Create mock directory structure
+    mkdir -p "$TEST_TEMP_DIR/logs"
+    mkdir -p "$TEST_TEMP_DIR/.claude"
+
+    echo "$TEST_TEMP_DIR"
+}
+
+#
+# cleanup_test_env - Clean up test environment
+#
+cleanup_test_env() {
+    if [ -n "$TEST_TEMP_DIR" ] && [ -d "$TEST_TEMP_DIR" ]; then
+        rm -rf "$TEST_TEMP_DIR"
+    fi
+    unset TEST_TEMP_DIR
+    unset CLAUDE_SESSION_ID
+    unset CLAUDE_MODEL
+}
+
+#
+# print_summary - Print test run summary
+#
+print_summary() {
+    echo ""
+    echo "━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━"
+    echo "Test Summary"
+    echo "━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━"
+    echo "Total:  $TESTS_RUN"
+    echo -e "Passed: ${GREEN}$TESTS_PASSED${NC}"
+    if [ "$TESTS_FAILED" -gt 0 ]; then
+        echo -e "Failed: ${RED}$TESTS_FAILED${NC}"
+    else
+        echo "Failed: $TESTS_FAILED"
+    fi
+    echo "━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━"
+
+    if [ "$TESTS_FAILED" -gt 0 ]; then
+        return 1
+    fi
+    return 0
+}
+
+#
+# json_fixture - Create JSON fixture using heredoc (avoids escaping issues)
+#
+# Usage:
+#   json_fixture FIXTURE_NAME << 'EOF'
+#   {"key": "value"}
+#   EOF
+#
+json_fixture() {
+    local name="$1"
+    eval "read -r -d '' $name" || true
+}
+
+# Export functions for use in subshells
+export -f run_test assert_contains assert_not_contains assert_equals assert_exit_code
+export -f setup_test_env cleanup_test_env print_summary json_fixture
+export TESTS_RUN TESTS_PASSED TESTS_FAILED
+export TEST_LIB_DIR PROJECT_ROOT

package/project/tests/test-statusline.sh

@@ -0,0 +1,165 @@
+#!/bin/bash
+#
+# Test Harness for statusline.sh
+#
+# Tests the Anvil statusline script with various JSON inputs.
+# Uses heredocs for JSON fixtures to avoid shell escaping issues.
+#
+# Usage:
+#   ./test-statusline.sh           # Run all tests
+#   ./test-statusline.sh --verbose # Run with verbose output
+#
+
+set -e
+
+# Load test library
+SCRIPT_DIR="$(cd "$(dirname "$0")" && pwd)"
+source "$SCRIPT_DIR/test-lib.sh"
+
+# Path to statusline script
+STATUSLINE="$PROJECT_ROOT/global/hooks/statusline.sh"
+
+# Check statusline exists
+if [ ! -f "$STATUSLINE" ]; then
+    echo "Error: statusline.sh not found at $STATUSLINE"
+    exit 1
+fi
+
+# ============================================================================
+# JSON Fixtures (heredocs avoid escaping issues)
+# ============================================================================
+
+read -r -d '' FIXTURE_BASIC << 'EOF' || true
+{
+  "cwd": "/Users/test/project",
+  "model": {
+    "display_name": "Claude Opus 4.5"
+  }
+}
+EOF
+
+read -r -d '' FIXTURE_WITH_MESSAGE << 'EOF' || true
+{
+  "cwd": "/Users/test/project",
+  "status_message": "Building feature",
+  "model": {
+    "display_name": "Claude 3.5 Sonnet"
+  }
+}
+EOF
+
+read -r -d '' FIXTURE_HAIKU << 'EOF' || true
+{
+  "cwd": "/Users/test/project",
+  "model": {
+    "display_name": "Claude 3 Haiku"
+  }
+}
+EOF
+
+read -r -d '' FIXTURE_EMPTY << 'EOF' || true
+{}
+EOF
+
+read -r -d '' FIXTURE_NO_MODEL << 'EOF' || true
+{
+  "cwd": "/Users/test/project",
+  "conversation": {}
+}
+EOF
+
+# ============================================================================
+# Test Functions
+# ============================================================================
+
+test_basic_output() {
+    local output
+    output=$(echo "$FIXTURE_BASIC" | "$STATUSLINE" 2>&1)
+
+    # Should produce some output
+    [ -n "$output" ] || return 1
+
+    # Should contain model indicator (Opus or Claude)
+    # Pattern with | is auto-detected as extended regex
+    assert_contains "$output" "Opus|opus|Claude|claude" "Should contain model name"
+}
+
+test_with_status_message() {
+    local output
+    output=$(echo "$FIXTURE_WITH_MESSAGE" | "$STATUSLINE" 2>&1)
+
+    # Should contain Sonnet model
+    assert_contains "$output" "Sonnet|sonnet" "Should contain Sonnet model"
+}
+
+test_haiku_model() {
+    local output
+    output=$(echo "$FIXTURE_HAIKU" | "$STATUSLINE" 2>&1)
+
+    # Should contain Haiku model
+    assert_contains "$output" "Haiku|haiku" "Should contain Haiku model"
+}
+
+test_empty_input() {
+    local output
+    # Should not crash on empty input
+    output=$(echo "$FIXTURE_EMPTY" | "$STATUSLINE" 2>&1) || true
+
+    # Should produce some output (even if minimal)
+    [ -n "$output" ] || return 0  # Empty output is acceptable for empty input
+}
+
+test_no_model() {
+    local output
+    # Should handle missing model gracefully
+    output=$(echo "$FIXTURE_NO_MODEL" | "$STATUSLINE" 2>&1) || true
+
+    # Should not crash
+    return 0
+}
+
+test_invalid_json() {
+    local output
+    # Should handle invalid JSON gracefully
+    output=$(echo "not valid json" | "$STATUSLINE" 2>&1) || true
+
+    # Should not crash - exit code 0 or non-empty fallback
+    return 0
+}
+
+test_performance() {
+    local start end elapsed
+
+    start=$(python3 -c "import time; print(int(time.time() * 1000))")
+    echo "$FIXTURE_BASIC" | "$STATUSLINE" > /dev/null 2>&1
+    end=$(python3 -c "import time; print(int(time.time() * 1000))")
+
+    elapsed=$((end - start))
+
+    # Should complete in under 200ms
+    if [ "$elapsed" -lt 200 ]; then
+        return 0
+    else
+        echo "   Performance: ${elapsed}ms (target: <200ms)"
+        return 1
+    fi
+}
+
+# ============================================================================
+# Run Tests
+# ============================================================================
+
+echo "Statusline Test Harness"
+echo "======================="
+echo "Script: $STATUSLINE"
+echo ""
+
+run_test "basic_output" test_basic_output
+run_test "with_status_message" test_with_status_message
+run_test "haiku_model" test_haiku_model
+run_test "empty_input" test_empty_input
+run_test "no_model" test_no_model
+run_test "invalid_json" test_invalid_json
+run_test "performance" test_performance
+
+print_summary