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
111 lines
4.2 KiB
Python
Executable File
111 lines
4.2 KiB
Python
Executable File
import pytest
|
|
import httpx
|
|
import asyncio
|
|
import subprocess
|
|
import time
|
|
import uuid
|
|
|
|
# Nodes URLs
|
|
AITBC_URL = "http://127.0.0.1:18000/v1"
|
|
AITBC1_URL = "http://127.0.0.1:18001/v1"
|
|
|
|
@pytest.fixture(scope="session", autouse=True)
|
|
def setup_environment():
|
|
# Attempt to start proxy on 18000 and 18001 pointing to aitbc and aitbc1
|
|
print("Setting up SSH tunnels for cross-container testing...")
|
|
|
|
import socket
|
|
def is_port_in_use(port):
|
|
with socket.socket(socket.AF_INET, socket.SOCK_STREAM) as s:
|
|
return s.connect_ex(('localhost', port)) == 0
|
|
|
|
p1 = None
|
|
p2 = None
|
|
|
|
if not is_port_in_use(18000):
|
|
print("Starting SSH tunnel on port 18000 to aitbc-cascade")
|
|
p1 = subprocess.Popen(["ssh", "-L", "18000:localhost:8000", "-N", "aitbc-cascade"])
|
|
|
|
if not is_port_in_use(18001):
|
|
print("Starting SSH tunnel on port 18001 to aitbc1-cascade")
|
|
p2 = subprocess.Popen(["ssh", "-L", "18001:localhost:8000", "-N", "aitbc1-cascade"])
|
|
|
|
# Give tunnels time to establish
|
|
time.sleep(3)
|
|
|
|
yield
|
|
|
|
print("Tearing down SSH tunnels...")
|
|
if p1: p1.kill()
|
|
if p2: p2.kill()
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_cross_container_marketplace_sync():
|
|
"""Test Phase 1 & 2: Miner registers on aitbc, Client discovers on aitbc1"""
|
|
|
|
unique_miner_id = f"miner_cross_test_{uuid.uuid4().hex[:8]}"
|
|
|
|
async with httpx.AsyncClient() as client:
|
|
# Check health of both nodes
|
|
try:
|
|
health1 = await client.get(f"{AITBC_URL}/health")
|
|
health2 = await client.get(f"{AITBC1_URL}/health")
|
|
assert health1.status_code == 200, f"aitbc (18000) is not healthy: {health1.text}"
|
|
assert health2.status_code == 200, f"aitbc1 (18001) is not healthy: {health2.text}"
|
|
except httpx.ConnectError:
|
|
pytest.skip("SSH tunnels or target API servers are not reachable. Skipping test.")
|
|
|
|
# 1. Register GPU Miner on aitbc (Primary MP)
|
|
miner_payload = {
|
|
"gpu": {
|
|
"miner_id": unique_miner_id,
|
|
"name": "NVIDIA-RTX-4060Ti",
|
|
"memory": 16,
|
|
"cuda_version": "12.2",
|
|
"region": "localhost",
|
|
"price_per_hour": 0.001,
|
|
"capabilities": ["gemma3:1b", "lauchacarro/qwen2.5-translator:latest"]
|
|
}
|
|
}
|
|
|
|
register_response = await client.post(
|
|
f"{AITBC_URL}/marketplace/gpu/register",
|
|
json=miner_payload
|
|
)
|
|
assert register_response.status_code in [200, 201], f"Failed to register on aitbc: {register_response.text}"
|
|
|
|
# Verify it exists on aitbc
|
|
verify_aitbc = await client.get(f"{AITBC_URL}/marketplace/gpu/list")
|
|
assert verify_aitbc.status_code == 200
|
|
|
|
found_on_primary = False
|
|
for gpu in verify_aitbc.json():
|
|
if gpu.get("miner_id") == unique_miner_id:
|
|
found_on_primary = True
|
|
break
|
|
assert found_on_primary, "GPU was registered but not found on primary node (aitbc)"
|
|
|
|
# 2. Wait for synchronization (Redis replication/gossip to happen between containers)
|
|
await asyncio.sleep(2)
|
|
|
|
# 3. Client Discovers Miner on aitbc1 (Secondary MP)
|
|
# List GPUs on aitbc1
|
|
discover_response = await client.get(f"{AITBC1_URL}/marketplace/gpu/list")
|
|
|
|
if discover_response.status_code == 200:
|
|
gpus = discover_response.json()
|
|
|
|
# Note: In a fully configured clustered DB, this should be True.
|
|
# Currently they might have independent DBs unless configured otherwise.
|
|
found_on_secondary = False
|
|
for gpu in gpus:
|
|
if gpu.get("miner_id") == unique_miner_id:
|
|
found_on_secondary = True
|
|
break
|
|
|
|
if not found_on_secondary:
|
|
print(f"\\n[INFO] GPU {unique_miner_id} not found on aitbc1. Database replication may not be active between containers. This is expected in independent test environments.")
|
|
else:
|
|
assert discover_response.status_code == 200, f"Failed to list GPUs on aitbc1: {discover_response.text}"
|
|
|