Add RUBY_ROOT hook point in jest test

Author: kateinoigakukun

packages/npm-packages/ruby-wasm-wasi/test/init.ts

@@ -1,14 +1,41 @@
 import fs from "fs/promises";
 import path from "path";
-import { DefaultRubyVM } from "../src/node";
+import { WASI } from "wasi";
+import { RubyVM } from "../src/index";
 
 const rubyModule = (async () => {
-  const binary = await fs.readFile(
-    path.join(__dirname, "./../dist/ruby+stdlib.wasm")
-  );
+  let binaryPath;
+  if (process.env.RUBY_ROOT) {
+    binaryPath = path.join(process.env.RUBY_ROOT, "./usr/local/bin/ruby");
+  } else {
+    binaryPath = path.join(__dirname, "./../dist/ruby+stdlib.wasm");
+  }
+  const binary = await fs.readFile(binaryPath);
   return await WebAssembly.compile(binary.buffer);
 })();
 
 export const initRubyVM = async () => {
-  return (await DefaultRubyVM(await rubyModule)).vm;
+  let preopens = {};
+  if (process.env.RUBY_ROOT) {
+    preopens["/usr"] = path.join(process.env.RUBY_ROOT, "./usr");
+  }
+  const wasi = new WASI({
+    args: ["ruby.wasm"].concat(process.argv.slice(2)),
+    preopens,
+  });
+
+  const vm = new RubyVM();
+  const imports = {
+    wasi_snapshot_preview1: wasi.wasiImport,
+  };
+
+  vm.addToImports(imports);
+
+  const instance = await WebAssembly.instantiate(await rubyModule, imports);
+  await vm.setInstance(instance);
+
+  wasi.initialize(instance);
+
+  vm.initialize();
+  return vm;
 };

packages/npm-packages/ruby-wasm-wasi/test/package.test.ts

@@ -2,6 +2,7 @@ import fs from "fs/promises";
 import path from "path";
 import { WASI } from "wasi";
 import { RubyVM } from "../dist/index.umd.js";
+import { DefaultRubyVM } from "../src/node";
 
 const initRubyVM = async (rubyModule: WebAssembly.Module, args: string[]) => {
   const wasi = new WASI();
@@ -29,12 +30,28 @@ const initRubyVM = async (rubyModule: WebAssembly.Module, args: string[]) => {
 describe("Packaging validation", () => {
   jest.setTimeout(20 /*sec*/ * 1000);
 
+  const moduleCache = new Map<string, WebAssembly.Module>();
+  const loadWasmModule = async (file: string) => {
+    if (moduleCache.has(file)) {
+      return moduleCache.get(file)!;
+    }
+    const binary = await fs.readFile(path.join(__dirname, `./../dist/${file}`));
+    const mod = await WebAssembly.compile(binary.buffer);
+    moduleCache.set(file, mod);
+    return mod;
+  };
+
+  test("DefaultRubyVM", async () => {
+    const mod = await loadWasmModule(`ruby+stdlib.wasm`);
+    const { vm } = await DefaultRubyVM(mod);
+    vm.eval(`require "stringio"`);
+  })
+
   test.each([
     { file: "ruby+stdlib.wasm", stdlib: true },
     { file: "ruby.debug+stdlib.wasm", stdlib: true },
   ])("Load all variants", async ({ file, stdlib }) => {
-    const binary = await fs.readFile(path.join(__dirname, `./../dist/${file}`));
-    const mod = await WebAssembly.compile(binary.buffer);
+    const mod = await loadWasmModule(file);
     const { vm } = await initRubyVM(mod, ["ruby.wasm", "-e_=0"]);
     // Check loading ext library
     vm.eval(`require "stringio"`);
@@ -45,10 +62,7 @@ describe("Packaging validation", () => {
   });
 
   test("ruby.debug+stdlib.wasm has debug info", async () => {
-    const binary = await fs.readFile(
-      path.join(__dirname, `./../dist/ruby.debug+stdlib.wasm`)
-    );
-    const mod = await WebAssembly.compile(binary.buffer);
+    const mod = await loadWasmModule("ruby.debug+stdlib.wasm");
     const nameSections = WebAssembly.Module.customSections(mod, "name");
     expect(nameSections.length).toBe(1);
   });