open-standards/wai/wai-quantum-js/test/smoke.mjs

// The package as an installer leaves it: its bindings and its WebAssembly, loaded in
// Node with nothing else installed.
//
//   node wai/wai-quantum-js/test/smoke.mjs PACKAGE_DIR
//
// Prints one line per check and exits non-zero on the first that fails.
import assert from 'node:assert/strict';
import { randomBytes } from 'node:crypto';
import { readFile } from 'node:fs/promises';
import { join, resolve } from 'node:path';
import { pathToFileURL } from 'node:url';

const dir = resolve(process.argv[2] ?? '.');
const pkg = JSON.parse(await readFile(join(dir, 'package.json'), 'utf8'));
const wq = await import(pathToFileURL(join(dir, pkg.main)).href);
// In Node the module is given its WebAssembly's bytes; a page lets it fetch them.
await wq.default({ module_or_path: await readFile(join(dir, 'wai_quantum_bg.wasm')) });
console.log('node', process.versions.node, pkg.name, pkg.version);

const BELL = 'OPENQASM 3.0; qubit[2] q; bit[2] c; h q[0]; cx q[0], q[1]; c = measure q;';

// A job run with a fresh key, and its receipt checked.
const engine = new wq.QuantumEngine(randomBytes(32), 256);
const signer = engine.signer_did();
assert.match(signer, /^did:key:z6Mk/, signer);
const job = JSON.parse(engine.run(JSON.stringify({ qasm: BELL, shots: 1000, seed: 7 })));
assert.equal(job.status, 'done', JSON.stringify(job).slice(0, 400));
const counts = job.result.counts;
assert.ok(Object.keys(counts).length > 0 && Object.keys(counts).every((k) => k === '00' || k === '11'), JSON.stringify(counts));
const v = JSON.parse(wq.engine_verify(JSON.stringify(job)));
assert.equal(v.valid, true, JSON.stringify(v));
assert.equal(v.signer, signer, JSON.stringify(v));
console.log(`run and verify: ${job.method}, ${Object.values(counts).reduce((a, b) => a + b, 0)} shots, signed by this engine's key`);

// A result changed after it was sealed does not verify.
const forged = structuredClone(job);
const k = Object.keys(forged.result.counts)[0];
forged.result.counts[k] += 1;
assert.notEqual(JSON.parse(wq.engine_verify(JSON.stringify(forged))).valid, true);
console.log('a forged result does not verify');

// A program read and drawn.
const lint = JSON.parse(wq.engine_lint(BELL));
assert.equal(lint.ok, true, JSON.stringify(lint));
assert.equal(lint.qubits, 2, JSON.stringify(lint));
const drawn = JSON.parse(wq.engine_draw(BELL, 80));
assert.ok(typeof drawn.text === 'string' && drawn.text.trim().length > 0, JSON.stringify(drawn));
console.log(`lint and draw: ${lint.qubits} qubits, ${lint.gates} gates; drawn in ${drawn.text.trimEnd().split('\n').length} lines`);

// A program the engine refuses says why, with its diagnostic code.
const WHILE = 'OPENQASM 3.0; qubit[1] q; bit[1] c; c[0] = measure q[0]; while (c[0]) { x q[0]; c[0] = measure q[0]; }';
const refused = JSON.parse(engine.run(JSON.stringify({ qasm: WHILE, shots: 10 })));
assert.ok(typeof refused.error === 'string' && /^Q\d{3}$/.test(refused.code ?? ''), JSON.stringify(refused));
console.log(`a refused program says why: [${refused.code}] ${refused.meaning}`);

console.log('smoke: every check passed');