import * as babel from "@babel/core";
import transformAmd from "../lib/index.js";
import externalHelpers from "@babel/plugin-external-helpers";

it("'importInterop' accepts a function", function () {
  const code = `
    import a from "a";
    import b from "b";
    import c from "c";

    a();
    b();
    c();
  `;

  const importInterop = source => {
    if (source === "a") return "babel";
    else if (source === "b") return "node";
    else if (source === "c") return "none";
  };

  const output = babel.transformSync(code, {
    configFile: false,
    ast: false,
    plugins: [
      [externalHelpers, { helperVersion: "7.100.0" }],
      [transformAmd, { importInterop }],
    ],
  }).code;

  expect(output).toMatchInlineSnapshot(`
    "define([\\"a\\", \\"b\\", \\"c\\"], function (_a, _b, _c) {
      \\"use strict\\";

      _a = babelHelpers.interopRequireDefault(_a);
      (0, _a.default)();

      _b();

      (0, _c.default)();
    });"
  `);
});