Add overwrite arg

9 years ago · f26061d836
1 changed files with 4 additions and 3 deletions
--- a/src/index.js
+++ b/src/index.js
@ -1,12 +1,13 @@
 import { jsdom } from 'jsdom';

-module.exports = (props) => {
+module.exports = (props, overwrite) => {
+  const presetProps = props instanceof Array ? props : false;
  const window = jsdom('<html><body></body></html>').defaultView;

  Object
    .keys(window)
-    .filter(prop => typeof global[prop] === 'undefined')
-    .filter(prop => !(props instanceof Array && props.indexOf(prop) === -1))
+    .filter(prop => typeof global[prop] === 'undefined' || presetProps && overwrite)
+    .filter(prop => !(presetProps && props.indexOf(prop) === -1))
    .forEach(prop => global[prop] = window[prop]);

  return window;