antigravity-ide 4.2.1 → 4.3.3

package/.agent/skills/loki-mode/tests/test-task-queue.sh

@@ -0,0 +1,396 @@
+#!/bin/bash
+# Test: Distributed Task Queue Functionality
+# Tests task creation, claiming, completion, and failure handling
+
+set -uo pipefail
+# Note: Not using -e to allow collecting all test results
+
+SCRIPT_DIR="$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)"
+TEST_DIR=$(mktemp -d)
+PASSED=0
+FAILED=0
+
+# Colors
+RED='\033[0;31m'
+GREEN='\033[0;32m'
+YELLOW='\033[1;33m'
+NC='\033[0m'
+
+log_pass() { echo -e "${GREEN}[PASS]${NC} $1"; ((PASSED++)); }
+log_fail() { echo -e "${RED}[FAIL]${NC} $1"; ((FAILED++)); }
+log_test() { echo -e "${YELLOW}[TEST]${NC} $1"; }
+
+cleanup() {
+    rm -rf "$TEST_DIR"
+}
+trap cleanup EXIT
+
+cd "$TEST_DIR"
+
+echo "========================================"
+echo "Loki Mode Task Queue Tests"
+echo "========================================"
+echo ""
+
+# Initialize structure
+mkdir -p .loki/{state/locks,queue}
+for f in pending in-progress completed failed dead-letter; do
+    echo '{"tasks":[]}' > ".loki/queue/$f.json"
+done
+
+# Helper function to add task
+add_task() {
+    local id="$1"
+    local type="$2"
+    local priority="${3:-5}"
+
+    local task=$(cat <<EOF
+{
+  "id": "$id",
+  "type": "$type",
+  "priority": $priority,
+  "payload": {"action": "test"},
+  "createdAt": "$(date -u +%Y-%m-%dT%H:%M:%SZ)",
+  "claimedBy": null,
+  "claimedAt": null,
+  "timeout": 3600,
+  "retries": 0,
+  "maxRetries": 3
+}
+EOF
+)
+
+    # Add to pending queue
+    if command -v jq &> /dev/null; then
+        jq --argjson task "$task" '.tasks += [$task]' .loki/queue/pending.json > tmp.json && mv tmp.json .loki/queue/pending.json
+    else
+        # Fallback without jq
+        python3 -c "
+import json
+with open('.loki/queue/pending.json', 'r') as f:
+    data = json.load(f)
+task = json.loads('''$task''')
+data['tasks'].append(task)
+with open('.loki/queue/pending.json', 'w') as f:
+    json.dump(data, f)
+"
+    fi
+}
+
+# Test 1: Add task to pending queue
+log_test "Add task to pending queue"
+add_task "task-001" "eng-backend" 5
+
+task_count=$(python3 -c "import json; print(len(json.load(open('.loki/queue/pending.json'))['tasks']))")
+if [ "$task_count" -eq 1 ]; then
+    log_pass "Task added to pending queue"
+else
+    log_fail "Failed to add task (count: $task_count)"
+fi
+
+# Test 2: Add multiple tasks with priorities
+log_test "Add multiple tasks with priorities"
+add_task "task-002" "eng-frontend" 3
+add_task "task-003" "eng-backend" 10
+add_task "task-004" "ops-devops" 1
+
+task_count=$(python3 -c "import json; print(len(json.load(open('.loki/queue/pending.json'))['tasks']))")
+if [ "$task_count" -eq 4 ]; then
+    log_pass "Multiple tasks added"
+else
+    log_fail "Failed to add multiple tasks (count: $task_count)"
+fi
+
+# Test 3: Priority ordering
+log_test "Priority ordering"
+highest_priority=$(python3 -c "
+import json
+data = json.load(open('.loki/queue/pending.json'))
+sorted_tasks = sorted(data['tasks'], key=lambda t: -t['priority'])
+print(sorted_tasks[0]['id'])
+")
+
+if [ "$highest_priority" = "task-003" ]; then
+    log_pass "Highest priority task is task-003 (priority 10)"
+else
+    log_fail "Priority ordering wrong: got $highest_priority, expected task-003"
+fi
+
+# Test 4: Claim task (atomic operation simulation)
+log_test "Claim task atomically"
+python3 << 'EOF'
+import json
+import os
+from datetime import datetime
+
+# Simulate atomic claim with file locking
+queue_file = '.loki/queue/pending.json'
+progress_file = '.loki/queue/in-progress.json'
+lock_file = '.loki/state/locks/queue.lock'
+
+# Read pending
+with open(queue_file, 'r') as f:
+    pending = json.load(f)
+
+# Find highest priority unclaimed task
+tasks = sorted(pending['tasks'], key=lambda t: -t['priority'])
+claimed_task = None
+for task in tasks:
+    if task.get('claimedBy') is None:
+        task['claimedBy'] = 'agent-001'
+        task['claimedAt'] = datetime.utcnow().isoformat() + 'Z'
+        claimed_task = task
+        break
+
+if claimed_task:
+    # Remove from pending
+    pending['tasks'] = [t for t in pending['tasks'] if t['id'] != claimed_task['id']]
+
+    # Add to in-progress
+    with open(progress_file, 'r') as f:
+        progress = json.load(f)
+    progress['tasks'].append(claimed_task)
+
+    # Write both files
+    with open(queue_file, 'w') as f:
+        json.dump(pending, f)
+    with open(progress_file, 'w') as f:
+        json.dump(progress, f)
+
+    print(f"CLAIMED:{claimed_task['id']}")
+else:
+    print("NONE")
+EOF
+
+claimed=$(python3 -c "
+import json
+data = json.load(open('.loki/queue/in-progress.json'))
+if data['tasks']:
+    print(data['tasks'][0]['id'])
+else:
+    print('NONE')
+")
+
+if [ "$claimed" = "task-003" ]; then
+    log_pass "Claimed highest priority task (task-003)"
+else
+    log_fail "Claim failed: got $claimed"
+fi
+
+# Test 5: Complete task
+log_test "Complete task"
+python3 << 'EOF'
+import json
+from datetime import datetime
+
+progress_file = '.loki/queue/in-progress.json'
+completed_file = '.loki/queue/completed.json'
+
+with open(progress_file, 'r') as f:
+    progress = json.load(f)
+
+with open(completed_file, 'r') as f:
+    completed = json.load(f)
+
+# Complete first task
+if progress['tasks']:
+    task = progress['tasks'][0]
+    task['completedAt'] = datetime.utcnow().isoformat() + 'Z'
+    task['result'] = {'status': 'success'}
+
+    completed['tasks'].append(task)
+    progress['tasks'] = progress['tasks'][1:]
+
+    with open(progress_file, 'w') as f:
+        json.dump(progress, f)
+    with open(completed_file, 'w') as f:
+        json.dump(completed, f)
+
+    print("COMPLETED")
+EOF
+
+completed_count=$(python3 -c "import json; print(len(json.load(open('.loki/queue/completed.json'))['tasks']))")
+if [ "$completed_count" -eq 1 ]; then
+    log_pass "Task completed successfully"
+else
+    log_fail "Task completion failed"
+fi
+
+# Test 6: Fail task with retry
+log_test "Fail task with retry"
+# First claim a task
+python3 << 'EOF'
+import json
+from datetime import datetime
+
+queue_file = '.loki/queue/pending.json'
+progress_file = '.loki/queue/in-progress.json'
+
+with open(queue_file, 'r') as f:
+    pending = json.load(f)
+
+if pending['tasks']:
+    task = pending['tasks'][0]
+    task['claimedBy'] = 'agent-002'
+    task['claimedAt'] = datetime.utcnow().isoformat() + 'Z'
+
+    with open(progress_file, 'r') as f:
+        progress = json.load(f)
+
+    progress['tasks'].append(task)
+    pending['tasks'] = pending['tasks'][1:]
+
+    with open(queue_file, 'w') as f:
+        json.dump(pending, f)
+    with open(progress_file, 'w') as f:
+        json.dump(progress, f)
+EOF
+
+# Now fail it
+python3 << 'EOF'
+import json
+from datetime import datetime
+
+progress_file = '.loki/queue/in-progress.json'
+pending_file = '.loki/queue/pending.json'
+
+with open(progress_file, 'r') as f:
+    progress = json.load(f)
+
+if progress['tasks']:
+    task = progress['tasks'][0]
+    task['retries'] = task.get('retries', 0) + 1
+    task['lastError'] = 'Test failure'
+    task['claimedBy'] = None
+    task['claimedAt'] = None
+    task['backoffSeconds'] = 60 * (2 ** (task['retries'] - 1))
+
+    # Move back to pending for retry
+    with open(pending_file, 'r') as f:
+        pending = json.load(f)
+
+    pending['tasks'].append(task)
+    progress['tasks'] = progress['tasks'][1:]
+
+    with open(progress_file, 'w') as f:
+        json.dump(progress, f)
+    with open(pending_file, 'w') as f:
+        json.dump(pending, f)
+
+    print(f"RETRY:{task['retries']}")
+EOF
+
+retry_count=$(python3 -c "
+import json
+data = json.load(open('.loki/queue/pending.json'))
+for t in data['tasks']:
+    if t.get('retries', 0) > 0:
+        print(t['retries'])
+        break
+else:
+    print(0)
+")
+
+if [ "$retry_count" -eq 1 ]; then
+    log_pass "Task moved back to pending with retry count"
+else
+    log_fail "Retry handling failed"
+fi
+
+# Test 7: Dead letter queue
+log_test "Move to dead letter queue after max retries"
+python3 << 'EOF'
+import json
+from datetime import datetime
+
+pending_file = '.loki/queue/pending.json'
+dlq_file = '.loki/queue/dead-letter.json'
+
+with open(pending_file, 'r') as f:
+    pending = json.load(f)
+
+with open(dlq_file, 'r') as f:
+    dlq = json.load(f)
+
+# Find task with retries and simulate max retries exceeded
+for task in pending['tasks']:
+    if task.get('retries', 0) > 0:
+        task['retries'] = task.get('maxRetries', 3)
+        task['lastError'] = 'Max retries exceeded'
+        task['movedToDLQ'] = datetime.utcnow().isoformat() + 'Z'
+
+        dlq['tasks'].append(task)
+        pending['tasks'] = [t for t in pending['tasks'] if t['id'] != task['id']]
+        break
+
+with open(pending_file, 'w') as f:
+    json.dump(pending, f)
+with open(dlq_file, 'w') as f:
+    json.dump(dlq, f)
+
+print("MOVED_TO_DLQ")
+EOF
+
+dlq_count=$(python3 -c "import json; print(len(json.load(open('.loki/queue/dead-letter.json'))['tasks']))")
+if [ "$dlq_count" -eq 1 ]; then
+    log_pass "Task moved to dead letter queue"
+else
+    log_fail "Dead letter queue handling failed"
+fi
+
+# Test 8: Idempotency check
+log_test "Idempotency check (duplicate prevention)"
+python3 << 'EOF'
+import json
+import hashlib
+
+pending_file = '.loki/queue/pending.json'
+
+with open(pending_file, 'r') as f:
+    pending = json.load(f)
+
+# Try to add duplicate task
+new_task = {
+    "id": "task-duplicate",
+    "type": "eng-backend",
+    "payload": {"action": "test"}
+}
+
+# Generate idempotency key
+idempotency_key = hashlib.md5(json.dumps(new_task['payload'], sort_keys=True).encode()).hexdigest()
+new_task['idempotencyKey'] = idempotency_key
+
+# Check if already exists
+existing = [t for t in pending['tasks'] if t.get('idempotencyKey') == idempotency_key]
+if not existing:
+    pending['tasks'].append(new_task)
+    print("ADDED")
+else:
+    print("DUPLICATE")
+
+# Try again with same payload
+existing = [t for t in pending['tasks'] if t.get('idempotencyKey') == idempotency_key]
+if existing:
+    print("DUPLICATE_DETECTED")
+
+with open(pending_file, 'w') as f:
+    json.dump(pending, f)
+EOF
+
+log_pass "Idempotency check works"
+
+echo ""
+echo "========================================"
+echo "Test Summary"
+echo "========================================"
+echo -e "${GREEN}Passed: $PASSED${NC}"
+echo -e "${RED}Failed: $FAILED${NC}"
+echo ""
+
+if [ $FAILED -eq 0 ]; then
+    echo -e "${GREEN}All tests passed!${NC}"
+    exit 0
+else
+    echo -e "${RED}Some tests failed!${NC}"
+    exit 1
+fi

package/.agent/skills/loki-mode/tests/test-wrapper.sh

@@ -0,0 +1,314 @@
+#!/bin/bash
+# Test: Loki Mode Wrapper Script
+# Tests the autonomous wrapper functionality
+
+set -uo pipefail
+
+TEST_DIR=$(mktemp -d)
+SCRIPT_DIR="$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)"
+WRAPPER_SCRIPT="$SCRIPT_DIR/../scripts/loki-wrapper.sh"
+PASSED=0
+FAILED=0
+
+# Colors
+RED='\033[0;31m'
+GREEN='\033[0;32m'
+YELLOW='\033[1;33m'
+NC='\033[0m'
+
+log_pass() { echo -e "${GREEN}[PASS]${NC} $1"; ((PASSED++)); }
+log_fail() { echo -e "${RED}[FAIL]${NC} $1"; ((FAILED++)); }
+log_test() { echo -e "${YELLOW}[TEST]${NC} $1"; }
+
+cleanup() {
+    rm -rf "$TEST_DIR"
+}
+trap cleanup EXIT
+
+cd "$TEST_DIR"
+
+echo "=========================================="
+echo "Loki Mode Wrapper Script Tests"
+echo "=========================================="
+echo ""
+
+# Test 1: Wrapper script exists and is executable
+log_test "Wrapper script exists and is executable"
+if [ -x "$WRAPPER_SCRIPT" ]; then
+    log_pass "Wrapper script is executable"
+else
+    log_fail "Wrapper script not found or not executable"
+fi
+
+# Test 2: Wrapper script has correct shebang
+log_test "Wrapper script has correct shebang"
+SHEBANG=$(head -1 "$WRAPPER_SCRIPT")
+if [ "$SHEBANG" = "#!/bin/bash" ]; then
+    log_pass "Correct shebang"
+else
+    log_fail "Incorrect shebang: $SHEBANG"
+fi
+
+# Test 3: Exponential backoff calculation
+log_test "Exponential backoff calculation"
+python3 << 'EOF'
+import os
+
+BASE_WAIT = 60
+MAX_WAIT = 3600
+
+def calculate_wait(retry):
+    wait_time = BASE_WAIT * (2 ** retry)
+    # Add jitter would be random, just test base calculation
+    if wait_time > MAX_WAIT:
+        wait_time = MAX_WAIT
+    return wait_time
+
+# Test exponential growth
+assert calculate_wait(0) == 60, f"Retry 0: expected 60, got {calculate_wait(0)}"
+assert calculate_wait(1) == 120, f"Retry 1: expected 120, got {calculate_wait(1)}"
+assert calculate_wait(2) == 240, f"Retry 2: expected 240, got {calculate_wait(2)}"
+assert calculate_wait(3) == 480, f"Retry 3: expected 480, got {calculate_wait(3)}"
+assert calculate_wait(4) == 960, f"Retry 4: expected 960, got {calculate_wait(4)}"
+assert calculate_wait(5) == 1920, f"Retry 5: expected 1920, got {calculate_wait(5)}"
+
+# Test max cap
+assert calculate_wait(6) == 3600, f"Retry 6: expected 3600 (capped), got {calculate_wait(6)}"
+assert calculate_wait(10) == 3600, f"Retry 10: expected 3600 (capped), got {calculate_wait(10)}"
+
+print("VERIFIED")
+EOF
+
+if [ $? -eq 0 ]; then
+    log_pass "Exponential backoff calculation works"
+else
+    log_fail "Exponential backoff calculation failed"
+fi
+
+# Test 4: State file JSON structure
+log_test "State file JSON structure"
+python3 << 'EOF'
+import json
+from datetime import datetime
+
+# Simulate wrapper state
+state = {
+    "retryCount": 3,
+    "status": "running",
+    "lastExitCode": 0,
+    "lastRun": datetime.utcnow().isoformat() + 'Z',
+    "prdPath": "./docs/requirements.md",
+    "pid": 12345
+}
+
+# Verify JSON serialization
+json_str = json.dumps(state)
+parsed = json.loads(json_str)
+
+assert parsed["retryCount"] == 3
+assert parsed["status"] == "running"
+assert parsed["pid"] == 12345
+print("VERIFIED")
+EOF
+
+if [ $? -eq 0 ]; then
+    log_pass "State file JSON structure is valid"
+else
+    log_fail "State file JSON structure failed"
+fi
+
+# Test 5: Completion detection logic
+log_test "Completion detection logic"
+mkdir -p "$TEST_DIR/.loki/state"
+cat > "$TEST_DIR/.loki/state/orchestrator.json" << 'EOF'
+{
+    "currentPhase": "COMPLETED",
+    "startedAt": "2025-01-15T10:00:00Z",
+    "completedAt": "2025-01-15T12:00:00Z"
+}
+EOF
+
+python3 << EOF
+import json
+
+with open("$TEST_DIR/.loki/state/orchestrator.json") as f:
+    state = json.load(f)
+
+phase = state.get("currentPhase", "")
+is_completed = phase == "COMPLETED"
+assert is_completed, f"Expected COMPLETED, got {phase}"
+print("VERIFIED")
+EOF
+
+if [ $? -eq 0 ]; then
+    log_pass "Completion detection works"
+else
+    log_fail "Completion detection failed"
+fi
+
+# Test 6: PRD path validation
+log_test "PRD path validation"
+touch "$TEST_DIR/test-prd.md"
+if [ -f "$TEST_DIR/test-prd.md" ]; then
+    log_pass "PRD path validation works"
+else
+    log_fail "PRD path validation failed"
+fi
+
+# Test 7: Resume prompt generation
+log_test "Resume prompt generation"
+python3 << 'EOF'
+def build_resume_prompt(retry, prd_path=None, initial_prompt="Loki Mode"):
+    if retry == 0:
+        return initial_prompt
+    else:
+        if prd_path:
+            return f"Loki Mode - Resume from checkpoint. PRD at {prd_path}. This is retry #{retry} after rate limit. Check .loki/state/ for current progress and continue from where we left off."
+        else:
+            return f"Loki Mode - Resume from checkpoint. This is retry #{retry} after rate limit. Check .loki/state/ for current progress and continue from where we left off."
+
+# Test initial prompt
+assert build_resume_prompt(0) == "Loki Mode"
+
+# Test resume prompt without PRD
+resume = build_resume_prompt(3)
+assert "Resume from checkpoint" in resume
+assert "retry #3" in resume
+assert ".loki/state/" in resume
+
+# Test resume prompt with PRD
+resume = build_resume_prompt(5, "./docs/req.md")
+assert "PRD at ./docs/req.md" in resume
+assert "retry #5" in resume
+
+print("VERIFIED")
+EOF
+
+if [ $? -eq 0 ]; then
+    log_pass "Resume prompt generation works"
+else
+    log_fail "Resume prompt generation failed"
+fi
+
+# Test 8: Rate limit detection logic
+log_test "Rate limit detection logic"
+python3 << 'EOF'
+def is_rate_limit(exit_code, log_content=""):
+    # Any non-zero exit is treated as potential rate limit
+    if exit_code != 0:
+        # Could check logs for specific indicators
+        rate_limit_indicators = ["rate limit", "429", "too many requests", "quota exceeded"]
+        for indicator in rate_limit_indicators:
+            if indicator.lower() in log_content.lower():
+                return True
+        # Conservative: treat any non-zero as rate limit
+        return True
+    return False
+
+# Test cases
+assert is_rate_limit(0) == False, "Exit 0 should not be rate limit"
+assert is_rate_limit(1) == True, "Exit 1 should be treated as rate limit"
+assert is_rate_limit(1, "Error: Rate limit exceeded") == True
+assert is_rate_limit(1, "HTTP 429 Too Many Requests") == True
+assert is_rate_limit(0, "Rate limit in logs but exit 0") == False
+
+print("VERIFIED")
+EOF
+
+if [ $? -eq 0 ]; then
+    log_pass "Rate limit detection logic works"
+else
+    log_fail "Rate limit detection logic failed"
+fi
+
+# Test 9: Log file creation
+log_test "Log file and directory creation"
+mkdir -p "$TEST_DIR/.loki"
+LOG_FILE="$TEST_DIR/.loki/wrapper.log"
+echo "[2025-01-15 10:00:00] [INFO] Test log entry" >> "$LOG_FILE"
+
+if [ -f "$LOG_FILE" ] && grep -q "Test log entry" "$LOG_FILE"; then
+    log_pass "Log file creation works"
+else
+    log_fail "Log file creation failed"
+fi
+
+# Test 10: COMPLETED file marker detection
+log_test "COMPLETED file marker detection"
+touch "$TEST_DIR/.loki/COMPLETED"
+if [ -f "$TEST_DIR/.loki/COMPLETED" ]; then
+    log_pass "COMPLETED file marker detection works"
+else
+    log_fail "COMPLETED file marker detection failed"
+fi
+
+# Test 11: Environment variable defaults
+log_test "Environment variable defaults"
+python3 << 'EOF'
+import os
+
+# Simulate reading with defaults
+MAX_RETRIES = int(os.environ.get('LOKI_MAX_RETRIES', '50'))
+BASE_WAIT = int(os.environ.get('LOKI_BASE_WAIT', '60'))
+MAX_WAIT = int(os.environ.get('LOKI_MAX_WAIT', '3600'))
+
+assert MAX_RETRIES == 50, f"Expected 50, got {MAX_RETRIES}"
+assert BASE_WAIT == 60, f"Expected 60, got {BASE_WAIT}"
+assert MAX_WAIT == 3600, f"Expected 3600, got {MAX_WAIT}"
+
+print("VERIFIED")
+EOF
+
+if [ $? -eq 0 ]; then
+    log_pass "Environment variable defaults work"
+else
+    log_fail "Environment variable defaults failed"
+fi
+
+# Test 12: Wrapper state loading
+log_test "Wrapper state loading and saving"
+STATE_FILE="$TEST_DIR/.loki/wrapper-state.json"
+cat > "$STATE_FILE" << 'EOF'
+{
+    "retryCount": 7,
+    "status": "running",
+    "lastExitCode": 1,
+    "lastRun": "2025-01-15T10:30:00Z",
+    "prdPath": "./test.md",
+    "pid": 99999
+}
+EOF
+
+python3 << EOF
+import json
+
+with open("$STATE_FILE") as f:
+    state = json.load(f)
+
+assert state["retryCount"] == 7
+assert state["status"] == "running"
+assert state["lastExitCode"] == 1
+print("VERIFIED")
+EOF
+
+if [ $? -eq 0 ]; then
+    log_pass "Wrapper state loading works"
+else
+    log_fail "Wrapper state loading failed"
+fi
+
+echo ""
+echo "=========================================="
+echo "Test Summary"
+echo "=========================================="
+echo -e "${GREEN}Passed: $PASSED${NC}"
+echo -e "${RED}Failed: $FAILED${NC}"
+echo ""
+
+if [ $FAILED -eq 0 ]; then
+    echo -e "${GREEN}All tests passed!${NC}"
+    exit 0
+else
+    echo -e "${RED}Some tests failed!${NC}"
+    exit 1
+fi