File size: 3,073 Bytes
13fc29d | 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 | """
End-to-end smoke test for the production API.
Usage:
python scripts/smoke_test.py # test local (localhost:7860)
python scripts/smoke_test.py https://my-space.hf.space # test remote
"""
from __future__ import annotations
import json
import sys
import time
import requests
def smoke_test(base_url: str) -> bool:
"""Run smoke tests against the API. Returns True if all pass."""
base = base_url.rstrip("/")
passed = 0
failed = 0
results = []
def check(name: str, method: str, path: str, expected_keys=None, body=None):
nonlocal passed, failed
url = f"{base}{path}"
try:
if method == "GET":
resp = requests.get(url, timeout=15)
else:
resp = requests.post(url, json=body, timeout=30)
if resp.status_code == 200:
data = resp.json()
if expected_keys:
missing = [k for k in expected_keys if k not in data]
if missing:
results.append(f" WARN {name}: missing keys {missing}")
passed += 1 # partial pass
return
results.append(f" PASS {name}")
passed += 1
else:
results.append(f" FAIL {name}: HTTP {resp.status_code} — {resp.text[:100]}")
failed += 1
except Exception as exc:
results.append(f" FAIL {name}: {exc}")
failed += 1
print(f"\nSmoke testing: {base}\n{'='*60}")
check("Health check",
"GET", "/api/health",
expected_keys=["status", "uptime_seconds"])
check("Weather current",
"GET", "/api/weather/current",
expected_keys=["air_temperature_c", "ghi_w_m2"])
check("Weather history",
"GET", "/api/weather/history")
check("Sensor snapshot",
"GET", "/api/sensors/snapshot")
check("Energy current",
"GET", "/api/energy/current")
check("Energy daily",
"GET", "/api/energy/daily/2026-03-17")
check("Photosynthesis current",
"GET", "/api/photosynthesis/current")
check("Control status",
"GET", "/api/control/status")
check("Control plan",
"GET", "/api/control/plan")
check("Biology rules",
"GET", "/api/biology/rules",
expected_keys=["rules"])
check("Biology phenology",
"GET", "/api/biology/phenology")
check("Chatbot message",
"POST", "/api/chatbot/message",
expected_keys=["message"],
body={"message": "What is the current temperature?", "session_id": "smoke-test"})
# Print results
print()
for r in results:
print(r)
print(f"\n{'='*60}")
print(f"Results: {passed} passed, {failed} failed out of {passed + failed}")
return failed == 0
if __name__ == "__main__":
url = sys.argv[1] if len(sys.argv) > 1 else "http://localhost:7860"
ok = smoke_test(url)
sys.exit(0 if ok else 1)
|