```
chore: remove obsolete payment architecture and integration test documentation - Remove AITBC_PAYMENT_ARCHITECTURE.md (dual-currency system documentation) - Remove IMPLEMENTATION_COMPLETE_SUMMARY.md (integration test completion summary) - Remove INTEGRATION_TEST_FIXES.md (test fixes documentation) - Remove INTEGRATION_TEST_UPDATES.md (real features implementation notes) - Remove PAYMENT_INTEGRATION_COMPLETE.md (wallet-coordinator integration docs) - Remove WALLET_COORDINATOR_INTEGRATION.md (payment
This commit is contained in:
oib
203
scripts/testing/test_block_import.py
Normal file
203
scripts/testing/test_block_import.py
Normal file
@@ -0,0 +1,203 @@
|
||||
#!/usr/bin/env python3
|
||||
"""
|
||||
Test script for block import endpoint
|
||||
Tests the /rpc/blocks/import POST endpoint functionality
|
||||
"""
|
||||
|
||||
import json
|
||||
import hashlib
|
||||
from datetime import datetime
|
||||
|
||||
# Test configuration
|
||||
BASE_URL = "https://aitbc.bubuit.net/rpc"
|
||||
CHAIN_ID = "ait-devnet"
|
||||
|
||||
def compute_block_hash(height, parent_hash, timestamp):
|
||||
"""Compute block hash using the same algorithm as PoA proposer"""
|
||||
payload = f"{CHAIN_ID}|{height}|{parent_hash}|{timestamp}".encode()
|
||||
return "0x" + hashlib.sha256(payload).hexdigest()
|
||||
|
||||
def test_block_import():
|
||||
"""Test the block import endpoint with various scenarios"""
|
||||
import requests
|
||||
|
||||
print("Testing Block Import Endpoint")
|
||||
print("=" * 50)
|
||||
|
||||
# Test 1: Invalid height (0)
|
||||
print("\n1. Testing invalid height (0)...")
|
||||
response = requests.post(
|
||||
f"{BASE_URL}/blocks/import",
|
||||
json={
|
||||
"height": 0,
|
||||
"hash": "0x123",
|
||||
"parent_hash": "0x00",
|
||||
"proposer": "test",
|
||||
"timestamp": "2026-01-29T10:20:00",
|
||||
"tx_count": 0
|
||||
}
|
||||
)
|
||||
print(f"Status: {response.status_code}")
|
||||
print(f"Response: {response.json()}")
|
||||
assert response.status_code == 422, "Should return validation error for height 0"
|
||||
print("✓ Correctly rejected height 0")
|
||||
|
||||
# Test 2: Block already exists with different hash
|
||||
print("\n2. Testing block conflict...")
|
||||
response = requests.post(
|
||||
f"{BASE_URL}/blocks/import",
|
||||
json={
|
||||
"height": 1,
|
||||
"hash": "0xinvalidhash",
|
||||
"parent_hash": "0x00",
|
||||
"proposer": "test",
|
||||
"timestamp": "2026-01-29T10:20:00",
|
||||
"tx_count": 0
|
||||
}
|
||||
)
|
||||
print(f"Status: {response.status_code}")
|
||||
print(f"Response: {response.json()}")
|
||||
assert response.status_code == 409, "Should return conflict for existing height with different hash"
|
||||
print("✓ Correctly detected block conflict")
|
||||
|
||||
# Test 3: Import existing block with correct hash
|
||||
print("\n3. Testing import of existing block with correct hash...")
|
||||
# Get actual block data
|
||||
response = requests.get(f"{BASE_URL}/blocks/1")
|
||||
block_data = response.json()
|
||||
|
||||
response = requests.post(
|
||||
f"{BASE_URL}/blocks/import",
|
||||
json={
|
||||
"height": block_data["height"],
|
||||
"hash": block_data["hash"],
|
||||
"parent_hash": block_data["parent_hash"],
|
||||
"proposer": block_data["proposer"],
|
||||
"timestamp": block_data["timestamp"],
|
||||
"tx_count": block_data["tx_count"]
|
||||
}
|
||||
)
|
||||
print(f"Status: {response.status_code}")
|
||||
print(f"Response: {response.json()}")
|
||||
assert response.status_code == 200, "Should accept existing block with correct hash"
|
||||
assert response.json()["status"] == "exists", "Should return 'exists' status"
|
||||
print("✓ Correctly handled existing block")
|
||||
|
||||
# Test 4: Invalid block hash (with valid parent)
|
||||
print("\n4. Testing invalid block hash...")
|
||||
# Get current head to use as parent
|
||||
response = requests.get(f"{BASE_URL}/head")
|
||||
head = response.json()
|
||||
|
||||
timestamp = "2026-01-29T10:20:00"
|
||||
parent_hash = head["hash"] # Use actual parent hash
|
||||
height = head["height"] + 1000 # Use high height to avoid conflicts
|
||||
invalid_hash = "0xinvalid"
|
||||
|
||||
response = requests.post(
|
||||
f"{BASE_URL}/blocks/import",
|
||||
json={
|
||||
"height": height,
|
||||
"hash": invalid_hash,
|
||||
"parent_hash": parent_hash,
|
||||
"proposer": "test",
|
||||
"timestamp": timestamp,
|
||||
"tx_count": 0
|
||||
}
|
||||
)
|
||||
print(f"Status: {response.status_code}")
|
||||
print(f"Response: {response.json()}")
|
||||
assert response.status_code == 400, "Should reject invalid hash"
|
||||
assert "Invalid block hash" in response.json()["detail"], "Should mention invalid hash"
|
||||
print("✓ Correctly rejected invalid hash")
|
||||
|
||||
# Test 5: Valid hash but parent not found
|
||||
print("\n5. Testing valid hash but parent not found...")
|
||||
height = head["height"] + 2000 # Use different height
|
||||
parent_hash = "0xnonexistentparent"
|
||||
timestamp = "2026-01-29T10:20:00"
|
||||
valid_hash = compute_block_hash(height, parent_hash, timestamp)
|
||||
|
||||
response = requests.post(
|
||||
f"{BASE_URL}/blocks/import",
|
||||
json={
|
||||
"height": height,
|
||||
"hash": valid_hash,
|
||||
"parent_hash": parent_hash,
|
||||
"proposer": "test",
|
||||
"timestamp": timestamp,
|
||||
"tx_count": 0
|
||||
}
|
||||
)
|
||||
print(f"Status: {response.status_code}")
|
||||
print(f"Response: {response.json()}")
|
||||
assert response.status_code == 400, "Should reject when parent not found"
|
||||
assert "Parent block not found" in response.json()["detail"], "Should mention parent not found"
|
||||
print("✓ Correctly rejected missing parent")
|
||||
|
||||
# Test 6: Valid block with transactions and receipts
|
||||
print("\n6. Testing valid block with transactions...")
|
||||
# Get current head to use as parent
|
||||
response = requests.get(f"{BASE_URL}/head")
|
||||
head = response.json()
|
||||
|
||||
height = head["height"] + 1
|
||||
parent_hash = head["hash"]
|
||||
timestamp = datetime.utcnow().isoformat() + "Z"
|
||||
valid_hash = compute_block_hash(height, parent_hash, timestamp)
|
||||
|
||||
test_block = {
|
||||
"height": height,
|
||||
"hash": valid_hash,
|
||||
"parent_hash": parent_hash,
|
||||
"proposer": "test-proposer",
|
||||
"timestamp": timestamp,
|
||||
"tx_count": 1,
|
||||
"transactions": [{
|
||||
"tx_hash": f"0xtx{height}",
|
||||
"sender": "0xsender",
|
||||
"recipient": "0xreceiver",
|
||||
"payload": {"to": "0xreceiver", "amount": 1000000}
|
||||
}],
|
||||
"receipts": [{
|
||||
"receipt_id": f"rx{height}",
|
||||
"job_id": f"job{height}",
|
||||
"payload": {"result": "success"},
|
||||
"miner_signature": "0xminer",
|
||||
"coordinator_attestations": ["0xatt1"],
|
||||
"minted_amount": 100,
|
||||
"recorded_at": timestamp
|
||||
}]
|
||||
}
|
||||
|
||||
response = requests.post(
|
||||
f"{BASE_URL}/blocks/import",
|
||||
json=test_block
|
||||
)
|
||||
print(f"Status: {response.status_code}")
|
||||
print(f"Response: {response.json()}")
|
||||
assert response.status_code == 200, "Should accept valid block with transactions"
|
||||
assert response.json()["status"] == "imported", "Should return 'imported' status"
|
||||
print("✓ Successfully imported block with transactions")
|
||||
|
||||
# Verify the block was imported
|
||||
print("\n7. Verifying imported block...")
|
||||
response = requests.get(f"{BASE_URL}/blocks/{height}")
|
||||
assert response.status_code == 200, "Should be able to retrieve imported block"
|
||||
imported_block = response.json()
|
||||
assert imported_block["hash"] == valid_hash, "Hash should match"
|
||||
assert imported_block["tx_count"] == 1, "Should have 1 transaction"
|
||||
print("✓ Block successfully imported and retrievable")
|
||||
|
||||
print("\n" + "=" * 50)
|
||||
print("All tests passed! ✅")
|
||||
print("\nBlock import endpoint is fully functional with:")
|
||||
print("- ✓ Input validation")
|
||||
print("- ✓ Hash validation")
|
||||
print("- ✓ Parent block verification")
|
||||
print("- ✓ Conflict detection")
|
||||
print("- ✓ Transaction and receipt import")
|
||||
print("- ✓ Proper error handling")
|
||||
|
||||
if __name__ == "__main__":
|
||||
test_block_import()
|
||||
Reference in New Issue
Block a user