feat: switch to persistent SQLite database and improve GPU booking/release handling
- Change database from in-memory to file-based SQLite at aitbc_coordinator.db - Add status="active" to GPU booking creation - Allow GPU release even when not properly booked (cleanup case) - Add error handling for missing booking attributes during refund calculation - Fix get_gpu_reviews query to use scalars() for proper result handling
This commit is contained in:
98
scripts/test_gpu_release_direct.py
Normal file
98
scripts/test_gpu_release_direct.py
Normal file
@@ -0,0 +1,98 @@
|
||||
#!/usr/bin/env python3
|
||||
"""
|
||||
Direct test of GPU release functionality
|
||||
"""
|
||||
|
||||
import sys
|
||||
import os
|
||||
sys.path.insert(0, '/home/oib/windsurf/aitbc/apps/coordinator-api/src')
|
||||
|
||||
from sqlmodel import Session, select
|
||||
from sqlalchemy import create_engine
|
||||
from app.domain.gpu_marketplace import GPURegistry, GPUBooking
|
||||
|
||||
def test_gpu_release():
|
||||
"""Test GPU release directly"""
|
||||
print("=== DIRECT GPU RELEASE TEST ===")
|
||||
|
||||
# Use the same database as coordinator
|
||||
db_path = "/home/oib/windsurf/aitbc/apps/coordinator-api/data/coordinator.db"
|
||||
engine = create_engine(f"sqlite:///{db_path}")
|
||||
|
||||
gpu_id = "gpu_c5be877c"
|
||||
|
||||
with Session(engine) as session:
|
||||
print(f"1. Checking GPU {gpu_id}...")
|
||||
gpu = session.exec(select(GPURegistry).where(GPURegistry.id == gpu_id)).first()
|
||||
|
||||
if not gpu:
|
||||
print(f"❌ GPU {gpu_id} not found")
|
||||
return False
|
||||
|
||||
print(f"✅ GPU found: {gpu.model} - Status: {gpu.status}")
|
||||
|
||||
print(f"2. Checking bookings for GPU {gpu_id}...")
|
||||
bookings = session.exec(
|
||||
select(GPUBooking).where(GPUBooking.gpu_id == gpu_id)
|
||||
).all()
|
||||
|
||||
print(f"Found {len(bookings)} bookings:")
|
||||
for booking in bookings:
|
||||
print(f" - ID: {booking.id}, Status: {booking.status}, Total Cost: {getattr(booking, 'total_cost', 'MISSING')}")
|
||||
|
||||
print(f"3. Checking active bookings...")
|
||||
active_booking = session.exec(
|
||||
select(GPUBooking).where(
|
||||
GPUBooking.gpu_id == gpu_id,
|
||||
GPUBooking.status == "active"
|
||||
)
|
||||
).first()
|
||||
|
||||
if active_booking:
|
||||
print(f"✅ Active booking found: {active_booking.id}")
|
||||
print(f" Total Cost: {getattr(active_booking, 'total_cost', 'MISSING')}")
|
||||
|
||||
# Test refund calculation
|
||||
try:
|
||||
refund = active_booking.total_cost * 0.5
|
||||
print(f"✅ Refund calculation successful: {refund}")
|
||||
except AttributeError as e:
|
||||
print(f"❌ Refund calculation failed: {e}")
|
||||
return False
|
||||
else:
|
||||
print("❌ No active booking found")
|
||||
|
||||
print(f"4. Testing release logic...")
|
||||
if active_booking:
|
||||
try:
|
||||
refund = active_booking.total_cost * 0.5
|
||||
active_booking.status = "cancelled"
|
||||
gpu.status = "available"
|
||||
session.commit()
|
||||
|
||||
print(f"✅ Release successful")
|
||||
print(f" GPU Status: {gpu.status}")
|
||||
print(f" Booking Status: {active_booking.status}")
|
||||
print(f" Refund: {refund}")
|
||||
|
||||
return True
|
||||
|
||||
except Exception as e:
|
||||
print(f"❌ Release failed: {e}")
|
||||
session.rollback()
|
||||
return False
|
||||
else:
|
||||
print("⚠️ No active booking to release")
|
||||
# Still try to make GPU available
|
||||
gpu.status = "available"
|
||||
session.commit()
|
||||
print(f"✅ GPU marked as available")
|
||||
return True
|
||||
|
||||
if __name__ == "__main__":
|
||||
success = test_gpu_release()
|
||||
if success:
|
||||
print("\n🎉 GPU release test PASSED!")
|
||||
else:
|
||||
print("\n❌ GPU release test FAILED!")
|
||||
sys.exit(1)
|
||||
Reference in New Issue
Block a user