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:
@@ -0,0 +1,252 @@
|
||||
"""Integration tests for compliance service"""
|
||||
|
||||
import pytest
|
||||
import sys
|
||||
import sys
|
||||
from pathlib import Path
|
||||
from unittest.mock import Mock, patch
|
||||
from fastapi.testclient import TestClient
|
||||
from datetime import datetime
|
||||
|
||||
|
||||
from main import app, KYCRequest, ComplianceReport, TransactionMonitoring, kyc_records, compliance_reports, suspicious_transactions, compliance_rules
|
||||
|
||||
|
||||
@pytest.fixture(autouse=True)
|
||||
def reset_state():
|
||||
"""Reset global state before each test"""
|
||||
kyc_records.clear()
|
||||
compliance_reports.clear()
|
||||
suspicious_transactions.clear()
|
||||
compliance_rules.clear()
|
||||
yield
|
||||
kyc_records.clear()
|
||||
compliance_reports.clear()
|
||||
suspicious_transactions.clear()
|
||||
compliance_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 Compliance Service"
|
||||
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 "kyc_records" in data
|
||||
assert "compliance_reports" in data
|
||||
|
||||
|
||||
@pytest.mark.integration
|
||||
def test_submit_kyc():
|
||||
"""Test KYC submission"""
|
||||
client = TestClient(app)
|
||||
kyc = KYCRequest(
|
||||
user_id="user123",
|
||||
name="John Doe",
|
||||
email="john@example.com",
|
||||
document_type="passport",
|
||||
document_number="ABC123",
|
||||
address={"street": "123 Main St", "city": "New York", "country": "USA"}
|
||||
)
|
||||
response = client.post("/api/v1/kyc/submit", json=kyc.model_dump())
|
||||
assert response.status_code == 200
|
||||
data = response.json()
|
||||
assert data["user_id"] == "user123"
|
||||
assert data["status"] == "approved"
|
||||
assert data["risk_score"] == "low"
|
||||
|
||||
|
||||
@pytest.mark.integration
|
||||
def test_submit_duplicate_kyc():
|
||||
"""Test submitting duplicate KYC"""
|
||||
client = TestClient(app)
|
||||
kyc = KYCRequest(
|
||||
user_id="user123",
|
||||
name="John Doe",
|
||||
email="john@example.com",
|
||||
document_type="passport",
|
||||
document_number="ABC123",
|
||||
address={"street": "123 Main St", "city": "New York", "country": "USA"}
|
||||
)
|
||||
|
||||
# First submission
|
||||
client.post("/api/v1/kyc/submit", json=kyc.model_dump())
|
||||
|
||||
# Second submission should fail
|
||||
response = client.post("/api/v1/kyc/submit", json=kyc.model_dump())
|
||||
assert response.status_code == 400
|
||||
|
||||
|
||||
@pytest.mark.integration
|
||||
def test_get_kyc_status():
|
||||
"""Test getting KYC status"""
|
||||
client = TestClient(app)
|
||||
kyc = KYCRequest(
|
||||
user_id="user123",
|
||||
name="John Doe",
|
||||
email="john@example.com",
|
||||
document_type="passport",
|
||||
document_number="ABC123",
|
||||
address={"street": "123 Main St", "city": "New York", "country": "USA"}
|
||||
)
|
||||
|
||||
# Submit KYC first
|
||||
client.post("/api/v1/kyc/submit", json=kyc.model_dump())
|
||||
|
||||
# Get KYC status
|
||||
response = client.get("/api/v1/kyc/user123")
|
||||
assert response.status_code == 200
|
||||
data = response.json()
|
||||
assert data["user_id"] == "user123"
|
||||
assert data["status"] == "approved"
|
||||
|
||||
|
||||
@pytest.mark.integration
|
||||
def test_get_kyc_status_not_found():
|
||||
"""Test getting KYC status for nonexistent user"""
|
||||
client = TestClient(app)
|
||||
response = client.get("/api/v1/kyc/nonexistent")
|
||||
assert response.status_code == 404
|
||||
|
||||
|
||||
@pytest.mark.integration
|
||||
def test_list_kyc_records():
|
||||
"""Test listing KYC records"""
|
||||
client = TestClient(app)
|
||||
response = client.get("/api/v1/kyc")
|
||||
assert response.status_code == 200
|
||||
data = response.json()
|
||||
assert "kyc_records" in data
|
||||
assert "total_records" in data
|
||||
|
||||
|
||||
@pytest.mark.integration
|
||||
def test_create_compliance_report():
|
||||
"""Test creating compliance report"""
|
||||
client = TestClient(app)
|
||||
report = ComplianceReport(
|
||||
report_type="suspicious_activity",
|
||||
description="Suspicious transaction detected",
|
||||
severity="high",
|
||||
details={"transaction_id": "tx123"}
|
||||
)
|
||||
response = client.post("/api/v1/compliance/report", json=report.model_dump())
|
||||
assert response.status_code == 200
|
||||
data = response.json()
|
||||
assert data["severity"] == "high"
|
||||
assert data["status"] == "created"
|
||||
|
||||
|
||||
@pytest.mark.integration
|
||||
def test_list_compliance_reports():
|
||||
"""Test listing compliance reports"""
|
||||
client = TestClient(app)
|
||||
response = client.get("/api/v1/compliance/reports")
|
||||
assert response.status_code == 200
|
||||
data = response.json()
|
||||
assert "reports" in data
|
||||
assert "total_reports" in data
|
||||
|
||||
|
||||
@pytest.mark.integration
|
||||
def test_monitor_transaction():
|
||||
"""Test transaction monitoring"""
|
||||
client = TestClient(app)
|
||||
tx = TransactionMonitoring(
|
||||
transaction_id="tx123",
|
||||
user_id="user123",
|
||||
amount=1000.0,
|
||||
currency="BTC",
|
||||
counterparty="counterparty1",
|
||||
timestamp=datetime.utcnow()
|
||||
)
|
||||
response = client.post("/api/v1/monitoring/transaction", json=tx.model_dump(mode='json'))
|
||||
assert response.status_code == 200
|
||||
data = response.json()
|
||||
assert data["transaction_id"] == "tx123"
|
||||
assert "risk_score" in data
|
||||
|
||||
|
||||
@pytest.mark.integration
|
||||
def test_monitor_suspicious_transaction():
|
||||
"""Test monitoring suspicious transaction"""
|
||||
client = TestClient(app)
|
||||
tx = TransactionMonitoring(
|
||||
transaction_id="tx123",
|
||||
user_id="user123",
|
||||
amount=100000.0,
|
||||
currency="BTC",
|
||||
counterparty="high_risk_entity_1",
|
||||
timestamp=datetime.utcnow()
|
||||
)
|
||||
response = client.post("/api/v1/monitoring/transaction", json=tx.model_dump(mode='json'))
|
||||
assert response.status_code == 200
|
||||
data = response.json()
|
||||
assert data["status"] == "flagged"
|
||||
assert len(data["flags"]) > 0
|
||||
|
||||
|
||||
@pytest.mark.integration
|
||||
def test_list_monitored_transactions():
|
||||
"""Test listing monitored transactions"""
|
||||
client = TestClient(app)
|
||||
response = client.get("/api/v1/monitoring/transactions")
|
||||
assert response.status_code == 200
|
||||
data = response.json()
|
||||
assert "transactions" in data
|
||||
assert "total_transactions" in data
|
||||
|
||||
|
||||
@pytest.mark.integration
|
||||
def test_create_compliance_rule():
|
||||
"""Test creating compliance rule"""
|
||||
client = TestClient(app)
|
||||
rule_data = {
|
||||
"name": "High Value Transaction Rule",
|
||||
"description": "Flag transactions over $50,000",
|
||||
"type": "transaction_monitoring",
|
||||
"conditions": {"min_amount": 50000},
|
||||
"actions": ["flag", "report"],
|
||||
"severity": "high"
|
||||
}
|
||||
response = client.post("/api/v1/rules/create", json=rule_data)
|
||||
assert response.status_code == 200
|
||||
data = response.json()
|
||||
assert data["name"] == "High Value Transaction Rule"
|
||||
assert data["active"] is True
|
||||
|
||||
|
||||
@pytest.mark.integration
|
||||
def test_list_compliance_rules():
|
||||
"""Test listing compliance 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_compliance_dashboard():
|
||||
"""Test compliance dashboard"""
|
||||
client = TestClient(app)
|
||||
response = client.get("/api/v1/dashboard")
|
||||
assert response.status_code == 200
|
||||
data = response.json()
|
||||
assert "summary" in data
|
||||
assert "risk_distribution" in data
|
||||
assert "recent_activity" in data
|
||||
Reference in New Issue
Block a user