feat: allow additional stubs to be passed (#5)

bcoe · Aug 16, 2019 · 81e3a76 · 81e3a76
1 parent 6dc2089
commit 81e3a76
Show file tree

Hide file tree

Showing 2 changed files with 16 additions and 2 deletions.
diff --git a/README.md b/README.md
@@ -20,6 +20,20 @@ const myClient = http2spy.require(require.resolve("./lib/my-client"));
 assert.strictEqual(http2spy.requests[0][":method"], 'GET');
 ```
 
+## Passing Additional Stubs
+
+http2spy is just a thin wrapper on top of
+[proxyquire](https://www.npmjs.com/package/proxyquire), if you would like to
+pass additional libraries to stub, simply provide them as a second parameter:
+
+```js
+const myClient = http2spy.require(require.resolve("./lib/my-client"), {
+  'second-library': {
+    foo: () => {}
+  }
+});
+```
+
 ## License
 
 Apache Version 2.0

diff --git a/index.js b/index.js
@@ -6,7 +6,7 @@ module.exports = {
   get requests() {
     return requests;
   },
-  require: modulePath => {
+  require: (modulePath, additionalStubs = {}) => {
     const stubs = {
       http2: {
         connect: (target, connectionOptions) => {
@@ -21,6 +21,6 @@ module.exports = {
         "@global": true
       }
     };
-    return proxyquire(modulePath, stubs);
+    return proxyquire(modulePath, Object.assign(additionalStubs, stubs));
   }
 };