[#4] 增加端到端测试

src/bind/tests/scripts/05_cleanup.sh

@@ -0,0 +1,107 @@
+#!/bin/bash
+
+# Clean up test environment and containers
+# Usage: ./05_cleanup.sh [--full]
+
+set -e
+
+SCRIPT_DIR="$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)"
+TEST_DIR="$(dirname "$SCRIPT_DIR")"
+
+# Parse command line arguments
+FULL_CLEANUP=false
+while [[ $# -gt 0 ]]; do
+    case $1 in
+        --full)
+            FULL_CLEANUP=true
+            shift
+            ;;
+        *)
+            echo "Unknown option: $1"
+            echo "Usage: $0 [--full]"
+            echo "  --full: Also remove persistent data and Docker image"
+            exit 1
+            ;;
+    esac
+done
+
+cd "$TEST_DIR"
+
+echo "=== Cleaning up BIND9 test environment ==="
+
+echo ""
+echo "Step 1: Stopping and removing containers..."
+
+# Stop and remove containers
+docker compose down -v
+
+echo "✓ Containers stopped and removed"
+
+echo ""
+echo "Step 2: Removing Docker networks..."
+
+# Clean up networks
+docker network prune -f > /dev/null 2>&1 || true
+
+echo "✓ Docker networks cleaned"
+
+if [ "$FULL_CLEANUP" = true ]; then
+    echo ""
+    echo "Step 3: Removing persistent data..."
+
+    # Remove persistent data directory
+    if [ -d "private" ]; then
+        rm -rf private
+        echo "✓ Persistent data directory removed"
+    else
+        echo "✓ No persistent data directory found"
+    fi
+
+    echo ""
+    echo "Step 4: Removing Docker image..."
+
+    # Remove the test image if it exists
+    if docker images --format "{{.Repository}}:{{.Tag}}" | grep -q "^argus-bind9:latest$"; then
+        docker rmi argus-bind9:latest
+        echo "✓ Docker image 'argus-bind9:latest' removed"
+    else
+        echo "✓ Docker image 'argus-bind9:latest' not found"
+    fi
+
+    echo ""
+    echo "Step 5: Cleaning up unused Docker resources..."
+
+    # Clean up any dangling images and unused volumes
+    docker system prune -f > /dev/null 2>&1 || true
+
+    echo "✓ Docker system cleaned"
+else
+    echo ""
+    echo "Step 3: Preserving persistent data and Docker image..."
+    echo "✓ Persistent data preserved in: private/"
+    echo "✓ Docker image 'argus-bind9:latest' preserved"
+    echo ""
+    echo "To perform full cleanup including persistent data and image, run:"
+    echo "  $0 --full"
+fi
+
+echo ""
+echo "=== Cleanup Summary ==="
+echo "✓ Containers stopped and removed"
+echo "✓ Docker networks cleaned"
+
+if [ "$FULL_CLEANUP" = true ]; then
+    echo "✓ Persistent data removed"
+    echo "✓ Docker image removed"
+    echo "✓ System resources cleaned"
+    echo ""
+    echo "Full cleanup completed! Test environment completely removed."
+else
+    echo "✓ Persistent data preserved"
+    echo "✓ Docker image preserved"
+    echo ""
+    echo "Basic cleanup completed! Run './01_start_container.sh' to restart testing."
+fi
+
+echo ""
+echo "Test environment cleanup finished."