fix falsy module mock

tapjs · Dec 17, 2020 · 7f6245c · 7f6245c
1 parent e0cdfa7
commit 7f6245c
Showing 1 changed file with 6 additions and 11 deletions.
diff --git a/lib/mock.js b/lib/mock.js
@@ -28,19 +28,14 @@ are the same used in ${filename} require calls`)
       this.mocks.set(mockFilePath, mocks[key])
     }
 
-    function tapRequireMock (id) {
-      const requiredFilePath = Module._resolveFilename(id, this)
-      const res = self.mocks.get(requiredFilePath)
-
-      if (res)
-        return res
-
-      return Module.prototype.require.call(this, id)
-    }
-
     class MockedModule extends Module {
       require (id) {
-        return tapRequireMock.call(this, id)
+        const requiredFilePath = Module._resolveFilename(id, this)
+
+        if (self.mocks.has(requiredFilePath))
+          return self.mocks.get(requiredFilePath)
+
+        return super.require(id)
       }
     }