Improve shebang test

jhnns · Dec 18, 2021 · b70fdc0 · b70fdc0
1 parent 0584c53
commit b70fdc0
Show file tree

Hide file tree

Showing 3 changed files with 7 additions and 5 deletions.
diff --git a/lib/moduleEnv.js b/lib/moduleEnv.js
@@ -44,7 +44,7 @@ var moduleWrapper0 = Module.wrapper[0],
     // Test the regular expresssion at https://regex101.com/r/dvnZPv/2 and also check out testLib/constModule.js.
     matchConst = /(^|\s|\}|;)const(\/\*|\s|{)/gm,
     // Required for importing modules with shebang declarations, since NodeJS 12.16.0
-    shebang = /^(#!).+/,
+    shebang = /^#!.+/,
     nodeRequire,
     currentModule;
 

diff --git a/testLib/sharedTestCases.js b/testLib/sharedTestCases.js
@@ -409,8 +409,10 @@ module.exports = function () {
     });
 
     it("should be possible to rewire shebang modules", function () {
-        expect(function () {
-            rewire("./shebangModule");
-        }).to.not.throwError();
+        var shebangModule = rewire("./shebangModule");
+        var shebangs = shebangModule.__get__("shebangs");
+
+        expect(typeof shebangs).to.be("function");
+        expect(shebangModule.shebangs()).to.be(true);
     });
 };
diff --git a/testLib/shebangModule.js b/testLib/shebangModule.js
@@ -4,4 +4,4 @@ function shebangs() {
     return true;
 }
 
-module.exports.shebangs = shebangs;
+exports.shebangs = shebangs;