Add sys import to test files and remove obsolete integration tests
Some checks failed
API Endpoint Tests / test-api-endpoints (push) Successful in 9s
Blockchain Synchronization Verification / sync-verification (push) Failing after 1s
CLI Tests / test-cli (push) Failing after 3s
Documentation Validation / validate-docs (push) Successful in 6s
Documentation Validation / validate-policies-strict (push) Successful in 2s
Integration Tests / test-service-integration (push) Successful in 40s
Multi-Node Blockchain Health Monitoring / health-check (push) Successful in 1s
P2P Network Verification / p2p-verification (push) Successful in 2s
Production Tests / Production Integration Tests (push) Successful in 21s
Python Tests / test-python (push) Successful in 13s
Security Scanning / security-scan (push) Failing after 46s
Smart Contract Tests / test-solidity (map[name:aitbc-token path:packages/solidity/aitbc-token]) (push) Successful in 17s
Smart Contract Tests / lint-solidity (push) Successful in 10s
Some checks failed
API Endpoint Tests / test-api-endpoints (push) Successful in 9s
Blockchain Synchronization Verification / sync-verification (push) Failing after 1s
CLI Tests / test-cli (push) Failing after 3s
Documentation Validation / validate-docs (push) Successful in 6s
Documentation Validation / validate-policies-strict (push) Successful in 2s
Integration Tests / test-service-integration (push) Successful in 40s
Multi-Node Blockchain Health Monitoring / health-check (push) Successful in 1s
P2P Network Verification / p2p-verification (push) Successful in 2s
Production Tests / Production Integration Tests (push) Successful in 21s
Python Tests / test-python (push) Successful in 13s
Security Scanning / security-scan (push) Failing after 46s
Smart Contract Tests / test-solidity (map[name:aitbc-token path:packages/solidity/aitbc-token]) (push) Successful in 17s
Smart Contract Tests / lint-solidity (push) Successful in 10s
- Add sys import to 29 test files across agent-coordinator, blockchain-event-bridge, blockchain-node, and coordinator-api - Remove apps/blockchain-event-bridge/tests/test_integration.py (obsolete bridge integration tests) - Remove apps/coordinator-api/tests/test_integration.py (obsolete API integration tests) - Implement GPU registration in marketplace_gpu.py with GPURegistry model persistence
This commit is contained in:
506
apps/marketplace/tests/test_integration_marketplace.py
Normal file
506
apps/marketplace/tests/test_integration_marketplace.py
Normal file
@@ -0,0 +1,506 @@
|
||||
"""Integration tests for agent marketplace service"""
|
||||
|
||||
import pytest
|
||||
import sys
|
||||
import sys
|
||||
from pathlib import Path
|
||||
from fastapi.testclient import TestClient
|
||||
|
||||
|
||||
from agent_marketplace import app, GPUOffering, DealRequest, DealConfirmation, MinerRegistration, gpu_offerings, marketplace_deals, miner_registrations, chain_offerings
|
||||
|
||||
|
||||
@pytest.fixture(autouse=True)
|
||||
def reset_state():
|
||||
"""Reset global state before each test"""
|
||||
gpu_offerings.clear()
|
||||
marketplace_deals.clear()
|
||||
miner_registrations.clear()
|
||||
chain_offerings.clear()
|
||||
yield
|
||||
gpu_offerings.clear()
|
||||
marketplace_deals.clear()
|
||||
miner_registrations.clear()
|
||||
chain_offerings.clear()
|
||||
|
||||
|
||||
@pytest.mark.integration
|
||||
def test_health_check():
|
||||
"""Test health check endpoint"""
|
||||
client = TestClient(app)
|
||||
response = client.get("/health")
|
||||
assert response.status_code == 200
|
||||
data = response.json()
|
||||
assert data["status"] == "ok"
|
||||
assert "supported_chains" in data
|
||||
|
||||
|
||||
@pytest.mark.integration
|
||||
def test_get_supported_chains():
|
||||
"""Test getting supported chains"""
|
||||
client = TestClient(app)
|
||||
response = client.get("/api/v1/chains")
|
||||
assert response.status_code == 200
|
||||
data = response.json()
|
||||
assert "chains" in data
|
||||
|
||||
|
||||
@pytest.mark.integration
|
||||
def test_register_miner():
|
||||
"""Test registering a miner"""
|
||||
client = TestClient(app)
|
||||
registration = MinerRegistration(
|
||||
miner_id="miner_123",
|
||||
wallet_address="0x1234567890abcdef",
|
||||
preferred_chains=["ait-devnet"],
|
||||
gpu_specs={"model": "RTX 4090"}
|
||||
)
|
||||
response = client.post("/api/v1/miners/register", json=registration.model_dump())
|
||||
assert response.status_code == 200
|
||||
data = response.json()
|
||||
assert data["success"] is True
|
||||
assert data["miner_id"] == "miner_123"
|
||||
|
||||
|
||||
@pytest.mark.integration
|
||||
def test_register_miner_update_existing():
|
||||
"""Test updating existing miner registration"""
|
||||
client = TestClient(app)
|
||||
registration = MinerRegistration(
|
||||
miner_id="miner_123",
|
||||
wallet_address="0x1234567890abcdef",
|
||||
preferred_chains=["ait-devnet"],
|
||||
gpu_specs={"model": "RTX 4090"}
|
||||
)
|
||||
client.post("/api/v1/miners/register", json=registration.model_dump())
|
||||
|
||||
# Update with new data
|
||||
registration.wallet_address = "0xabcdef1234567890"
|
||||
response = client.post("/api/v1/miners/register", json=registration.model_dump())
|
||||
assert response.status_code == 200
|
||||
|
||||
|
||||
@pytest.mark.integration
|
||||
def test_create_gpu_offering():
|
||||
"""Test creating a GPU offering"""
|
||||
client = TestClient(app)
|
||||
offering = GPUOffering(
|
||||
miner_id="miner_123",
|
||||
gpu_model="RTX 4090",
|
||||
gpu_memory=24576,
|
||||
cuda_cores=16384,
|
||||
price_per_hour=0.50,
|
||||
available_hours=24,
|
||||
chains=["ait-devnet"],
|
||||
capabilities=["inference"]
|
||||
)
|
||||
response = client.post("/api/v1/offerings/create", json=offering.model_dump())
|
||||
assert response.status_code == 200
|
||||
data = response.json()
|
||||
assert data["success"] is True
|
||||
assert "offering_id" in data
|
||||
|
||||
|
||||
@pytest.mark.integration
|
||||
def test_create_gpu_offering_invalid_chain():
|
||||
"""Test creating GPU offering with invalid chain"""
|
||||
client = TestClient(app)
|
||||
offering = GPUOffering(
|
||||
miner_id="miner_123",
|
||||
gpu_model="RTX 4090",
|
||||
gpu_memory=24576,
|
||||
cuda_cores=16384,
|
||||
price_per_hour=0.50,
|
||||
available_hours=24,
|
||||
chains=["invalid-chain"],
|
||||
capabilities=["inference"]
|
||||
)
|
||||
response = client.post("/api/v1/offerings/create", json=offering.model_dump())
|
||||
assert response.status_code == 400
|
||||
|
||||
|
||||
@pytest.mark.integration
|
||||
def test_get_gpu_offerings():
|
||||
"""Test getting GPU offerings"""
|
||||
client = TestClient(app)
|
||||
# Create an offering first
|
||||
offering = GPUOffering(
|
||||
miner_id="miner_123",
|
||||
gpu_model="RTX 4090",
|
||||
gpu_memory=24576,
|
||||
cuda_cores=16384,
|
||||
price_per_hour=0.50,
|
||||
available_hours=24,
|
||||
chains=["ait-devnet"],
|
||||
capabilities=["inference"]
|
||||
)
|
||||
client.post("/api/v1/offerings/create", json=offering.model_dump())
|
||||
|
||||
response = client.get("/api/v1/offerings")
|
||||
assert response.status_code == 200
|
||||
data = response.json()
|
||||
assert "offerings" in data
|
||||
|
||||
|
||||
@pytest.mark.integration
|
||||
def test_get_gpu_offerings_with_filters():
|
||||
"""Test getting GPU offerings with filters"""
|
||||
client = TestClient(app)
|
||||
# Create offerings
|
||||
offering1 = GPUOffering(
|
||||
miner_id="miner_123",
|
||||
gpu_model="RTX 4090",
|
||||
gpu_memory=24576,
|
||||
cuda_cores=16384,
|
||||
price_per_hour=0.50,
|
||||
available_hours=24,
|
||||
chains=["ait-devnet"],
|
||||
capabilities=["inference"]
|
||||
)
|
||||
offering2 = GPUOffering(
|
||||
miner_id="miner_456",
|
||||
gpu_model="RTX 3080",
|
||||
gpu_memory=10240,
|
||||
cuda_cores=8704,
|
||||
price_per_hour=0.30,
|
||||
available_hours=24,
|
||||
chains=["ait-testnet"],
|
||||
capabilities=["inference"]
|
||||
)
|
||||
client.post("/api/v1/offerings/create", json=offering1.model_dump())
|
||||
client.post("/api/v1/offerings/create", json=offering2.model_dump())
|
||||
|
||||
response = client.get("/api/v1/offerings?chain=ait-devnet&gpu_model=RTX")
|
||||
assert response.status_code == 200
|
||||
|
||||
|
||||
@pytest.mark.integration
|
||||
def test_get_gpu_offering():
|
||||
"""Test getting specific GPU offering"""
|
||||
client = TestClient(app)
|
||||
# Create an offering first
|
||||
offering = GPUOffering(
|
||||
miner_id="miner_123",
|
||||
gpu_model="RTX 4090",
|
||||
gpu_memory=24576,
|
||||
cuda_cores=16384,
|
||||
price_per_hour=0.50,
|
||||
available_hours=24,
|
||||
chains=["ait-devnet"],
|
||||
capabilities=["inference"]
|
||||
)
|
||||
create_response = client.post("/api/v1/offerings/create", json=offering.model_dump())
|
||||
offering_id = create_response.json()["offering_id"]
|
||||
|
||||
response = client.get(f"/api/v1/offerings/{offering_id}")
|
||||
assert response.status_code == 200
|
||||
data = response.json()
|
||||
assert data["offering_id"] == offering_id
|
||||
|
||||
|
||||
@pytest.mark.integration
|
||||
def test_get_gpu_offering_not_found():
|
||||
"""Test getting nonexistent GPU offering"""
|
||||
client = TestClient(app)
|
||||
response = client.get("/api/v1/offerings/nonexistent")
|
||||
assert response.status_code == 404
|
||||
|
||||
|
||||
@pytest.mark.integration
|
||||
def test_request_deal():
|
||||
"""Test requesting a deal"""
|
||||
client = TestClient(app)
|
||||
# Create an offering first
|
||||
offering = GPUOffering(
|
||||
miner_id="miner_123",
|
||||
gpu_model="RTX 4090",
|
||||
gpu_memory=24576,
|
||||
cuda_cores=16384,
|
||||
price_per_hour=0.50,
|
||||
available_hours=24,
|
||||
chains=["ait-devnet"],
|
||||
capabilities=["inference"]
|
||||
)
|
||||
create_response = client.post("/api/v1/offerings/create", json=offering.model_dump())
|
||||
offering_id = create_response.json()["offering_id"]
|
||||
|
||||
deal_request = DealRequest(
|
||||
offering_id=offering_id,
|
||||
buyer_id="buyer_123",
|
||||
rental_hours=10,
|
||||
chain="ait-devnet"
|
||||
)
|
||||
response = client.post("/api/v1/deals/request", json=deal_request.model_dump())
|
||||
assert response.status_code == 200
|
||||
data = response.json()
|
||||
assert data["success"] is True
|
||||
assert "deal_id" in data
|
||||
|
||||
|
||||
@pytest.mark.integration
|
||||
def test_request_deal_offering_not_found():
|
||||
"""Test requesting deal for nonexistent offering"""
|
||||
client = TestClient(app)
|
||||
deal_request = DealRequest(
|
||||
offering_id="nonexistent",
|
||||
buyer_id="buyer_123",
|
||||
rental_hours=10,
|
||||
chain="ait-devnet"
|
||||
)
|
||||
response = client.post("/api/v1/deals/request", json=deal_request.model_dump())
|
||||
assert response.status_code == 404
|
||||
|
||||
|
||||
@pytest.mark.integration
|
||||
def test_request_deal_chain_not_supported():
|
||||
"""Test requesting deal with unsupported chain"""
|
||||
client = TestClient(app)
|
||||
# Create an offering
|
||||
offering = GPUOffering(
|
||||
miner_id="miner_123",
|
||||
gpu_model="RTX 4090",
|
||||
gpu_memory=24576,
|
||||
cuda_cores=16384,
|
||||
price_per_hour=0.50,
|
||||
available_hours=24,
|
||||
chains=["ait-devnet"],
|
||||
capabilities=["inference"]
|
||||
)
|
||||
create_response = client.post("/api/v1/offerings/create", json=offering.model_dump())
|
||||
offering_id = create_response.json()["offering_id"]
|
||||
|
||||
deal_request = DealRequest(
|
||||
offering_id=offering_id,
|
||||
buyer_id="buyer_123",
|
||||
rental_hours=10,
|
||||
chain="ait-testnet"
|
||||
)
|
||||
response = client.post("/api/v1/deals/request", json=deal_request.model_dump())
|
||||
assert response.status_code == 400
|
||||
|
||||
|
||||
@pytest.mark.integration
|
||||
def test_confirm_deal():
|
||||
"""Test confirming a deal"""
|
||||
client = TestClient(app)
|
||||
# Create offering and request deal
|
||||
offering = GPUOffering(
|
||||
miner_id="miner_123",
|
||||
gpu_model="RTX 4090",
|
||||
gpu_memory=24576,
|
||||
cuda_cores=16384,
|
||||
price_per_hour=0.50,
|
||||
available_hours=24,
|
||||
chains=["ait-devnet"],
|
||||
capabilities=["inference"]
|
||||
)
|
||||
create_response = client.post("/api/v1/offerings/create", json=offering.model_dump())
|
||||
offering_id = create_response.json()["offering_id"]
|
||||
|
||||
deal_request = DealRequest(
|
||||
offering_id=offering_id,
|
||||
buyer_id="buyer_123",
|
||||
rental_hours=10,
|
||||
chain="ait-devnet"
|
||||
)
|
||||
deal_response = client.post("/api/v1/deals/request", json=deal_request.model_dump())
|
||||
deal_id = deal_response.json()["deal_id"]
|
||||
|
||||
confirmation = DealConfirmation(
|
||||
deal_id=deal_id,
|
||||
miner_confirmation=True,
|
||||
chain="ait-devnet"
|
||||
)
|
||||
response = client.post(f"/api/v1/deals/{deal_id}/confirm", json=confirmation.model_dump())
|
||||
assert response.status_code == 200
|
||||
data = response.json()
|
||||
assert data["success"] is True
|
||||
assert data["status"] == "confirmed"
|
||||
|
||||
|
||||
@pytest.mark.integration
|
||||
def test_confirm_deal_reject():
|
||||
"""Test rejecting a deal"""
|
||||
client = TestClient(app)
|
||||
# Create offering and request deal
|
||||
offering = GPUOffering(
|
||||
miner_id="miner_123",
|
||||
gpu_model="RTX 4090",
|
||||
gpu_memory=24576,
|
||||
cuda_cores=16384,
|
||||
price_per_hour=0.50,
|
||||
available_hours=24,
|
||||
chains=["ait-devnet"],
|
||||
capabilities=["inference"]
|
||||
)
|
||||
create_response = client.post("/api/v1/offerings/create", json=offering.model_dump())
|
||||
offering_id = create_response.json()["offering_id"]
|
||||
|
||||
deal_request = DealRequest(
|
||||
offering_id=offering_id,
|
||||
buyer_id="buyer_123",
|
||||
rental_hours=10,
|
||||
chain="ait-devnet"
|
||||
)
|
||||
deal_response = client.post("/api/v1/deals/request", json=deal_request.model_dump())
|
||||
deal_id = deal_response.json()["deal_id"]
|
||||
|
||||
confirmation = DealConfirmation(
|
||||
deal_id=deal_id,
|
||||
miner_confirmation=False,
|
||||
chain="ait-devnet"
|
||||
)
|
||||
response = client.post(f"/api/v1/deals/{deal_id}/confirm", json=confirmation.model_dump())
|
||||
assert response.status_code == 200
|
||||
data = response.json()
|
||||
assert data["status"] == "rejected"
|
||||
|
||||
|
||||
@pytest.mark.integration
|
||||
def test_confirm_deal_not_found():
|
||||
"""Test confirming nonexistent deal"""
|
||||
client = TestClient(app)
|
||||
confirmation = DealConfirmation(
|
||||
deal_id="nonexistent",
|
||||
miner_confirmation=True,
|
||||
chain="ait-devnet"
|
||||
)
|
||||
response = client.post("/api/v1/deals/nonexistent/confirm", json=confirmation.model_dump())
|
||||
assert response.status_code == 404
|
||||
|
||||
|
||||
@pytest.mark.integration
|
||||
def test_get_deals():
|
||||
"""Test getting deals"""
|
||||
client = TestClient(app)
|
||||
response = client.get("/api/v1/deals")
|
||||
assert response.status_code == 200
|
||||
data = response.json()
|
||||
assert "deals" in data
|
||||
|
||||
|
||||
@pytest.mark.integration
|
||||
def test_get_deals_with_filters():
|
||||
"""Test getting deals with filters"""
|
||||
client = TestClient(app)
|
||||
# Create offering and request deal
|
||||
offering = GPUOffering(
|
||||
miner_id="miner_123",
|
||||
gpu_model="RTX 4090",
|
||||
gpu_memory=24576,
|
||||
cuda_cores=16384,
|
||||
price_per_hour=0.50,
|
||||
available_hours=24,
|
||||
chains=["ait-devnet"],
|
||||
capabilities=["inference"]
|
||||
)
|
||||
create_response = client.post("/api/v1/offerings/create", json=offering.model_dump())
|
||||
offering_id = create_response.json()["offering_id"]
|
||||
|
||||
deal_request = DealRequest(
|
||||
offering_id=offering_id,
|
||||
buyer_id="buyer_123",
|
||||
rental_hours=10,
|
||||
chain="ait-devnet"
|
||||
)
|
||||
client.post("/api/v1/deals/request", json=deal_request.model_dump())
|
||||
|
||||
response = client.get("/api/v1/deals?miner_id=miner_123")
|
||||
assert response.status_code == 200
|
||||
|
||||
|
||||
@pytest.mark.integration
|
||||
def test_get_miner_offerings():
|
||||
"""Test getting offerings for a specific miner"""
|
||||
client = TestClient(app)
|
||||
# Create an offering
|
||||
offering = GPUOffering(
|
||||
miner_id="miner_123",
|
||||
gpu_model="RTX 4090",
|
||||
gpu_memory=24576,
|
||||
cuda_cores=16384,
|
||||
price_per_hour=0.50,
|
||||
available_hours=24,
|
||||
chains=["ait-devnet"],
|
||||
capabilities=["inference"]
|
||||
)
|
||||
client.post("/api/v1/offerings/create", json=offering.model_dump())
|
||||
|
||||
response = client.get("/api/v1/miners/miner_123/offerings")
|
||||
assert response.status_code == 200
|
||||
data = response.json()
|
||||
assert data["miner_id"] == "miner_123"
|
||||
|
||||
|
||||
@pytest.mark.integration
|
||||
def test_get_chain_offerings():
|
||||
"""Test getting offerings for a specific chain"""
|
||||
client = TestClient(app)
|
||||
# Create an offering
|
||||
offering = GPUOffering(
|
||||
miner_id="miner_123",
|
||||
gpu_model="RTX 4090",
|
||||
gpu_memory=24576,
|
||||
cuda_cores=16384,
|
||||
price_per_hour=0.50,
|
||||
available_hours=24,
|
||||
chains=["ait-devnet"],
|
||||
capabilities=["inference"]
|
||||
)
|
||||
client.post("/api/v1/offerings/create", json=offering.model_dump())
|
||||
|
||||
response = client.get("/api/v1/chains/ait-devnet/offerings")
|
||||
assert response.status_code == 200
|
||||
data = response.json()
|
||||
assert data["chain"] == "ait-devnet"
|
||||
|
||||
|
||||
@pytest.mark.integration
|
||||
def test_get_chain_offerings_unsupported_chain():
|
||||
"""Test getting offerings for unsupported chain"""
|
||||
client = TestClient(app)
|
||||
response = client.get("/api/v1/chains/unsupported-chain/offerings")
|
||||
assert response.status_code == 400
|
||||
|
||||
|
||||
@pytest.mark.integration
|
||||
def test_remove_offering():
|
||||
"""Test removing a GPU offering"""
|
||||
client = TestClient(app)
|
||||
# Create an offering
|
||||
offering = GPUOffering(
|
||||
miner_id="miner_123",
|
||||
gpu_model="RTX 4090",
|
||||
gpu_memory=24576,
|
||||
cuda_cores=16384,
|
||||
price_per_hour=0.50,
|
||||
available_hours=24,
|
||||
chains=["ait-devnet"],
|
||||
capabilities=["inference"]
|
||||
)
|
||||
create_response = client.post("/api/v1/offerings/create", json=offering.model_dump())
|
||||
offering_id = create_response.json()["offering_id"]
|
||||
|
||||
response = client.delete(f"/api/v1/offerings/{offering_id}")
|
||||
assert response.status_code == 200
|
||||
data = response.json()
|
||||
assert data["success"] is True
|
||||
|
||||
|
||||
@pytest.mark.integration
|
||||
def test_remove_offering_not_found():
|
||||
"""Test removing nonexistent offering"""
|
||||
client = TestClient(app)
|
||||
response = client.delete("/api/v1/offerings/nonexistent")
|
||||
assert response.status_code == 404
|
||||
|
||||
|
||||
@pytest.mark.integration
|
||||
def test_get_marketplace_stats():
|
||||
"""Test getting marketplace statistics"""
|
||||
client = TestClient(app)
|
||||
response = client.get("/api/v1/stats")
|
||||
assert response.status_code == 200
|
||||
data = response.json()
|
||||
assert "total_offerings" in data
|
||||
assert "chain_stats" in data
|
||||
Reference in New Issue
Block a user