1 files changed, 126 insertions, 0 deletions
diff --git a/std/wasi/snapshot_preview1_test.ts b/std/wasi/snapshot_preview1_test.ts
new file mode 100644
index 000000000..9d8892f9c
--- /dev/null
+++ b/std/wasi/snapshot_preview1_test.ts
@@ -0,0 +1,126 @@
+/* eslint-disable */
+
+import { assert, assertEquals } from "../testing/asserts.ts";
+import * as path from "../path/mod.ts";
+import WASI from "./snapshot_preview1.ts";
+
+if (import.meta.main) {
+  const options = JSON.parse(Deno.args[0]);
+  const binary = await Deno.readFile(Deno.args[1]);
+  const module = await WebAssembly.compile(binary);
+
+  const wasi = new WASI({
+    env: options.env,
+    args: options.args,
+    preopens: options.preopens,
+  });
+
+  const instance = new WebAssembly.Instance(module, {
+    wasi_snapshot_preview1: wasi.exports,
+  });
+
+  wasi.memory = instance.exports.memory;
+
+  instance.exports._start();
+} else {
+  const rootdir = path.dirname(path.fromFileUrl(import.meta.url));
+  const testdir = path.join(rootdir, "testdata");
+  const outdir = path.join(testdir, "snapshot_preview1");
+
+  for await (const entry of Deno.readDir(testdir)) {
+    if (!entry.name.endsWith(".rs")) {
+      continue;
+    }
+
+    const process = Deno.run({
+      cmd: [
+        "rustc",
+        "--target",
+        "wasm32-wasi",
+        "--out-dir",
+        outdir,
+        path.join(testdir, entry.name),
+      ],
+      stdout: "inherit",
+      stderr: "inherit",
+    });
+
+    const status = await process.status();
+    assert(status.success);
+
+    process.close();
+
+    // TODO(caspervonb) allow the prelude to span multiple lines
+    const source = await Deno.readTextFile(path.join(testdir, entry.name));
+    const prelude = source.match(/^\/\/\s*\{.*/);
+    if (prelude) {
+      const basename = entry.name.replace(/.rs$/, ".json");
+      await Deno.writeTextFile(
+        path.join(outdir, basename),
+        prelude[0].slice(2)
+      );
+    }
+  }
+
+  for await (const entry of Deno.readDir(outdir)) {
+    if (!entry.name.endsWith(".wasm")) {
+      continue;
+    }
+
+    Deno.test(entry.name, async function () {
+      const basename = entry.name.replace(/\.wasm$/, ".json");
+      const prelude = await Deno.readTextFile(path.resolve(outdir, basename));
+      const options = JSON.parse(prelude);
+
+      const process = await Deno.run({
+        cwd: testdir,
+        cmd: [
+          `${Deno.execPath()}`,
+          "run",
+          "--quiet",
+          "--unstable",
+          "--v8-flags=--experimental-wasm-bigint",
+          "--allow-all",
+          import.meta.url,
+          prelude,
+          path.resolve(outdir, entry.name),
+        ],
+        stdin: "piped",
+        stdout: "piped",
+        stderr: "piped",
+      });
+
+      if (options.stdin) {
+        const stdin = new TextEncoder().encode(options.stdin);
+        await Deno.writeAll(process.stdin, stdin);
+      }
+
+      process.stdin.close();
+
+      const stdout = await Deno.readAll(process.stdout);
+
+      if (options.stdout) {
+        assertEquals(new TextDecoder().decode(stdout), options.stdout);
+      } else {
+        await Deno.writeAll(Deno.stdout, stdout);
+      }
+
+      process.stdout.close();
+
+      const stderr = await Deno.readAll(process.stderr);
+
+      if (options.stderr) {
+        assertEquals(new TextDecoder().decode(stderr), options.stderr);
+      } else {
+        await Deno.writeAll(Deno.stderr, stderr);
+      }
+
+      process.stderr.close();
+
+      const status = await process.status();
+      assertEquals(status.code, options.exitCode ? +options.exitCode : 0);
+
+      process.close();
+    });
+  }
+}