15427c96c0
- Change file mode from 644 to 755 for all project files - Add chain_id parameter to get_balance RPC endpoint with default "ait-devnet" - Rename Miner.extra_meta_data to extra_metadata for consistency
129 lines
3.9 KiB
Python
Executable File
129 lines
3.9 KiB
Python
Executable File
#!/usr/bin/env python3
|
|
"""
|
|
AITBC Phase 5 Performance Testing Script
|
|
Tests system performance for production deployment requirements
|
|
"""
|
|
|
|
import time
|
|
import requests
|
|
import statistics
|
|
import concurrent.futures
|
|
import sys
|
|
import os
|
|
|
|
def test_api_response_time():
|
|
"""Test API response times"""
|
|
print("⚡ Testing API Response Time...")
|
|
response_times = []
|
|
|
|
for i in range(10):
|
|
start_time = time.time()
|
|
response = requests.get('http://127.0.0.1:8000/health/live', timeout=5)
|
|
end_time = time.time()
|
|
if response.status_code == 200:
|
|
response_times.append((end_time - start_time) * 1000) # Convert to ms
|
|
|
|
if response_times:
|
|
avg_time = statistics.mean(response_times)
|
|
min_time = min(response_times)
|
|
max_time = max(response_times)
|
|
|
|
print(f"✅ API Response Time: PASSED")
|
|
print(f" Average: {avg_time:.2f}ms")
|
|
print(f" Min: {min_time:.2f}ms")
|
|
print(f" Max: {max_time:.2f}ms")
|
|
|
|
if avg_time < 200: # Target: <200ms
|
|
print(" ✅ Performance Target Met")
|
|
return True
|
|
else:
|
|
print(" ⚠️ Performance Target Not Met")
|
|
return False
|
|
return False
|
|
|
|
def test_concurrent_load():
|
|
"""Test concurrent load handling"""
|
|
print("\n🔄 Testing Concurrent Load...")
|
|
|
|
def make_request():
|
|
try:
|
|
response = requests.get('http://127.0.0.1:8000/health/live', timeout=5)
|
|
return response.status_code == 200
|
|
except:
|
|
return False
|
|
|
|
start_time = time.time()
|
|
with concurrent.futures.ThreadPoolExecutor(max_workers=10) as executor:
|
|
futures = [executor.submit(make_request) for _ in range(50)]
|
|
results = [f.result() for f in futures]
|
|
end_time = time.time()
|
|
|
|
success_rate = sum(results) / len(results) * 100
|
|
total_time = end_time - start_time
|
|
|
|
print(f"✅ Concurrent Load Testing: PASSED")
|
|
print(f" Requests: 50")
|
|
print(f" Success Rate: {success_rate:.1f}%")
|
|
print(f" Total Time: {total_time:.2f}s")
|
|
print(f" Requests/sec: {50/total_time:.1f}")
|
|
|
|
return success_rate > 95 # Target: 95%+ success rate
|
|
|
|
def test_ml_zk_performance():
|
|
"""Test ML-ZK circuit performance"""
|
|
print("\n🤖 Testing ML-ZK Circuit Performance...")
|
|
|
|
start_time = time.time()
|
|
response = requests.get('http://127.0.0.1:8000/v1/ml-zk/circuits', timeout=5)
|
|
end_time = time.time()
|
|
|
|
if response.status_code == 200:
|
|
response_time = (end_time - start_time) * 1000
|
|
circuits = response.json()['circuits']
|
|
print(f"✅ ML-ZK Circuit Performance: PASSED")
|
|
print(f" Response Time: {response_time:.2f}ms")
|
|
print(f" Circuits Returned: {len(circuits)}")
|
|
|
|
if response_time < 500: # Target: <500ms for complex endpoint
|
|
print(" ✅ Performance Target Met")
|
|
return True
|
|
else:
|
|
print(" ⚠️ Performance Target Not Met")
|
|
return False
|
|
return False
|
|
|
|
def main():
|
|
"""Run all performance tests"""
|
|
print("🚀 Phase 5.1 Performance Testing - Starting Now!")
|
|
print("=" * 60)
|
|
|
|
tests = [
|
|
test_api_response_time,
|
|
test_concurrent_load,
|
|
test_ml_zk_performance
|
|
]
|
|
|
|
results = []
|
|
for test in tests:
|
|
results.append(test())
|
|
|
|
print("\n" + "=" * 60)
|
|
print("🎯 Performance Testing Summary:")
|
|
|
|
passed = sum(results)
|
|
total = len(results)
|
|
|
|
print(f" Tests Passed: {passed}/{total}")
|
|
print(f" Success Rate: {(passed/total)*100:.1f}%")
|
|
|
|
if passed == total:
|
|
print("\n🚀 Phase 5.1 Performance Testing: COMPLETED!")
|
|
print("📋 System meets production performance requirements!")
|
|
return 0
|
|
else:
|
|
print("\n⚠️ Some performance tests failed.")
|
|
return 1
|
|
|
|
if __name__ == "__main__":
|
|
sys.exit(main())
|