make babel injectable in babel-register

2018-01-25 08:16:34 +01:00
parent ca18ea5e79
commit 0e570eceb2
2 changed files with 22 additions and 1 deletions
--- a/packages/babel-register/test/index.js
+++ b/packages/babel-register/test/index.js
@@ -103,4 +103,20 @@ describe("@babel/register", function() {
    chai.expect((gen_error = require(GEN_ERROR))).to.be.ok;
    chai.expect(gen_error()).to.match(/gen_error\.js:2:86/);
  });
+
+  it("uses custom babel when requested", () => {
+    setupRegister({
+      babel: {
+        transform() {
+          return { code: "module.exports = 'hello';" };
+        },
+        getEnv() {
+          return {};
+        },
+        version: "custom",
+      },
+    });
+
+    chai.expect(require(DATA_ES2015)).to.equal("hello");
+  });
 });