Merge commit 'd803bfe2b1fe7f5e219e50ac20d6801a0a58ac75' as 'vendor/ruvector'
This commit is contained in:
ruv
334
vendor/ruvector/examples/edge-net/tests/qdag-persistence.test.ts
vendored
Normal file
334
vendor/ruvector/examples/edge-net/tests/qdag-persistence.test.ts
vendored
Normal file
@@ -0,0 +1,334 @@
|
||||
/**
|
||||
* QDAG Credit Persistence Test Suite
|
||||
*
|
||||
* Tests the Edge-Net QDAG credit persistence system to verify:
|
||||
* - Credits persist across sessions in Firestore
|
||||
* - Same public key returns same balance from different node IDs
|
||||
* - Ledger sync correctly retrieves balances from QDAG
|
||||
*/
|
||||
|
||||
import WebSocket from 'ws';
|
||||
|
||||
interface EdgeNetMessage {
|
||||
type: string;
|
||||
from?: string;
|
||||
to?: string;
|
||||
payload?: any;
|
||||
[key: string]: any;
|
||||
}
|
||||
|
||||
interface LedgerSyncResponse {
|
||||
type: 'ledger-sync';
|
||||
credits: number;
|
||||
publicKey: string;
|
||||
timestamp: number;
|
||||
}
|
||||
|
||||
interface TestResult {
|
||||
success: boolean;
|
||||
balance: number | null;
|
||||
error?: string;
|
||||
timestamp: number;
|
||||
}
|
||||
|
||||
const RELAY_URL = 'wss://edge-net-relay-875130704813.us-central1.run.app';
|
||||
const TEST_PUBLIC_KEY = '38a3bcd1732fe04c4a0358a058fd8f81ed8325fcf6f372b91aab0f983f3a2ca5';
|
||||
const CONNECTION_TIMEOUT = 10000; // 10 seconds
|
||||
const RESPONSE_TIMEOUT = 15000; // 15 seconds
|
||||
|
||||
/**
|
||||
* Create a WebSocket connection with timeout
|
||||
*/
|
||||
function connectWithTimeout(url: string, timeout: number): Promise<WebSocket> {
|
||||
return new Promise((resolve, reject) => {
|
||||
const ws = new WebSocket(url);
|
||||
const timer = setTimeout(() => {
|
||||
ws.close();
|
||||
reject(new Error('Connection timeout'));
|
||||
}, timeout);
|
||||
|
||||
ws.on('open', () => {
|
||||
clearTimeout(timer);
|
||||
resolve(ws);
|
||||
});
|
||||
|
||||
ws.on('error', (error) => {
|
||||
clearTimeout(timer);
|
||||
reject(error);
|
||||
});
|
||||
});
|
||||
}
|
||||
|
||||
/**
|
||||
* Wait for a specific message type
|
||||
*/
|
||||
function waitForMessage(
|
||||
ws: WebSocket,
|
||||
messageType: string,
|
||||
timeout: number
|
||||
): Promise<EdgeNetMessage> {
|
||||
return new Promise((resolve, reject) => {
|
||||
const timer = setTimeout(() => {
|
||||
reject(new Error(`Timeout waiting for ${messageType}`));
|
||||
}, timeout);
|
||||
|
||||
const handler = (data: WebSocket.Data) => {
|
||||
try {
|
||||
const message: EdgeNetMessage = JSON.parse(data.toString());
|
||||
if (message.type === messageType) {
|
||||
clearTimeout(timer);
|
||||
ws.removeListener('message', handler);
|
||||
resolve(message);
|
||||
}
|
||||
} catch (error) {
|
||||
// Ignore parse errors, wait for valid message
|
||||
}
|
||||
};
|
||||
|
||||
ws.on('message', handler);
|
||||
});
|
||||
}
|
||||
|
||||
/**
|
||||
* Test 1: Connect to relay and verify connection
|
||||
*/
|
||||
async function testConnection(): Promise<TestResult> {
|
||||
console.log('\n📡 Test 1: Testing relay connection...');
|
||||
const startTime = Date.now();
|
||||
|
||||
try {
|
||||
const ws = await connectWithTimeout(RELAY_URL, CONNECTION_TIMEOUT);
|
||||
console.log('✅ Successfully connected to relay');
|
||||
ws.close();
|
||||
|
||||
return {
|
||||
success: true,
|
||||
balance: null,
|
||||
timestamp: Date.now() - startTime
|
||||
};
|
||||
} catch (error) {
|
||||
console.error('❌ Connection failed:', error);
|
||||
return {
|
||||
success: false,
|
||||
balance: null,
|
||||
error: error instanceof Error ? error.message : String(error),
|
||||
timestamp: Date.now() - startTime
|
||||
};
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* Test 2: Register with public key and request ledger sync
|
||||
*/
|
||||
async function testLedgerSync(nodeId: string): Promise<TestResult> {
|
||||
console.log(`\n💳 Test 2: Testing ledger sync with node ID: ${nodeId.substring(0, 8)}...`);
|
||||
const startTime = Date.now();
|
||||
|
||||
try {
|
||||
// Connect to relay
|
||||
const ws = await connectWithTimeout(RELAY_URL, CONNECTION_TIMEOUT);
|
||||
console.log('✅ Connected to relay');
|
||||
|
||||
// Register with public key
|
||||
const registerMessage = {
|
||||
type: 'register',
|
||||
nodeId: nodeId,
|
||||
publicKey: TEST_PUBLIC_KEY,
|
||||
capabilities: ['test'],
|
||||
timestamp: Date.now()
|
||||
};
|
||||
|
||||
ws.send(JSON.stringify(registerMessage));
|
||||
console.log('📤 Sent registration message');
|
||||
|
||||
// Wait for welcome message (registration confirmation)
|
||||
await waitForMessage(ws, 'welcome', RESPONSE_TIMEOUT);
|
||||
console.log('✅ Registration confirmed (received welcome)');
|
||||
|
||||
// Request ledger sync
|
||||
const ledgerSyncRequest = {
|
||||
type: 'ledger_sync',
|
||||
nodeId: nodeId,
|
||||
publicKey: TEST_PUBLIC_KEY
|
||||
};
|
||||
|
||||
ws.send(JSON.stringify(ledgerSyncRequest));
|
||||
console.log('📤 Sent ledger sync request');
|
||||
|
||||
// Wait for ledger sync response
|
||||
const response = await waitForMessage(ws, 'ledger_sync_response', RESPONSE_TIMEOUT);
|
||||
const ledgerData = response.ledger as any;
|
||||
const credits = BigInt(ledgerData.earned || '0') - BigInt(ledgerData.spent || '0');
|
||||
|
||||
console.log(`✅ Received ledger sync response:`);
|
||||
console.log(` Earned: ${ledgerData.earned}`);
|
||||
console.log(` Spent: ${ledgerData.spent}`);
|
||||
console.log(` Available: ${credits.toString()} credits`);
|
||||
|
||||
ws.close();
|
||||
|
||||
return {
|
||||
success: true,
|
||||
balance: Number(credits),
|
||||
timestamp: Date.now() - startTime
|
||||
};
|
||||
} catch (error) {
|
||||
console.error('❌ Ledger sync failed:', error);
|
||||
return {
|
||||
success: false,
|
||||
balance: null,
|
||||
error: error instanceof Error ? error.message : String(error),
|
||||
timestamp: Date.now() - startTime
|
||||
};
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* Test 3: Verify same balance from different node IDs
|
||||
*/
|
||||
async function testBalanceConsistency(): Promise<TestResult> {
|
||||
console.log('\n🔄 Test 3: Testing balance consistency across node IDs...');
|
||||
const startTime = Date.now();
|
||||
|
||||
try {
|
||||
// Generate multiple random node IDs
|
||||
const nodeIds = [
|
||||
`test-node-${Math.random().toString(36).substring(7)}`,
|
||||
`test-node-${Math.random().toString(36).substring(7)}`,
|
||||
`test-node-${Math.random().toString(36).substring(7)}`
|
||||
];
|
||||
|
||||
const balances: number[] = [];
|
||||
|
||||
// Test with each node ID
|
||||
for (const nodeId of nodeIds) {
|
||||
console.log(`\n Testing with node ID: ${nodeId}`);
|
||||
const result = await testLedgerSync(nodeId);
|
||||
|
||||
if (!result.success || result.balance === null) {
|
||||
throw new Error(`Failed to get balance for node ${nodeId}`);
|
||||
}
|
||||
|
||||
balances.push(result.balance);
|
||||
console.log(` Balance: ${result.balance} credits`);
|
||||
|
||||
// Wait a bit between requests
|
||||
await new Promise(resolve => setTimeout(resolve, 1000));
|
||||
}
|
||||
|
||||
// Verify all balances are the same
|
||||
const allSame = balances.every(balance => balance === balances[0]);
|
||||
|
||||
if (allSame) {
|
||||
console.log(`\n✅ Balance consistency verified: All node IDs returned ${balances[0]} credits`);
|
||||
return {
|
||||
success: true,
|
||||
balance: balances[0],
|
||||
timestamp: Date.now() - startTime
|
||||
};
|
||||
} else {
|
||||
console.error(`\n❌ Balance inconsistency detected: ${balances.join(', ')}`);
|
||||
return {
|
||||
success: false,
|
||||
balance: null,
|
||||
error: `Inconsistent balances: ${balances.join(', ')}`,
|
||||
timestamp: Date.now() - startTime
|
||||
};
|
||||
}
|
||||
} catch (error) {
|
||||
console.error('❌ Balance consistency test failed:', error);
|
||||
return {
|
||||
success: false,
|
||||
balance: null,
|
||||
error: error instanceof Error ? error.message : String(error),
|
||||
timestamp: Date.now() - startTime
|
||||
};
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* Main test runner
|
||||
*/
|
||||
async function runTests() {
|
||||
console.log('🧪 Edge-Net QDAG Credit Persistence Test Suite');
|
||||
console.log('='.repeat(60));
|
||||
console.log(`📋 Testing public key: ${TEST_PUBLIC_KEY}`);
|
||||
console.log(`🌐 Relay URL: ${RELAY_URL}`);
|
||||
console.log('='.repeat(60));
|
||||
|
||||
const results: TestResult[] = [];
|
||||
|
||||
// Test 1: Connection
|
||||
const connectionResult = await testConnection();
|
||||
results.push(connectionResult);
|
||||
|
||||
if (!connectionResult.success) {
|
||||
console.error('\n❌ Connection test failed. Aborting remaining tests.');
|
||||
printSummary(results);
|
||||
return;
|
||||
}
|
||||
|
||||
// Test 2: Single ledger sync
|
||||
const nodeId = `test-node-${Math.random().toString(36).substring(7)}`;
|
||||
const ledgerSyncResult = await testLedgerSync(nodeId);
|
||||
results.push(ledgerSyncResult);
|
||||
|
||||
if (!ledgerSyncResult.success) {
|
||||
console.error('\n❌ Ledger sync test failed. Aborting remaining tests.');
|
||||
printSummary(results);
|
||||
return;
|
||||
}
|
||||
|
||||
// Test 3: Balance consistency
|
||||
const consistencyResult = await testBalanceConsistency();
|
||||
results.push(consistencyResult);
|
||||
|
||||
// Print summary
|
||||
printSummary(results);
|
||||
}
|
||||
|
||||
/**
|
||||
* Print test summary
|
||||
*/
|
||||
function printSummary(results: TestResult[]) {
|
||||
console.log('\n' + '='.repeat(60));
|
||||
console.log('📊 Test Summary');
|
||||
console.log('='.repeat(60));
|
||||
|
||||
const totalTests = results.length;
|
||||
const passedTests = results.filter(r => r.success).length;
|
||||
const failedTests = totalTests - passedTests;
|
||||
|
||||
results.forEach((result, index) => {
|
||||
const status = result.success ? '✅ PASS' : '❌ FAIL';
|
||||
const testName = ['Connection Test', 'Ledger Sync Test', 'Balance Consistency Test'][index];
|
||||
console.log(`\n${status} - ${testName}`);
|
||||
console.log(` Duration: ${result.timestamp}ms`);
|
||||
if (result.balance !== null) {
|
||||
console.log(` Balance: ${result.balance} credits`);
|
||||
}
|
||||
if (result.error) {
|
||||
console.log(` Error: ${result.error}`);
|
||||
}
|
||||
});
|
||||
|
||||
console.log('\n' + '='.repeat(60));
|
||||
console.log(`Total: ${totalTests} | Passed: ${passedTests} | Failed: ${failedTests}`);
|
||||
console.log('='.repeat(60));
|
||||
|
||||
// Report final balance
|
||||
const balanceResult = results.find(r => r.balance !== null);
|
||||
if (balanceResult && balanceResult.balance !== null) {
|
||||
console.log(`\n🏆 Final Balance for public key ${TEST_PUBLIC_KEY.substring(0, 16)}...:`);
|
||||
console.log(` ${balanceResult.balance} credits`);
|
||||
}
|
||||
|
||||
// Exit with appropriate code
|
||||
process.exit(failedTests > 0 ? 1 : 0);
|
||||
}
|
||||
|
||||
// Run tests
|
||||
runTests().catch(error => {
|
||||
console.error('Fatal error running tests:', error);
|
||||
process.exit(1);
|
||||
});
|
||||
Reference in New Issue
Block a user