Spaces:
Running
Running
| # 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}" |