4f35df1781
🤖 Generated with [Claude Code](https://claude.com/claude-code) Co-Authored-By: Claude Opus 4.5 <noreply@anthropic.com>
161 lines
4.4 KiB
Bash
Executable File
161 lines
4.4 KiB
Bash
Executable File
#!/bin/bash
|
|
# Test Dangerous Pi custom image
|
|
# Run this script after flashing and booting the image
|
|
|
|
set -e
|
|
|
|
# Configuration
|
|
PI_HOST="${1:-10.3.141.1}" # Default to AP mode IP
|
|
PI_USER="dt"
|
|
API_PORT="8000"
|
|
|
|
# Colors
|
|
GREEN='\033[0;32m'
|
|
RED='\033[0;31m'
|
|
YELLOW='\033[1;33m'
|
|
NC='\033[0m'
|
|
|
|
echo "========================================"
|
|
echo "Dangerous Pi Image Test Suite"
|
|
echo "========================================"
|
|
echo "Target: $PI_USER@$PI_HOST"
|
|
echo ""
|
|
|
|
# Function to run test
|
|
run_test() {
|
|
local test_name="$1"
|
|
local command="$2"
|
|
|
|
echo -n "Testing $test_name... "
|
|
|
|
if eval "$command" &>/dev/null; then
|
|
echo -e "${GREEN}✓ PASS${NC}"
|
|
return 0
|
|
else
|
|
echo -e "${RED}✗ FAIL${NC}"
|
|
return 1
|
|
fi
|
|
}
|
|
|
|
# Test 1: SSH Connectivity
|
|
echo "=== Connectivity Tests ==="
|
|
run_test "SSH connectivity" \
|
|
"ssh -o ConnectTimeout=5 -o StrictHostKeyChecking=no $PI_USER@$PI_HOST 'echo ok'"
|
|
|
|
# Test 2: User exists
|
|
run_test "User 'pi' exists" \
|
|
"ssh -o ConnectTimeout=5 $PI_USER@$PI_HOST 'id pi'"
|
|
|
|
# Test 3: Dangerous Pi directory
|
|
run_test "Dangerous Pi installed at /opt" \
|
|
"ssh $PI_USER@$PI_HOST '[ -d /opt/dangerous-pi ]'"
|
|
|
|
# Test 4: Required files
|
|
echo ""
|
|
echo "=== File Structure Tests ==="
|
|
run_test "Backend code exists" \
|
|
"ssh $PI_USER@$PI_HOST '[ -d /opt/dangerous-pi/app/backend ]'"
|
|
|
|
run_test "Systemd service file" \
|
|
"ssh $PI_USER@$PI_HOST '[ -f /etc/systemd/system/dangerous-pi.service ]'"
|
|
|
|
run_test "Environment file" \
|
|
"ssh $PI_USER@$PI_HOST '[ -f /opt/dangerous-pi/.env ]'"
|
|
|
|
run_test "Scripts executable" \
|
|
"ssh $PI_USER@$PI_HOST '[ -x /opt/dangerous-pi/scripts/resolve-port-conflict.sh ]'"
|
|
|
|
# Test 5: Service Status
|
|
echo ""
|
|
echo "=== Service Tests ==="
|
|
run_test "Service is enabled" \
|
|
"ssh $PI_USER@$PI_HOST 'systemctl is-enabled dangerous-pi.service'"
|
|
|
|
run_test "Service is running" \
|
|
"ssh $PI_USER@$PI_HOST 'systemctl is-active dangerous-pi.service'"
|
|
|
|
# Test 6: Hardware Access
|
|
echo ""
|
|
echo "=== Hardware Access Tests ==="
|
|
run_test "User in 'i2c' group" \
|
|
"ssh $PI_USER@$PI_HOST 'groups pi | grep -q i2c'"
|
|
|
|
run_test "User in 'dialout' group" \
|
|
"ssh $PI_USER@$PI_HOST 'groups pi | grep -q dialout'"
|
|
|
|
run_test "User in 'gpio' group" \
|
|
"ssh $PI_USER@$PI_HOST 'groups pi | grep -q gpio'"
|
|
|
|
run_test "User in 'bluetooth' group" \
|
|
"ssh $PI_USER@$PI_HOST 'groups pi | grep -q bluetooth'"
|
|
|
|
run_test "I2C interface enabled" \
|
|
"ssh $PI_USER@$PI_HOST 'grep -q \"^dtparam=i2c_arm=on\" /boot/config.txt'"
|
|
|
|
run_test "I2C device available" \
|
|
"ssh $PI_USER@$PI_HOST '[ -e /dev/i2c-1 ]'"
|
|
|
|
# Test 7: Python Dependencies
|
|
echo ""
|
|
echo "=== Python Dependencies Tests ==="
|
|
run_test "FastAPI installed" \
|
|
"ssh $PI_USER@$PI_HOST 'python3 -c \"import fastapi\"'"
|
|
|
|
run_test "Uvicorn installed" \
|
|
"ssh $PI_USER@$PI_HOST 'python3 -c \"import uvicorn\"'"
|
|
|
|
run_test "aiosqlite installed" \
|
|
"ssh $PI_USER@$PI_HOST 'python3 -c \"import aiosqlite\"'"
|
|
|
|
run_test "psutil installed" \
|
|
"ssh $PI_USER@$PI_HOST 'python3 -c \"import psutil\"'"
|
|
|
|
run_test "smbus2 installed" \
|
|
"ssh $PI_USER@$PI_HOST 'python3 -c \"import smbus2\"'"
|
|
|
|
# Test 8: API Endpoints (if service is running)
|
|
echo ""
|
|
echo "=== API Endpoint Tests ==="
|
|
|
|
if ssh $PI_USER@$PI_HOST "systemctl is-active dangerous-pi.service" &>/dev/null; then
|
|
# Wait for service to be fully up
|
|
sleep 3
|
|
|
|
run_test "Health endpoint" \
|
|
"curl -f -s http://$PI_HOST:$API_PORT/api/health"
|
|
|
|
run_test "System info endpoint" \
|
|
"curl -f -s http://$PI_HOST:$API_PORT/api/system/info"
|
|
|
|
run_test "PM3 status endpoint" \
|
|
"curl -f -s http://$PI_HOST:$API_PORT/api/pm3/status"
|
|
|
|
run_test "WiFi status endpoint" \
|
|
"curl -f -s http://$PI_HOST:$API_PORT/api/wifi/status"
|
|
|
|
run_test "Plugin list endpoint" \
|
|
"curl -f -s http://$PI_HOST:$API_PORT/api/plugins/list"
|
|
else
|
|
echo -e "${YELLOW}Service not running, skipping API tests${NC}"
|
|
fi
|
|
|
|
# Test 9: Disk Space
|
|
echo ""
|
|
echo "=== System Health Tests ==="
|
|
DISK_USAGE=$(ssh $PI_USER@$PI_HOST "df / | tail -1 | awk '{print \$5}' | sed 's/%//'")
|
|
if [ "$DISK_USAGE" -lt 80 ]; then
|
|
echo -e "Disk usage: ${GREEN}$DISK_USAGE%${NC} ✓"
|
|
else
|
|
echo -e "Disk usage: ${RED}$DISK_USAGE%${NC} (Warning: >80%)"
|
|
fi
|
|
|
|
# Test 10: Service Logs
|
|
echo ""
|
|
echo "=== Service Logs (last 10 lines) ==="
|
|
ssh $PI_USER@$PI_HOST "sudo journalctl -u dangerous-pi.service -n 10 --no-pager" || true
|
|
|
|
echo ""
|
|
echo "========================================"
|
|
echo "Test Suite Complete"
|
|
echo "========================================"
|