test_worker.js

const { Worker } = require("worker_threads");
const path = require("path");

class SQLWorker {
  constructor(worker) {
    this.worker = worker;
    this.callbacks = new Map();
    this.nextId = 1;

    this.worker.stderr.on('data', (data) => {
      console.log(data);
    });

    this.worker.stdout.on('data', (data) => {
      console.log(data);
    });

    this.worker.on('message', (data) => {
      if (data.error) {
        console.log("Worker error: ", data.error);
        for (const callback of this.callbacks.values()) {
          callback.reject(data.error);
        }
        return;
      }
      const callback = this.callbacks.get(data.id);
      if (callback) {
        this.callbacks.delete(data.id);
        callback.resolve(data);
      } else {
        console.log("Received message from worker but no callback found for id", data);
      }
    });

    this.worker.on('error', (err) => {
      console.log("Worker error", err);
      for (const callback of this.callbacks.values()) {
        callback.reject(err);
      }
      this.callbacks.clear();
    });
  }

  static async fromFile(file) {
    // Create a worker directly from the file
    const worker = new Worker(file);
    return new SQLWorker(worker);
  }

  async postMessage(msg) {
    return new Promise((resolve, reject) => {
      const id = msg.id || this.nextId++;
      const messageWithId = { ...msg, id };
      
      this.callbacks.set(id, { resolve, reject });
      
      // Set a timeout to reject the promise if no response is received
      setTimeout(() => {
        if (this.callbacks.has(id)) {
          this.callbacks.delete(id);
          reject(new Error("Worker response timeout"));
        }
      }, 20000);
      
      // Send the message to the worker
      this.worker.postMessage(messageWithId);
    });
  }
}

exports.test = async function test(SQL, assert) {
  var target = process.argv[2];
  var file = target ? "sql-" + target : "sql-wasm";
  if (file.indexOf("wasm-browser") > -1) {
    // Browser worker bundle expects web-style wasm loading; node worker_threads
    // should exercise the node-compatible worker bundle instead.
    file = "sql-wasm";
  }
  if (file.indexOf('memory-growth') > -1) {
    console.error("Skipping worker test for " + file + ". Not implemented yet");
    return;
  };
  const filename = "../dist/worker." + file + ".js";
  var worker = await SQLWorker.fromFile(path.join(__dirname, filename));
  var data = await worker.postMessage({ id: 1, action: 'open' });
  assert.strictEqual(data.id, 1, "Return the given id in the correct format");
  assert.deepEqual(data, { id: 1, ready: true }, 'Correct data answered to the "open" query');

  data = await worker.postMessage({
    id: 2,
    action: 'exec',
    params: {
        ":num2": 2,
        "@str2": 'b',
        // test_worker.js has issue message-passing Uint8Array
        // but it works fine in real-world browser-usage
        // "$hex2": new Uint8Array([0x00, 0x42]),
        ":num3": 3,
        "@str3": 'c'
        // "$hex3": new Uint8Array([0x00, 0x44])
    },
    sql: "CREATE TABLE test (num, str, hex);" +
      "INSERT INTO test VALUES (1, 'a', x'0042');" +
      "INSERT INTO test VALUES (:num2, @str2, x'0043');" +
      // test passing params split across multi-statement "exec"
      "INSERT INTO test VALUES (:num3, @str3, x'0044');" +
      "SELECT * FROM test;"
  });
  assert.strictEqual(data.id, 2, "Correct id");
  // debug error
  assert.strictEqual(data.error, undefined, data.error);
  var results = data.results;
  assert.ok(Array.isArray(results), 'Correct result type');
  assert.strictEqual(results.length, 1, 'Expected exactly 1 table');
  var table = results[0];
  assert.strictEqual(typeof table, 'object', 'Type of the returned table');
  assert.deepEqual(table.columns, ['num', 'str', 'hex'], 'Reading column names');
  assert.strictEqual(table.values[0][0], 1, 'Reading number');
  assert.strictEqual(table.values[0][1], 'a', 'Reading string');
  assert.deepEqual(obj2array(table.values[0][2]), [0x00, 0x42], 'Reading BLOB byte');
  assert.strictEqual(table.values[1][0], 2, 'Reading number');
  assert.strictEqual(table.values[1][1], 'b', 'Reading string');
  assert.deepEqual(obj2array(table.values[1][2]), [0x00, 0x43], 'Reading BLOB byte');
  assert.strictEqual(table.values[2][0], 3, 'Reading number');
  assert.strictEqual(table.values[2][1], 'c', 'Reading string');
  assert.deepEqual(obj2array(table.values[2][2]), [0x00, 0x44], 'Reading BLOB byte');

  data = await worker.postMessage({ action: 'getRowsModified' });
  assert.equal(data.rowsModified, 1, 'Number of changed rows')

  data = await worker.postMessage({ action: 'export' });
  var header = "SQLite format 3\0";
  var actual = "";
  for (let i = 0; i < header.length; i++) actual += String.fromCharCode(data.buffer[i]);
  assert.equal(actual, header, 'Data returned is an SQLite database file');

  // test worker properly opens db after closing
  await worker.postMessage({ action: "close" });
  await worker.postMessage({ action: "open" });
  data = await worker.postMessage({ action: "exec", sql: "SELECT 1" });
  assert.deepEqual(data.results, [{"columns":["1"],"values":[[1]]}]);
}

function obj2array(obj) {
  var buffer = []
  for (var p in obj) { buffer[p] = obj[p] }
  return buffer;
}

if (module == require.main) {
  process.on('unhandledRejection', r => console.log(r));

  require('test').run({
    'test worker': function (assert, done) {
      exports.test(null, assert).then(done);
    }
  });
}