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
66 lines
2.0 KiB
Python
Executable File
66 lines
2.0 KiB
Python
Executable File
import asyncio
|
|
from aitbc_chain.config import settings
|
|
from aitbc_chain.main import node_app
|
|
import httpx
|
|
import time
|
|
import os
|
|
|
|
# Create an alternate config just for this test process
|
|
os.environ["SUPPORTED_CHAINS"] = "ait-devnet,ait-testnet"
|
|
os.environ["DB_PATH"] = "./data/test_chain.db"
|
|
from aitbc_chain.config import settings as test_settings
|
|
|
|
# Make sure we use a clean DB for the test
|
|
if os.path.exists("./data/test_chain.db"):
|
|
os.remove("./data/test_chain.db")
|
|
if os.path.exists("./data/test_chain.db-journal"):
|
|
os.remove("./data/test_chain.db-journal")
|
|
|
|
async def run_test():
|
|
print(f"Testing with chains: {test_settings.supported_chains}")
|
|
|
|
# Start the app and the node
|
|
import uvicorn
|
|
from aitbc_chain.app import app
|
|
from threading import Thread
|
|
import requests
|
|
|
|
def run_server():
|
|
uvicorn.run(app, host="127.0.0.1", port=8181, log_level="error")
|
|
|
|
server_thread = Thread(target=run_server, daemon=True)
|
|
server_thread.start()
|
|
|
|
time.sleep(2) # Give server time to start
|
|
|
|
try:
|
|
# Check health which should report supported chains
|
|
resp = requests.get("http://127.0.0.1:8181/health")
|
|
print("Health status:", resp.json())
|
|
assert "ait-devnet" in resp.json()["supported_chains"]
|
|
assert "ait-testnet" in resp.json()["supported_chains"]
|
|
|
|
# The lifepan started the node with both chains.
|
|
# Wait for a couple blocks to be proposed
|
|
time.sleep(5)
|
|
|
|
# Check block head for devnet
|
|
resp = requests.get("http://127.0.0.1:8181/rpc/head?chain_id=ait-devnet")
|
|
print("Devnet head:", resp.json())
|
|
assert "hash" in resp.json()
|
|
|
|
# Check block head for testnet
|
|
resp = requests.get("http://127.0.0.1:8181/rpc/head?chain_id=ait-testnet")
|
|
print("Testnet head:", resp.json())
|
|
assert "hash" in resp.json()
|
|
|
|
print("SUCCESS! Multi-chain support is working.")
|
|
|
|
except Exception as e:
|
|
print("Test failed:", e)
|
|
|
|
if __name__ == "__main__":
|
|
import sys
|
|
sys.path.append('src')
|
|
asyncio.run(run_test())
|