plumber differences between node and browser so that tests run on both

1 files changed, 23 insertions, 0 deletions

diff --git a/src/utils/window.js b/src/utils/window.js
index 9e51339..626fde3 100644
--- a/src/utils/window.js
+++ b/src/utils/window.js
@@ -7,3 +7,26 @@ export function registerWindow (win = null, doc = null) {
   globals.window = win
   globals.document = doc
 }
+
+const save = {}
+
+export function saveWindow () {
+  save.window = globals.window
+  save.document = globals.document
+}
+
+export function restoreWindow () {
+  globals.window = save.window
+  globals.document = save.document
+}
+
+export function withWindow (win, fn) {
+  saveWindow()
+  registerWindow(win, win.document)
+  fn(win, win.document)
+  restoreWindow()
+}
+
+export function getWindow () {
+  return globals.window
+}