1 files changed, 9 insertions, 10 deletions
diff --git a/crates/chat-crypto/tests/browser.ts b/crates/chat-crypto/tests/browser.ts
index 75095ce..fd16529 100644
--- a/crates/chat-crypto/tests/browser.ts
+++ b/crates/chat-crypto/tests/browser.ts
@@ -13,33 +13,32 @@
 // You should have received a copy of the GNU Affero General Public License
 // along with this program.  If not, see <https://www.gnu.org/licenses/>.
 
-import { createServer } from 'node:http';
 import { readFile } from 'node:fs/promises';
 import { resolve } from 'node:path';
 import assert from 'node:assert/strict';
 import { chromium } from 'playwright';
-import { serverOrigin } from '../../../tests/browser-support.js';
+import { asyncServer, closeBrowser, serverOrigin } from '../../../tests/browser-support.js';
 
 const assets = new Map<string, [string, string]>([
     ['/wasi.mjs', [resolve('var/build/typescript/crates/chat-crypto/browser/wasi.js'), 'text/javascript']],
     ['/probe.wasm', [resolve('target/wasm32-wasip1/release/stack-chat-crypto.wasm'), 'application/wasm']],
 ]);
-const server = createServer(async (request, response) => {
+const server = asyncServer(async (request, response) => {
     response.setHeader('Cache-Control', 'no-store');
     if (request.url === '/') {
         response.setHeader('Content-Type', 'text/html');
         response.end('<!doctype html><title>Local cryptographic compatibility test</title>');
         return;
     }
-    const asset = assets.get(request.url || '');
-    if (!asset) { response.writeHead(404); response.end(); return; }
+    const asset = assets.get((request.url ?? ''));
+    if (asset === undefined) { response.writeHead(404); response.end(); return; }
     response.setHeader('Content-Type', asset[1]);
     response.end(await readFile(asset[0]));
 });
 await new Promise<void>(resolveReady => server.listen(0, '127.0.0.1', resolveReady));
 const origin = serverOrigin(server);
 const browser = await chromium.launch({headless: true, channel: "chromium"});
-const deadline = setTimeout(() => browser.close(), 120000);
+const deadline = setTimeout(() => { closeBrowser(browser); }, 120000);
 try {
     const page = await browser.newPage();
     const external: string[] = [];
@@ -50,13 +49,13 @@ try {
     const result = await page.evaluate(async () => {
         const host = '/wasi.mjs';
         await import(host);
-        const { runProbe } = globalThis.stackChatWasi;
+        const { runProbe } = stackChatWasi;
         const bytes = await (await fetch('/probe.wasm', { cache: 'no-store' })).arrayBuffer();
         const valid = await runProbe(bytes);
         const again = await runProbe(bytes);
-        const getRandomValues = crypto.getRandomValues;
+        const getRandomValues = crypto.getRandomValues.bind(crypto);
         let rejected = false;
-        crypto.getRandomValues = () => { throw new Error('Entropy unavailable'); };
+        crypto.getRandomValues = (): never => { throw new Error('Entropy unavailable'); };
         try { await runProbe(bytes); } catch (error) { rejected = (error instanceof Error ? error.message : String(error)).includes("Entropy unavailable"); }
         finally { crypto.getRandomValues = getRandomValues; }
         const modified = new Uint8Array(bytes.slice(0));
@@ -80,7 +79,7 @@ try {
     assert.deepEqual(await page.evaluate(async () => indexedDB.databases()), []);
     assert.equal(await page.evaluate(async () => (await navigator.serviceWorker.getRegistrations()).length), 0);
     console.log(result.valid.output.trim());
-    console.log(`Browser CSPRNG: ${result.valid.randomCalls} calls, ${result.valid.randomBytes} bytes; missing entropy and unknown imports rejected.`);
+    console.log(`Browser CSPRNG: ${String(result.valid.randomCalls)} calls, ${String(result.valid.randomBytes)} bytes; missing entropy and unknown imports rejected.`);
     console.log('PASS: actual Chromium WASM execution; no external requests, cookies, web storage, or IndexedDB.');
 } finally {
     clearTimeout(deadline);