scriptc/packages/compiler/test/library-computed-methods.test.ts

import { execFileSync } from "node:child_process";
import { mkdtempSync, rmSync, writeFileSync } from "node:fs";
import { join } from "node:path";
import { expect, test } from "vitest";
import { compileLibrary } from "../src/index.js";

test("library data lookups do not reach dormant async methods", async () => {
  const dir = mkdtempSync("/tmp/scriptc-library-computed-");
  try {
    const entry = join(dir, "entry.mjs");
    const source = `
class Counter {
  constructor() { this.value = 7; }
  *values() { yield this.value; }
  *[Symbol.iterator]() { yield this.value; }
  async later() { return this.value; }
}
const counter = new Counter();
const reader = {
  read(object, key) { return object[key]; },
  sum(source) { let total = 0; for (const value of source) total += value; return Number(total); }
};
`;
    writeFileSync(
      entry,
      source +
        'export function result() { return Number(reader.read(counter, "value")) + reader.sum([1, 2]); }\n',
    );
    const profilePath = join(dir, "profile.json");
    writeFileSync(
      profilePath,
      JSON.stringify({
        profile_format: 1,
        name: "computed-methods",
        entry: "entry.mjs",
        emission: "llvm",
        abi: {
          prefix: "cm_",
          init_symbol: "cm_init",
          sink_register_symbol: "cm_sink",
          collect_symbol: "cm_collect",
        },
        exports: [{ export: "result", symbol: "cm_result", params: [], returns: "f64" }],
      }),
    );
    const result = await compileLibrary({ profilePath, outDir: join(dir, "out") });
    expect(result.ok, JSON.stringify(result.diagnostics)).toBe(true);
    if (!result.ok) return;
    if (process.env["SCRIPTC_PORTABLE_ONLY"] !== "1") {
      const host = join(dir, "host.c"),
        binary = join(dir, "host");
      writeFileSync(
        host,
        '#include <stdio.h>\nextern void cm_init(void); extern double cm_result(void);\nint main(void) { cm_init(); printf("%.0f\\n", cm_result()); }\n',
      );
      execFileSync("clang", [host, result.archivePath, "-lm", "-o", binary]);
      const actual = execFileSync(binary, { encoding: "utf8" });
      const expected = execFileSync(
        process.execPath,
        [
          "--input-type=module",
          "-e",
          `import { result } from ${JSON.stringify(entry)}; console.log(result());`,
        ],
        { encoding: "utf8" },
      );
      expect(actual).toBe(expected);
      expect(actual).toBe("10\n");
    }
    writeFileSync(
      entry,
      source + "export function result() { return Number(counter.values().next().value); }\n",
    );
    const refused = await compileLibrary({ profilePath, outDir: join(dir, "refused") });
    expect(refused.ok).toBe(false);
    expect(refused.diagnostics.some((diagnostic) => diagnostic.code === "SC4005")).toBe(true);
  } finally {
    rmSync(dir, { recursive: true, force: true });
  }
});