1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
|
// A relay socket that dies while the app is on screen must not leave the
// room deaf. Every relay socket under an open coinos room is made to ERROR
// (what a wifi hop or a phone freezing the tab does): nostr-tools drops an
// errored relay from its pool with every subscription it carried and never
// dials it again, so it's the watchdog's job to notice and rebuild — with no
// 'online' event and no resume hook to lean on. Afterwards the rebuilt
// subscriptions must stay quiet (no further rebuilds) and the sockets open.
//
// Chrome's offline emulation leaves established sockets alone, so the
// sockets are reached through a registry shimmed onto window.WebSocket.
//
// Run: bun tools/chat-watchdog-test.js
import puppeteer from './puppeteer.js';
import { generateMnemonic } from '@scure/bip39';
import { wordlist } from '@scure/bip39/wordlists/english';
import { buildHtml } from '../build.js';
const sleep = (ms) => new Promise((r) => setTimeout(r, ms));
let ok = true;
const check = (n, c, d = '') => { console.log(` ${c ? '✓' : '✗'} ${n}${d ? ' — ' + d : ''}`); if (!c) ok = false; };
const html = await buildHtml({ minify: true, pwa: false });
const server = Bun.serve({ port: 5232, fetch: () => new Response(html, { headers: { 'content-type': 'text/html' } }) });
const browser = await puppeteer.launch({ relays: 'read', executablePath: '/usr/bin/google-chrome', headless: 'new', args: ['--no-sandbox'] });
const page = await browser.newPage();
// every socket the page opens, so the test can kill them the way the OS does
await page.evaluateOnNewDocument(() => {
const Real = window.WebSocket;
const live = new Set();
window.__sockets = live;
window.WebSocket = new Proxy(Real, {
construct(target, args) {
const ws = new target(...args);
live.add(ws);
ws.addEventListener('close', () => live.delete(ws));
return ws;
},
});
});
const warns = [];
page.on('console', (m) => { if (m.type() === 'warn' || m.type() === 'warning') warns.push({ t: Date.now(), text: m.text() }); });
page.on('pageerror', (e) => console.log(' [pageerror]', e.message));
const click = (sel, t) => page.evaluate((s, x) => { const e = [...document.querySelectorAll(s)].find((n) => n.textContent.trim().toLowerCase().includes(x.toLowerCase())); if (e) { e.click(); return true; } return false; }, sel, t);
const waitText = async (x, ms = 25000) => { for (let i = 0; i < ms / 250; i++) { if ((await page.evaluate(() => document.body.innerText)).toLowerCase().includes(x.toLowerCase())) return true; await sleep(250); } return false; };
const rebuilds = (since = 0) => warns.filter((w) => w.t >= since && /resubscribing/.test(w.text));
const waitRebuild = async (since, ms) => { for (let i = 0; i < ms / 500; i++) { if (rebuilds(since).length) return true; await sleep(500); } return false; };
try {
await page.setViewport({ width: 420, height: 860 });
await page.goto('http://localhost:5232/', { waitUntil: 'domcontentloaded' });
await page.evaluate(() => localStorage.setItem('btc-wallet-network', 'regtest'));
await page.reload({ waitUntil: 'domcontentloaded' });
await sleep(400);
await click('button', 'Create a new wallet'); await sleep(300);
await click('button', 'Have an existing seed'); await sleep(300);
await page.waitForSelector('textarea');
await page.type('textarea', generateMnemonic(wordlist));
await click('button', 'Open wallet');
await waitText('receive', 15000);
await page.evaluate(() => [...document.querySelectorAll('button')].find((b) => /message/i.test(b.getAttribute('aria-label') || ''))?.click());
await sleep(800);
await page.evaluate(() => { const e = [...document.querySelectorAll('.chat-thread-row')].find((n) => (n.querySelector('.chat-name') || {}).textContent === 'coinos'); if (e) e.click(); });
check('the coinos room opens with messages', await waitText('GM Coinos community', 30000));
await sleep(12000); // a full watchdog tick with everything healthy
check('a healthy room triggers no rebuild', rebuilds().length === 0, rebuilds().map((w) => w.text).join(' | ').slice(0, 200) || 'none');
console.log('\n[every relay socket errors under the open room]');
const before = await page.evaluate(() => [...window.__sockets].filter((w) => w.readyState === 1).map((w) => w.url));
check('relay sockets are open to begin with', before.length >= 2, before.join(' '));
const cut = Date.now();
await page.evaluate(() => {
for (const ws of [...window.__sockets]) {
// the handler nostr-tools installed: an error drops the relay from
// the pool and closes its subscriptions for good
try { ws.onerror && ws.onerror(new Event('error')); } catch {}
try { ws.close(); } catch {}
}
});
check('the watchdog notices and rebuilds within seconds', await waitRebuild(cut, 15000),
rebuilds(cut).map((w) => w.text).join(' | ').slice(0, 200));
await sleep(6000);
const after = await page.evaluate(() => [...window.__sockets].filter((w) => w.readyState === 1).map((w) => w.url));
check('the relays are dialed again', after.length >= 2, after.join(' '));
console.log('\n[and then it stays quiet]');
const settled = Date.now();
await sleep(25000);
const late = rebuilds(settled);
check('the rebuilt subscriptions stay up (no further rebuilds)', late.length === 0, late.map((w) => w.text).join(' | ').slice(0, 200));
const fresh = await page.evaluate(() => document.querySelectorAll('.chat-bubble').length);
check('the room still shows its messages', fresh > 5, `${fresh} bubbles`);
} catch (e) { console.log('ERROR', e); ok = false; }
await browser.close();
server.stop(true);
console.log(ok ? '\nALL GOOD' : '\nFAILURES');
process.exit(ok ? 0 : 1);
|
