Add sys import to test files and remove obsolete integration tests

- 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

apps/multi-region-load-balancer/tests/__init__.py

@@ -0,0 +1 @@
+"""Multi-region load balancer service tests"""

apps/multi-region-load-balancer/tests/test_edge_cases_multi_region_load_balancer.py

@@ -0,0 +1,199 @@
+"""Edge case and error handling tests for multi-region load balancer service"""
+
+import pytest
+import sys
+import sys
+from pathlib import Path
+from fastapi.testclient import TestClient
+from datetime import datetime
+
+
+from main import app, LoadBalancingRule, RegionHealth, LoadBalancingMetrics, GeographicRule, load_balancing_rules, region_health_status, balancing_metrics, geographic_rules
+
+
+@pytest.fixture(autouse=True)
+def reset_state():
+    """Reset global state before each test"""
+    load_balancing_rules.clear()
+    region_health_status.clear()
+    balancing_metrics.clear()
+    geographic_rules.clear()
+    yield
+    load_balancing_rules.clear()
+    region_health_status.clear()
+    balancing_metrics.clear()
+    geographic_rules.clear()
+
+
+@pytest.mark.unit
+def test_load_balancing_rule_empty_target_regions():
+    """Test LoadBalancingRule with empty target regions"""
+    rule = LoadBalancingRule(
+        rule_id="rule_123",
+        name="Test Rule",
+        algorithm="round_robin",
+        target_regions=[],
+        weights={},
+        health_check_path="/health",
+        failover_enabled=False,
+        session_affinity=False
+    )
+    assert rule.target_regions == []
+
+
+@pytest.mark.unit
+def test_region_health_negative_success_rate():
+    """Test RegionHealth with negative success rate"""
+    health = RegionHealth(
+        region_id="us-east-1",
+        status="healthy",
+        response_time_ms=45.5,
+        success_rate=-0.5,
+        active_connections=100,
+        last_check=datetime.utcnow()
+    )
+    assert health.success_rate == -0.5
+
+
+@pytest.mark.unit
+def test_region_health_negative_connections():
+    """Test RegionHealth with negative connections"""
+    health = RegionHealth(
+        region_id="us-east-1",
+        status="healthy",
+        response_time_ms=45.5,
+        success_rate=0.99,
+        active_connections=-100,
+        last_check=datetime.utcnow()
+    )
+    assert health.active_connections == -100
+
+
+@pytest.mark.unit
+def test_load_balancing_metrics_negative_requests():
+    """Test LoadBalancingMetrics with negative requests"""
+    metrics = LoadBalancingMetrics(
+        balancer_id="lb_123",
+        timestamp=datetime.utcnow(),
+        total_requests=-1000,
+        requests_per_region={},
+        average_response_time=50.5,
+        error_rate=0.001,
+        throughput=100.0
+    )
+    assert metrics.total_requests == -1000
+
+
+@pytest.mark.unit
+def test_load_balancing_metrics_negative_response_time():
+    """Test LoadBalancingMetrics with negative response time"""
+    metrics = LoadBalancingMetrics(
+        balancer_id="lb_123",
+        timestamp=datetime.utcnow(),
+        total_requests=1000,
+        requests_per_region={},
+        average_response_time=-50.5,
+        error_rate=0.001,
+        throughput=100.0
+    )
+    assert metrics.average_response_time == -50.5
+
+
+@pytest.mark.unit
+def test_geographic_rule_empty_source_regions():
+    """Test GeographicRule with empty source regions"""
+    rule = GeographicRule(
+        rule_id="geo_123",
+        source_regions=[],
+        target_regions=["us-east-1"],
+        priority=1,
+        latency_threshold_ms=50.0
+    )
+    assert rule.source_regions == []
+
+
+@pytest.mark.unit
+def test_geographic_rule_negative_priority():
+    """Test GeographicRule with negative priority"""
+    rule = GeographicRule(
+        rule_id="geo_123",
+        source_regions=["us-east"],
+        target_regions=["us-east-1"],
+        priority=-5,
+        latency_threshold_ms=50.0
+    )
+    assert rule.priority == -5
+
+
+@pytest.mark.unit
+def test_geographic_rule_negative_latency_threshold():
+    """Test GeographicRule with negative latency threshold"""
+    rule = GeographicRule(
+        rule_id="geo_123",
+        source_regions=["us-east"],
+        target_regions=["us-east-1"],
+        priority=1,
+        latency_threshold_ms=-50.0
+    )
+    assert rule.latency_threshold_ms == -50.0
+
+
+@pytest.mark.integration
+def test_list_rules_with_no_rules():
+    """Test listing rules when no rules exist"""
+    client = TestClient(app)
+    response = client.get("/api/v1/rules")
+    assert response.status_code == 200
+    data = response.json()
+    assert data["total_rules"] == 0
+
+
+@pytest.mark.integration
+def test_get_region_health_with_no_regions():
+    """Test getting region health when no regions exist"""
+    client = TestClient(app)
+    response = client.get("/api/v1/health")
+    assert response.status_code == 200
+    data = response.json()
+    assert data["total_regions"] == 0
+
+
+@pytest.mark.integration
+def test_get_balancing_metrics_hours_parameter():
+    """Test getting balancing metrics with custom hours parameter"""
+    client = TestClient(app)
+    # Create a rule first
+    rule = LoadBalancingRule(
+        rule_id="rule_123",
+        name="Test Rule",
+        algorithm="weighted_round_robin",
+        target_regions=["us-east-1"],
+        weights={"us-east-1": 1.0},
+        health_check_path="/health",
+        failover_enabled=True,
+        session_affinity=False
+    )
+    client.post("/api/v1/rules/create", json=rule.model_dump())
+    
+    response = client.get("/api/v1/metrics/rule_123?hours=12")
+    assert response.status_code == 200
+    data = response.json()
+    assert data["period_hours"] == 12
+
+
+@pytest.mark.integration
+def test_get_optimal_region_nonexistent_rule():
+    """Test getting optimal region with nonexistent rule"""
+    client = TestClient(app)
+    response = client.get("/api/v1/route/us-east?rule_id=nonexistent")
+    assert response.status_code == 404
+
+
+@pytest.mark.integration
+def test_dashboard_with_no_data():
+    """Test dashboard with no data"""
+    client = TestClient(app)
+    response = client.get("/api/v1/dashboard")
+    assert response.status_code == 200
+    data = response.json()
+    assert data["dashboard"]["overview"]["total_rules"] == 0

apps/multi-region-load-balancer/tests/test_integration_multi_region_load_balancer.py

@@ -0,0 +1,341 @@
+"""Integration tests for multi-region load balancer service"""
+
+import pytest
+import sys
+import sys
+from pathlib import Path
+from fastapi.testclient import TestClient
+from datetime import datetime
+
+
+from main import app, LoadBalancingRule, RegionHealth, LoadBalancingMetrics, GeographicRule, load_balancing_rules, region_health_status, balancing_metrics, geographic_rules
+
+
+@pytest.fixture(autouse=True)
+def reset_state():
+    """Reset global state before each test"""
+    load_balancing_rules.clear()
+    region_health_status.clear()
+    balancing_metrics.clear()
+    geographic_rules.clear()
+    yield
+    load_balancing_rules.clear()
+    region_health_status.clear()
+    balancing_metrics.clear()
+    geographic_rules.clear()
+
+
+@pytest.mark.integration
+def test_root_endpoint():
+    """Test root endpoint"""
+    client = TestClient(app)
+    response = client.get("/")
+    assert response.status_code == 200
+    data = response.json()
+    assert data["service"] == "AITBC Multi-Region Load Balancer"
+    assert data["status"] == "running"
+
+
+@pytest.mark.integration
+def test_health_check_endpoint():
+    """Test health check endpoint"""
+    client = TestClient(app)
+    response = client.get("/health")
+    assert response.status_code == 200
+    data = response.json()
+    assert data["status"] == "healthy"
+    assert "total_rules" in data
+
+
+@pytest.mark.integration
+def test_create_load_balancing_rule():
+    """Test creating a load balancing rule"""
+    client = TestClient(app)
+    rule = LoadBalancingRule(
+        rule_id="rule_123",
+        name="Test Rule",
+        algorithm="weighted_round_robin",
+        target_regions=["us-east-1"],
+        weights={"us-east-1": 1.0},
+        health_check_path="/health",
+        failover_enabled=True,
+        session_affinity=False
+    )
+    response = client.post("/api/v1/rules/create", json=rule.model_dump())
+    assert response.status_code == 200
+    data = response.json()
+    assert data["rule_id"] == "rule_123"
+    assert data["status"] == "created"
+
+
+@pytest.mark.integration
+def test_create_duplicate_rule():
+    """Test creating duplicate load balancing rule"""
+    client = TestClient(app)
+    rule = LoadBalancingRule(
+        rule_id="rule_123",
+        name="Test Rule",
+        algorithm="weighted_round_robin",
+        target_regions=["us-east-1"],
+        weights={"us-east-1": 1.0},
+        health_check_path="/health",
+        failover_enabled=True,
+        session_affinity=False
+    )
+    client.post("/api/v1/rules/create", json=rule.model_dump())
+    
+    response = client.post("/api/v1/rules/create", json=rule.model_dump())
+    assert response.status_code == 400
+
+
+@pytest.mark.integration
+def test_list_load_balancing_rules():
+    """Test listing load balancing rules"""
+    client = TestClient(app)
+    response = client.get("/api/v1/rules")
+    assert response.status_code == 200
+    data = response.json()
+    assert "rules" in data
+    assert "total_rules" in data
+
+
+@pytest.mark.integration
+def test_get_load_balancing_rule():
+    """Test getting specific load balancing rule"""
+    client = TestClient(app)
+    rule = LoadBalancingRule(
+        rule_id="rule_123",
+        name="Test Rule",
+        algorithm="weighted_round_robin",
+        target_regions=["us-east-1"],
+        weights={"us-east-1": 1.0},
+        health_check_path="/health",
+        failover_enabled=True,
+        session_affinity=False
+    )
+    client.post("/api/v1/rules/create", json=rule.model_dump())
+    
+    response = client.get("/api/v1/rules/rule_123")
+    assert response.status_code == 200
+    data = response.json()
+    assert data["rule_id"] == "rule_123"
+
+
+@pytest.mark.integration
+def test_get_load_balancing_rule_not_found():
+    """Test getting nonexistent load balancing rule"""
+    client = TestClient(app)
+    response = client.get("/api/v1/rules/nonexistent")
+    assert response.status_code == 404
+
+
+@pytest.mark.integration
+def test_update_rule_weights():
+    """Test updating rule weights"""
+    client = TestClient(app)
+    rule = LoadBalancingRule(
+        rule_id="rule_123",
+        name="Test Rule",
+        algorithm="weighted_round_robin",
+        target_regions=["us-east-1", "eu-west-1"],
+        weights={"us-east-1": 0.5, "eu-west-1": 0.5},
+        health_check_path="/health",
+        failover_enabled=True,
+        session_affinity=False
+    )
+    client.post("/api/v1/rules/create", json=rule.model_dump())
+    
+    new_weights = {"us-east-1": 0.7, "eu-west-1": 0.3}
+    response = client.post("/api/v1/rules/rule_123/update-weights", json=new_weights)
+    assert response.status_code == 200
+    data = response.json()
+    assert data["rule_id"] == "rule_123"
+    assert "new_weights" in data
+
+
+@pytest.mark.integration
+def test_update_rule_weights_not_found():
+    """Test updating weights for nonexistent rule"""
+    client = TestClient(app)
+    new_weights = {"us-east-1": 1.0}
+    response = client.post("/api/v1/rules/nonexistent/update-weights", json=new_weights)
+    assert response.status_code == 404
+
+
+@pytest.mark.integration
+def test_update_rule_weights_zero_total():
+    """Test updating weights with zero total"""
+    client = TestClient(app)
+    rule = LoadBalancingRule(
+        rule_id="rule_123",
+        name="Test Rule",
+        algorithm="weighted_round_robin",
+        target_regions=["us-east-1"],
+        weights={"us-east-1": 1.0},
+        health_check_path="/health",
+        failover_enabled=True,
+        session_affinity=False
+    )
+    client.post("/api/v1/rules/create", json=rule.model_dump())
+    
+    new_weights = {"us-east-1": 0.0}
+    response = client.post("/api/v1/rules/rule_123/update-weights", json=new_weights)
+    assert response.status_code == 400
+
+
+@pytest.mark.integration
+def test_register_region_health():
+    """Test registering region health"""
+    client = TestClient(app)
+    health = RegionHealth(
+        region_id="us-east-1",
+        status="healthy",
+        response_time_ms=45.5,
+        success_rate=0.99,
+        active_connections=100,
+        last_check=datetime.utcnow()
+    )
+    response = client.post("/api/v1/health/register", json=health.model_dump(mode='json'))
+    assert response.status_code == 200
+    data = response.json()
+    assert data["region_id"] == "us-east-1"
+
+
+@pytest.mark.integration
+def test_get_all_region_health():
+    """Test getting all region health"""
+    client = TestClient(app)
+    response = client.get("/api/v1/health")
+    assert response.status_code == 200
+    data = response.json()
+    assert "region_health" in data
+
+
+@pytest.mark.integration
+def test_create_geographic_rule():
+    """Test creating geographic rule"""
+    client = TestClient(app)
+    rule = GeographicRule(
+        rule_id="geo_123",
+        source_regions=["us-east"],
+        target_regions=["us-east-1"],
+        priority=1,
+        latency_threshold_ms=50.0
+    )
+    response = client.post("/api/v1/geographic-rules/create", json=rule.model_dump())
+    assert response.status_code == 200
+    data = response.json()
+    assert data["rule_id"] == "geo_123"
+    assert data["status"] == "created"
+
+
+@pytest.mark.integration
+def test_create_duplicate_geographic_rule():
+    """Test creating duplicate geographic rule"""
+    client = TestClient(app)
+    rule = GeographicRule(
+        rule_id="geo_123",
+        source_regions=["us-east"],
+        target_regions=["us-east-1"],
+        priority=1,
+        latency_threshold_ms=50.0
+    )
+    client.post("/api/v1/geographic-rules/create", json=rule.model_dump())
+    
+    response = client.post("/api/v1/geographic-rules/create", json=rule.model_dump())
+    assert response.status_code == 400
+
+
+@pytest.mark.integration
+def test_get_optimal_region():
+    """Test getting optimal region"""
+    client = TestClient(app)
+    response = client.get("/api/v1/route/us-east")
+    assert response.status_code == 200
+    data = response.json()
+    assert "client_region" in data
+    assert "optimal_region" in data
+
+
+@pytest.mark.integration
+def test_get_optimal_region_with_rule():
+    """Test getting optimal region with specific rule"""
+    client = TestClient(app)
+    # Create a rule first
+    rule = LoadBalancingRule(
+        rule_id="rule_123",
+        name="Test Rule",
+        algorithm="weighted_round_robin",
+        target_regions=["us-east-1"],
+        weights={"us-east-1": 1.0},
+        health_check_path="/health",
+        failover_enabled=True,
+        session_affinity=False
+    )
+    client.post("/api/v1/rules/create", json=rule.model_dump())
+    
+    response = client.get("/api/v1/route/us-east?rule_id=rule_123")
+    assert response.status_code == 200
+    data = response.json()
+    assert data["rule_id"] == "rule_123"
+
+
+@pytest.mark.integration
+def test_record_balancing_metrics():
+    """Test recording balancing metrics"""
+    client = TestClient(app)
+    metrics = LoadBalancingMetrics(
+        balancer_id="lb_123",
+        timestamp=datetime.utcnow(),
+        total_requests=1000,
+        requests_per_region={"us-east-1": 500},
+        average_response_time=50.5,
+        error_rate=0.001,
+        throughput=100.0
+    )
+    response = client.post("/api/v1/metrics/record", json=metrics.model_dump(mode='json'))
+    assert response.status_code == 200
+    data = response.json()
+    assert data["metrics_id"]
+    assert data["status"] == "recorded"
+
+
+@pytest.mark.integration
+def test_get_balancing_metrics():
+    """Test getting balancing metrics"""
+    client = TestClient(app)
+    # Create a rule first
+    rule = LoadBalancingRule(
+        rule_id="rule_123",
+        name="Test Rule",
+        algorithm="weighted_round_robin",
+        target_regions=["us-east-1"],
+        weights={"us-east-1": 1.0},
+        health_check_path="/health",
+        failover_enabled=True,
+        session_affinity=False
+    )
+    client.post("/api/v1/rules/create", json=rule.model_dump())
+    
+    response = client.get("/api/v1/metrics/rule_123")
+    assert response.status_code == 200
+    data = response.json()
+    assert data["rule_id"] == "rule_123"
+
+
+@pytest.mark.integration
+def test_get_balancing_metrics_not_found():
+    """Test getting metrics for nonexistent rule"""
+    client = TestClient(app)
+    response = client.get("/api/v1/metrics/nonexistent")
+    assert response.status_code == 404
+
+
+@pytest.mark.integration
+def test_get_load_balancing_dashboard():
+    """Test getting load balancing dashboard"""
+    client = TestClient(app)
+    response = client.get("/api/v1/dashboard")
+    assert response.status_code == 200
+    data = response.json()
+    assert "dashboard" in data

apps/multi-region-load-balancer/tests/test_unit_multi_region_load_balancer.py

@@ -0,0 +1,120 @@
+"""Unit tests for multi-region load balancer service"""
+
+import pytest
+import sys
+import sys
+from pathlib import Path
+from datetime import datetime
+
+
+from main import app, LoadBalancingRule, RegionHealth, LoadBalancingMetrics, GeographicRule
+
+
+@pytest.mark.unit
+def test_app_initialization():
+    """Test that the FastAPI app initializes correctly"""
+    assert app is not None
+    assert app.title == "AITBC Multi-Region Load Balancer"
+    assert app.version == "1.0.0"
+
+
+@pytest.mark.unit
+def test_load_balancing_rule_model():
+    """Test LoadBalancingRule model"""
+    rule = LoadBalancingRule(
+        rule_id="rule_123",
+        name="Test Rule",
+        algorithm="weighted_round_robin",
+        target_regions=["us-east-1", "eu-west-1"],
+        weights={"us-east-1": 0.5, "eu-west-1": 0.5},
+        health_check_path="/health",
+        failover_enabled=True,
+        session_affinity=False
+    )
+    assert rule.rule_id == "rule_123"
+    assert rule.name == "Test Rule"
+    assert rule.algorithm == "weighted_round_robin"
+    assert rule.failover_enabled is True
+    assert rule.session_affinity is False
+
+
+@pytest.mark.unit
+def test_region_health_model():
+    """Test RegionHealth model"""
+    health = RegionHealth(
+        region_id="us-east-1",
+        status="healthy",
+        response_time_ms=45.5,
+        success_rate=0.99,
+        active_connections=100,
+        last_check=datetime.utcnow()
+    )
+    assert health.region_id == "us-east-1"
+    assert health.status == "healthy"
+    assert health.response_time_ms == 45.5
+    assert health.success_rate == 0.99
+    assert health.active_connections == 100
+
+
+@pytest.mark.unit
+def test_load_balancing_metrics_model():
+    """Test LoadBalancingMetrics model"""
+    metrics = LoadBalancingMetrics(
+        balancer_id="lb_123",
+        timestamp=datetime.utcnow(),
+        total_requests=1000,
+        requests_per_region={"us-east-1": 500, "eu-west-1": 500},
+        average_response_time=50.5,
+        error_rate=0.001,
+        throughput=100.0
+    )
+    assert metrics.balancer_id == "lb_123"
+    assert metrics.total_requests == 1000
+    assert metrics.average_response_time == 50.5
+    assert metrics.error_rate == 0.001
+
+
+@pytest.mark.unit
+def test_geographic_rule_model():
+    """Test GeographicRule model"""
+    rule = GeographicRule(
+        rule_id="geo_123",
+        source_regions=["us-east", "us-west"],
+        target_regions=["us-east-1", "us-west-1"],
+        priority=1,
+        latency_threshold_ms=50.0
+    )
+    assert rule.rule_id == "geo_123"
+    assert rule.source_regions == ["us-east", "us-west"]
+    assert rule.priority == 1
+    assert rule.latency_threshold_ms == 50.0
+
+
+@pytest.mark.unit
+def test_load_balancing_rule_empty_weights():
+    """Test LoadBalancingRule with empty weights"""
+    rule = LoadBalancingRule(
+        rule_id="rule_123",
+        name="Test Rule",
+        algorithm="round_robin",
+        target_regions=["us-east-1"],
+        weights={},
+        health_check_path="/health",
+        failover_enabled=False,
+        session_affinity=False
+    )
+    assert rule.weights == {}
+
+
+@pytest.mark.unit
+def test_region_health_negative_response_time():
+    """Test RegionHealth with negative response time"""
+    health = RegionHealth(
+        region_id="us-east-1",
+        status="healthy",
+        response_time_ms=-45.5,
+        success_rate=0.99,
+        active_connections=100,
+        last_check=datetime.utcnow()
+    )
+    assert health.response_time_ms == -45.5