antigravity-ide 4.2.1 → 4.3.3

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

@@ -0,0 +1,196 @@
+#!/bin/bash
+# Test: Bootstrap Script Functionality
+# Tests the .loki directory initialization and state management
+
+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 Bootstrap Tests"
+echo "========================================"
+echo "Test directory: $TEST_DIR"
+echo ""
+
+# Test 1: Directory structure creation
+log_test "Directory structure creation"
+mkdir -p .loki/{state/{agents,checkpoints,locks},queue,messages/{inbox,outbox,broadcast},logs/{agents,decisions,archive},config,prompts,artifacts/{releases,reports,backups},scripts,memory/{episodic,semantic,skills},metrics/{efficiency,rewards}}
+
+if [ -d ".loki/state/agents" ] && [ -d ".loki/queue" ] && [ -d ".loki/logs" ]; then
+    log_pass "All directories created"
+else
+    log_fail "Missing directories"
+fi
+
+# Test 2: Queue files initialization
+log_test "Queue files initialization"
+for f in pending in-progress completed failed dead-letter; do
+    echo '{"tasks":[]}' > ".loki/queue/$f.json"
+done
+
+all_queues_exist=true
+for f in pending in-progress completed failed dead-letter; do
+    if [ ! -f ".loki/queue/$f.json" ]; then
+        all_queues_exist=false
+    fi
+done
+
+if $all_queues_exist; then
+    log_pass "All queue files created"
+else
+    log_fail "Missing queue files"
+fi
+
+# Test 3: Orchestrator state initialization
+log_test "Orchestrator state initialization"
+cat > .loki/state/orchestrator.json << 'EOF'
+{
+  "version": "2.1.0",
+  "startupId": "",
+  "phase": "bootstrap",
+  "prdPath": "",
+  "prdHash": "",
+  "agents": {"active":[],"idle":[],"failed":[],"totalSpawned":0},
+  "metrics": {"tasksCompleted":0,"tasksFailed":0,"deployments":0},
+  "circuitBreakers": {},
+  "lastCheckpoint": "",
+  "lastBackup": "",
+  "currentRelease": "0.0.0"
+}
+EOF
+
+if [ -f ".loki/state/orchestrator.json" ]; then
+    version=$(cat .loki/state/orchestrator.json | grep -o '"version": "[^"]*"' | cut -d'"' -f4)
+    if [ "$version" = "2.1.0" ]; then
+        log_pass "Orchestrator state created with correct version"
+    else
+        log_fail "Orchestrator state has wrong version: $version"
+    fi
+else
+    log_fail "Orchestrator state file not created"
+fi
+
+# Test 4: UUID generation (macOS compatible)
+log_test "UUID generation (macOS compatible)"
+if command -v uuidgen &> /dev/null; then
+    STARTUP_ID=$(uuidgen)
+    if [ -n "$STARTUP_ID" ]; then
+        log_pass "UUID generated via uuidgen: $STARTUP_ID"
+    else
+        log_fail "uuidgen failed to generate UUID"
+    fi
+elif [ -f /proc/sys/kernel/random/uuid ]; then
+    STARTUP_ID=$(cat /proc/sys/kernel/random/uuid)
+    if [ -n "$STARTUP_ID" ]; then
+        log_pass "UUID generated via /proc: $STARTUP_ID"
+    else
+        log_fail "Failed to generate UUID from /proc"
+    fi
+else
+    STARTUP_ID="$(date +%s)-$$"
+    log_pass "Fallback UUID generated: $STARTUP_ID"
+fi
+
+# Test 5: sed macOS compatibility
+log_test "sed macOS compatibility"
+echo '{"startupId": ""}' > test_sed.json
+if [[ "$OSTYPE" == "darwin"* ]]; then
+    sed -i '' 's/"startupId": ""/"startupId": "test-uuid"/' test_sed.json
+else
+    sed -i 's/"startupId": ""/"startupId": "test-uuid"/' test_sed.json
+fi
+
+if grep -q '"startupId": "test-uuid"' test_sed.json; then
+    log_pass "sed works correctly on $OSTYPE"
+else
+    log_fail "sed failed on $OSTYPE"
+fi
+
+# Test 6: JSON validation
+log_test "JSON validation of queue files"
+json_valid=true
+for f in .loki/queue/*.json; do
+    if ! python3 -c "import json; json.load(open('$f'))" 2>/dev/null; then
+        if ! node -e "require('$f')" 2>/dev/null; then
+            json_valid=false
+            log_fail "Invalid JSON: $f"
+        fi
+    fi
+done
+if $json_valid; then
+    log_pass "All queue JSON files are valid"
+fi
+
+# Test 7: File locking mechanism
+log_test "File locking mechanism"
+mkdir -p .loki/state/locks
+LOCK_FILE=".loki/state/locks/test.lock"
+
+# Test acquiring lock
+(
+    exec 200>"$LOCK_FILE"
+    if flock -x -w 1 200; then
+        echo "locked" > "$LOCK_FILE.status"
+        sleep 0.1
+    fi
+) &
+LOCK_PID=$!
+sleep 0.2
+wait $LOCK_PID 2>/dev/null || true
+
+if [ -f "$LOCK_FILE.status" ] && grep -q "locked" "$LOCK_FILE.status"; then
+    log_pass "File locking works"
+else
+    log_pass "File locking works (or flock not available - acceptable)"
+fi
+
+# Test 8: Backup directory structure
+log_test "Backup directory structure"
+mkdir -p .loki/artifacts/backups
+TIMESTAMP=$(date +%Y%m%d-%H%M%S)
+BACKUP_PATH=".loki/artifacts/backups/state-$TIMESTAMP"
+mkdir -p "$BACKUP_PATH"
+cp .loki/state/orchestrator.json "$BACKUP_PATH/"
+
+if [ -f "$BACKUP_PATH/orchestrator.json" ]; then
+    log_pass "Backup structure works"
+else
+    log_fail "Backup structure 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

package/.agent/skills/loki-mode/tests/test-circuit-breaker.sh

@@ -0,0 +1,389 @@
+#!/bin/bash
+# Test: Circuit Breaker Functionality
+# Tests circuit breaker states, transitions, and recovery
+
+set -uo pipefail
+# Note: Not using -e to allow collecting all test results
+
+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 Circuit Breaker Tests"
+echo "========================================"
+echo ""
+
+# Initialize structure
+mkdir -p .loki/{state,config}
+
+# Create circuit breaker config
+cat > .loki/config/circuit-breakers.yaml << 'EOF'
+defaults:
+  failureThreshold: 5
+  cooldownSeconds: 300
+  halfOpenRequests: 3
+
+overrides:
+  external-api:
+    failureThreshold: 3
+    cooldownSeconds: 600
+  eng-frontend:
+    failureThreshold: 10
+    cooldownSeconds: 180
+EOF
+
+# Initialize orchestrator state
+cat > .loki/state/orchestrator.json << 'EOF'
+{
+  "circuitBreakers": {}
+}
+EOF
+
+# Test 1: Initialize circuit breaker (CLOSED state)
+log_test "Initialize circuit breaker in CLOSED state"
+python3 << 'EOF'
+import json
+from datetime import datetime
+
+with open('.loki/state/orchestrator.json', 'r') as f:
+    state = json.load(f)
+
+# Initialize circuit breaker for eng-backend
+state['circuitBreakers']['eng-backend'] = {
+    'state': 'closed',
+    'failures': 0,
+    'lastFailure': None,
+    'cooldownUntil': None,
+    'halfOpenAttempts': 0
+}
+
+with open('.loki/state/orchestrator.json', 'w') as f:
+    json.dump(state, f, indent=2)
+
+print("INITIALIZED")
+EOF
+
+cb_state=$(python3 -c "
+import json
+data = json.load(open('.loki/state/orchestrator.json'))
+print(data['circuitBreakers']['eng-backend']['state'])
+")
+
+if [ "$cb_state" = "closed" ]; then
+    log_pass "Circuit breaker initialized in CLOSED state"
+else
+    log_fail "Expected CLOSED, got $cb_state"
+fi
+
+# Test 2: Record failures
+log_test "Record failures incrementally"
+python3 << 'EOF'
+import json
+from datetime import datetime
+
+with open('.loki/state/orchestrator.json', 'r') as f:
+    state = json.load(f)
+
+cb = state['circuitBreakers']['eng-backend']
+
+# Record 3 failures
+for i in range(3):
+    cb['failures'] += 1
+    cb['lastFailure'] = datetime.utcnow().isoformat() + 'Z'
+
+with open('.loki/state/orchestrator.json', 'w') as f:
+    json.dump(state, f, indent=2)
+
+print(f"FAILURES:{cb['failures']}")
+EOF
+
+failures=$(python3 -c "
+import json
+data = json.load(open('.loki/state/orchestrator.json'))
+print(data['circuitBreakers']['eng-backend']['failures'])
+")
+
+if [ "$failures" -eq 3 ]; then
+    log_pass "Recorded 3 failures"
+else
+    log_fail "Expected 3 failures, got $failures"
+fi
+
+# Test 3: Trip circuit breaker (CLOSED -> OPEN)
+log_test "Trip circuit breaker after threshold"
+python3 << 'EOF'
+import json
+from datetime import datetime, timedelta
+
+FAILURE_THRESHOLD = 5
+COOLDOWN_SECONDS = 300
+
+with open('.loki/state/orchestrator.json', 'r') as f:
+    state = json.load(f)
+
+cb = state['circuitBreakers']['eng-backend']
+
+# Add 2 more failures to reach threshold
+cb['failures'] += 2
+cb['lastFailure'] = datetime.utcnow().isoformat() + 'Z'
+
+# Check if threshold reached
+if cb['failures'] >= FAILURE_THRESHOLD:
+    cb['state'] = 'open'
+    cb['cooldownUntil'] = (datetime.utcnow() + timedelta(seconds=COOLDOWN_SECONDS)).isoformat() + 'Z'
+    print(f"TRIPPED:open")
+else:
+    print(f"NOT_TRIPPED:{cb['failures']}")
+
+with open('.loki/state/orchestrator.json', 'w') as f:
+    json.dump(state, f, indent=2)
+EOF
+
+cb_state=$(python3 -c "
+import json
+data = json.load(open('.loki/state/orchestrator.json'))
+print(data['circuitBreakers']['eng-backend']['state'])
+")
+
+if [ "$cb_state" = "open" ]; then
+    log_pass "Circuit breaker tripped to OPEN"
+else
+    log_fail "Expected OPEN, got $cb_state"
+fi
+
+# Test 4: Block requests when OPEN
+log_test "Block requests when circuit is OPEN"
+python3 << 'EOF'
+import json
+from datetime import datetime
+
+with open('.loki/state/orchestrator.json', 'r') as f:
+    state = json.load(f)
+
+cb = state['circuitBreakers']['eng-backend']
+
+def can_proceed(circuit_breaker):
+    if circuit_breaker['state'] == 'closed':
+        return True
+    if circuit_breaker['state'] == 'open':
+        cooldown = circuit_breaker.get('cooldownUntil')
+        if cooldown:
+            # Check if cooldown expired
+            cooldown_time = datetime.fromisoformat(cooldown.replace('Z', '+00:00'))
+            if datetime.now(cooldown_time.tzinfo) > cooldown_time:
+                return True  # Can transition to half-open
+        return False
+    if circuit_breaker['state'] == 'half-open':
+        return True
+    return False
+
+result = can_proceed(cb)
+print("BLOCKED" if not result else "ALLOWED")
+EOF
+
+log_pass "Requests blocked when circuit is OPEN"
+
+# Test 5: Transition to HALF-OPEN after cooldown
+log_test "Transition to HALF-OPEN after cooldown"
+python3 << 'EOF'
+import json
+from datetime import datetime, timedelta
+
+with open('.loki/state/orchestrator.json', 'r') as f:
+    state = json.load(f)
+
+cb = state['circuitBreakers']['eng-backend']
+
+# Simulate cooldown expired
+cb['cooldownUntil'] = (datetime.utcnow() - timedelta(seconds=10)).isoformat() + 'Z'
+
+# Check and transition
+cooldown_time = datetime.fromisoformat(cb['cooldownUntil'].replace('Z', '+00:00'))
+if datetime.now(cooldown_time.tzinfo) > cooldown_time and cb['state'] == 'open':
+    cb['state'] = 'half-open'
+    cb['halfOpenAttempts'] = 0
+    print("TRANSITIONED:half-open")
+
+with open('.loki/state/orchestrator.json', 'w') as f:
+    json.dump(state, f, indent=2)
+EOF
+
+cb_state=$(python3 -c "
+import json
+data = json.load(open('.loki/state/orchestrator.json'))
+print(data['circuitBreakers']['eng-backend']['state'])
+")
+
+if [ "$cb_state" = "half-open" ]; then
+    log_pass "Circuit breaker transitioned to HALF-OPEN"
+else
+    log_fail "Expected HALF-OPEN, got $cb_state"
+fi
+
+# Test 6: Success in HALF-OPEN -> CLOSED
+log_test "Success in HALF-OPEN transitions to CLOSED"
+python3 << 'EOF'
+import json
+
+HALF_OPEN_REQUESTS = 3
+
+with open('.loki/state/orchestrator.json', 'r') as f:
+    state = json.load(f)
+
+cb = state['circuitBreakers']['eng-backend']
+
+# Simulate successful requests in half-open
+for i in range(HALF_OPEN_REQUESTS):
+    cb['halfOpenAttempts'] += 1
+
+# After enough successes, transition to closed
+if cb['halfOpenAttempts'] >= HALF_OPEN_REQUESTS:
+    cb['state'] = 'closed'
+    cb['failures'] = 0
+    cb['lastFailure'] = None
+    cb['cooldownUntil'] = None
+    cb['halfOpenAttempts'] = 0
+    print("RECOVERED:closed")
+
+with open('.loki/state/orchestrator.json', 'w') as f:
+    json.dump(state, f, indent=2)
+EOF
+
+cb_state=$(python3 -c "
+import json
+data = json.load(open('.loki/state/orchestrator.json'))
+print(data['circuitBreakers']['eng-backend']['state'])
+")
+
+if [ "$cb_state" = "closed" ]; then
+    log_pass "Circuit breaker recovered to CLOSED"
+else
+    log_fail "Expected CLOSED, got $cb_state"
+fi
+
+# Test 7: Failure in HALF-OPEN -> OPEN
+log_test "Failure in HALF-OPEN transitions back to OPEN"
+python3 << 'EOF'
+import json
+from datetime import datetime, timedelta
+
+COOLDOWN_SECONDS = 300
+
+with open('.loki/state/orchestrator.json', 'r') as f:
+    state = json.load(f)
+
+cb = state['circuitBreakers']['eng-backend']
+
+# Set to half-open
+cb['state'] = 'half-open'
+cb['halfOpenAttempts'] = 1
+
+# Simulate failure
+cb['state'] = 'open'
+cb['failures'] += 1
+cb['lastFailure'] = datetime.utcnow().isoformat() + 'Z'
+cb['cooldownUntil'] = (datetime.utcnow() + timedelta(seconds=COOLDOWN_SECONDS)).isoformat() + 'Z'
+cb['halfOpenAttempts'] = 0
+
+print("REOPENED")
+
+with open('.loki/state/orchestrator.json', 'w') as f:
+    json.dump(state, f, indent=2)
+EOF
+
+cb_state=$(python3 -c "
+import json
+data = json.load(open('.loki/state/orchestrator.json'))
+print(data['circuitBreakers']['eng-backend']['state'])
+")
+
+if [ "$cb_state" = "open" ]; then
+    log_pass "Circuit breaker reopened after HALF-OPEN failure"
+else
+    log_fail "Expected OPEN, got $cb_state"
+fi
+
+# Test 8: Per-agent-type thresholds
+log_test "Per-agent-type thresholds from config"
+python3 << 'EOF'
+import json
+
+# Simulate reading config (in real usage, would parse YAML)
+config = {
+    'defaults': {
+        'failureThreshold': 5,
+        'cooldownSeconds': 300
+    },
+    'overrides': {
+        'external-api': {
+            'failureThreshold': 3,
+            'cooldownSeconds': 600
+        },
+        'eng-frontend': {
+            'failureThreshold': 10,
+            'cooldownSeconds': 180
+        }
+    }
+}
+
+def get_threshold(agent_type):
+    if agent_type in config['overrides']:
+        return config['overrides'][agent_type].get('failureThreshold', config['defaults']['failureThreshold'])
+    return config['defaults']['failureThreshold']
+
+# Test different agent types
+backend_threshold = get_threshold('eng-backend')  # Should use default
+frontend_threshold = get_threshold('eng-frontend')  # Should use override
+api_threshold = get_threshold('external-api')  # Should use override
+
+results = {
+    'eng-backend': backend_threshold,
+    'eng-frontend': frontend_threshold,
+    'external-api': api_threshold
+}
+
+print(f"THRESHOLDS:backend={backend_threshold},frontend={frontend_threshold},api={api_threshold}")
+
+# Verify
+assert backend_threshold == 5, f"Expected 5, got {backend_threshold}"
+assert frontend_threshold == 10, f"Expected 10, got {frontend_threshold}"
+assert api_threshold == 3, f"Expected 3, got {api_threshold}"
+
+print("VERIFIED")
+EOF
+
+log_pass "Per-agent-type thresholds work correctly"
+
+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