File size: 996 Bytes
55a76a3 | 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 | import requests
import time
import sys
BASE_URL = "http://localhost:8000"
def wait_for_server():
print("Waiting for server to start...")
for _ in range(30):
try:
res = requests.get(f"{BASE_URL}/health")
if res.status_code == 200:
print("Server is up!")
return True
except requests.exceptions.ConnectionError:
pass
time.sleep(1)
print("Server failed to start within 30 seconds.")
return False
def test_endpoints():
print("Testing /health...")
res = requests.get(f"{BASE_URL}/health")
assert res.status_code == 200, f"Health check failed: {res.text}"
print("Testing /documents (empty list expected)...")
res = requests.get(f"{BASE_URL}/documents")
assert res.status_code == 200, f"List documents failed: {res.text}"
print("Smoke tests passed successfully!")
if __name__ == "__main__":
if not wait_for_server():
sys.exit(1)
test_endpoints()
|