57 lines
1.9 KiB
Python
57 lines
1.9 KiB
Python
#!/usr/bin/env python3
|
|
"""
|
|
Simple test script for file upload API
|
|
"""
|
|
import requests
|
|
import os
|
|
from pathlib import Path
|
|
|
|
def test_upload_api():
|
|
"""Test the file upload API"""
|
|
base_url = "http://localhost:5000/api"
|
|
|
|
# Test health check
|
|
print("Testing health check...")
|
|
try:
|
|
response = requests.get(f"{base_url}/health")
|
|
print(f"Health check: {response.status_code} - {response.json()}")
|
|
except Exception as e:
|
|
print(f"Health check failed: {e}")
|
|
return
|
|
|
|
# Test file upload with blade.step
|
|
blade_file = Path("resource/blade.step")
|
|
if blade_file.exists():
|
|
print(f"\nTesting file upload with {blade_file}...")
|
|
try:
|
|
with open(blade_file, 'rb') as f:
|
|
files = {'file': (blade_file.name, f, 'application/octet-stream')}
|
|
response = requests.post(f"{base_url}/upload", files=files)
|
|
print(f"Upload response: {response.status_code}")
|
|
print(f"Response data: {response.json()}")
|
|
except Exception as e:
|
|
print(f"Upload test failed: {e}")
|
|
else:
|
|
print(f"Test file {blade_file} not found")
|
|
|
|
# Test get current file
|
|
print("\nTesting get current file...")
|
|
try:
|
|
response = requests.get(f"{base_url}/files/current")
|
|
print(f"Current file: {response.status_code} - {response.json()}")
|
|
except Exception as e:
|
|
print(f"Get current file failed: {e}")
|
|
|
|
# Test mesh status
|
|
print("\nTesting mesh status...")
|
|
try:
|
|
response = requests.get(f"{base_url}/mesh/status")
|
|
print(f"Mesh status: {response.status_code} - {response.json()}")
|
|
except Exception as e:
|
|
print(f"Mesh status failed: {e}")
|
|
|
|
if __name__ == '__main__':
|
|
print("CAE Mesh Generator API Test")
|
|
print("Make sure the server is running on http://localhost:5000")
|
|
print("=" * 50)
|
|
test_upload_api() |