Spaces:
Running
Running
File size: 4,814 Bytes
02eac4b bf6e620 02eac4b |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 148 149 150 151 152 153 154 155 156 157 158 159 160 161 |
#!/bin/bash
# Test script for LeRobot Arena Transport Server Docker setup
set -e
echo "🤖 Testing LeRobot Arena Transport Server Docker Setup"
echo "=================================================="
# Colors for output
RED='\033[0;31m'
GREEN='\033[0;32m'
YELLOW='\033[1;33m'
NC='\033[0m' # No Color
# Configuration
CONTAINER_NAME="lerobot-arena-transport-test"
PORT=7860
MAX_WAIT=60
# Cleanup function
cleanup() {
echo -e "\n${YELLOW}Cleaning up...${NC}"
docker stop $CONTAINER_NAME 2>/dev/null || true
docker rm $CONTAINER_NAME 2>/dev/null || true
}
# Set trap to cleanup on exit
trap cleanup EXIT
# Step 1: Build the Docker image
echo -e "\n${YELLOW}Step 1: Building Docker image...${NC}"
docker build -t lerobot-arena-transport . || {
echo -e "${RED}❌ Failed to build Docker image${NC}"
exit 1
}
echo -e "${GREEN}✅ Docker image built successfully${NC}"
# Step 2: Start the container
echo -e "\n${YELLOW}Step 2: Starting container...${NC}"
docker run -d \
--name $CONTAINER_NAME \
-p $PORT:7860 \
-e SERVE_FRONTEND=true \
lerobot-arena-transport || {
echo -e "${RED}❌ Failed to start container${NC}"
exit 1
}
echo -e "${GREEN}✅ Container started successfully${NC}"
# Step 3: Wait for container to be ready
echo -e "\n${YELLOW}Step 3: Waiting for container to be ready...${NC}"
COUNTER=0
while [ $COUNTER -lt $MAX_WAIT ]; do
if curl -s http://localhost:$PORT/health > /dev/null 2>&1; then
echo -e "${GREEN}✅ Container is ready!${NC}"
break
fi
echo "Waiting... ($COUNTER/$MAX_WAIT)"
sleep 2
COUNTER=$((COUNTER + 2))
done
if [ $COUNTER -ge $MAX_WAIT ]; then
echo -e "${RED}❌ Container failed to start within $MAX_WAIT seconds${NC}"
echo "Container logs:"
docker logs $CONTAINER_NAME
exit 1
fi
# Step 4: Test endpoints
echo -e "\n${YELLOW}Step 4: Testing endpoints...${NC}"
# Test health endpoint
echo "Testing /health endpoint..."
HEALTH_RESPONSE=$(curl -s http://localhost:$PORT/health)
if echo "$HEALTH_RESPONSE" | grep -q "healthy"; then
echo -e "${GREEN}✅ /health endpoint working${NC}"
else
echo -e "${RED}❌ /health endpoint failed${NC}"
echo "Response: $HEALTH_RESPONSE"
exit 1
fi
# Test API health endpoint
echo "Testing /api/health endpoint..."
API_HEALTH_RESPONSE=$(curl -s http://localhost:$PORT/api/health)
if echo "$API_HEALTH_RESPONSE" | grep -q "healthy"; then
echo -e "${GREEN}✅ /api/health endpoint working${NC}"
else
echo -e "${RED}❌ /api/health endpoint failed${NC}"
echo "Response: $API_HEALTH_RESPONSE"
exit 1
fi
# Test frontend
echo "Testing frontend..."
FRONTEND_RESPONSE=$(curl -s http://localhost:$PORT/)
if echo "$FRONTEND_RESPONSE" | grep -q "<!doctype html>" && echo "$FRONTEND_RESPONSE" | grep -q "_app/immutable"; then
echo -e "${GREEN}✅ Frontend is served correctly${NC}"
else
echo -e "${RED}❌ Frontend test failed${NC}"
echo "Response preview: $(echo "$FRONTEND_RESPONSE" | head -5)"
exit 1
fi
# Test API documentation
echo "Testing API documentation..."
DOCS_RESPONSE=$(curl -s http://localhost:$PORT/api/docs)
if echo "$DOCS_RESPONSE" | grep -q "swagger"; then
echo -e "${GREEN}✅ API documentation accessible${NC}"
else
echo -e "${RED}❌ API documentation test failed${NC}"
exit 1
fi
# Test robotics API
echo "Testing robotics API..."
ROBOTICS_RESPONSE=$(curl -s http://localhost:$PORT/api/robotics/rooms)
if [ $? -eq 0 ]; then
echo -e "${GREEN}✅ Robotics API accessible${NC}"
else
echo -e "${RED}❌ Robotics API test failed${NC}"
exit 1
fi
# Test video API
echo "Testing video API..."
VIDEO_RESPONSE=$(curl -s http://localhost:$PORT/api/video/rooms)
if [ $? -eq 0 ]; then
echo -e "${GREEN}✅ Video API accessible${NC}"
else
echo -e "${RED}❌ Video API test failed${NC}"
exit 1
fi
# Step 5: Show container information
echo -e "\n${YELLOW}Step 5: Container information${NC}"
echo "Container status:"
docker ps | grep $CONTAINER_NAME
echo -e "\nContainer resource usage:"
docker stats --no-stream $CONTAINER_NAME
# Step 6: Success summary
echo -e "\n${GREEN}🎉 All tests passed successfully!${NC}"
echo "=================================================="
echo "✅ Docker image builds correctly"
echo "✅ Container starts and runs"
echo "✅ Health endpoints respond correctly"
echo "✅ Frontend is served"
echo "✅ API documentation is accessible"
echo "✅ Robotics API is working"
echo "✅ Video API is working"
echo ""
echo "🌐 Access the application at: http://localhost:$PORT"
echo "📚 API docs available at: http://localhost:$PORT/api/docs"
echo ""
echo "To manually test:"
echo " docker run -p 7860:7860 -e SERVE_FRONTEND=true lerobot-arena-transport"
echo ""
echo -e "${YELLOW}Container will be cleaned up automatically.${NC}" |