ㅔㅛ#!/bin/bash

# MuscleCare API 테스트 스크립트

# 서버 URL 설정
# 로컬: http://localhost:7860
# 배포: https://[username]-musclecare-api.hf.space
BASE_URL="${1:-http://localhost:7860}"

echo "======================================"
echo "🧪 MuscleCare API 테스트"
echo "======================================"
echo "서버: $BASE_URL"
echo ""

# 1. Health 체크
echo "1️⃣ Health 체크"
echo "--------------------------------------"
curl -X GET "$BASE_URL/health" \
  -H "accept: application/json" | jq .
echo ""
echo ""

# 2. 사용자 상태 업로드
echo "2️⃣ 사용자 상태 업로드"
echo "--------------------------------------"
curl -X POST "$BASE_URL/upload_state" \
  -H "Content-Type: application/json" \
  -H "accept: application/json" \
  -d '{
    "user_id": "test_user_001",
    "rms_base": 0.11,
    "freq_base": 50.1,
    "user_emb": [0.135, 0.012, 49.3, 0.8, 0.84, 0.02, 0.11, 0.0021, 0.35, 0.71, 0.92, 0.88],
    "model_version": "v1.0.0"
  }' | jq .
echo ""
echo ""

# 3. Dataset 업로드
echo "3️⃣ Dataset 업로드"
echo "--------------------------------------"
curl -X POST "$BASE_URL/upload_dataset" \
  -H "Content-Type: application/json" \
  -H "accept: application/json" \
  -d '{
    "user_id": "test_user_001",
    "rms": 0.42,
    "freq": 45.3,
    "fatigue": 0.25,
    "timestamp": "2024-01-15T10:30:00Z"
  }' | jq .
echo ""
echo ""

# 4. 사용자 Dataset 조회
echo "4️⃣ 사용자 Dataset 조회"
echo "--------------------------------------"
curl -X GET "$BASE_URL/user_dataset/test_user_001" \
  -H "accept: application/json" | jq .
echo ""
echo ""

echo "======================================"
echo "✅ 테스트 완료"
echo "======================================"