Attempt to fix possible race condition with sinon.spy

electron · Nov 5, 2018 · 229c69f · 229c69f
1 parent 8b9c16f
commit 229c69f
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/test/_util.js b/test/_util.js
@@ -31,11 +31,15 @@ test.after.always(t => {
 test.beforeEach(t => {
   t.context.workDir = tempy.directory()
   t.context.tempDir = tempy.directory()
-  sinon.spy(console, 'warn')
+  if (!console.warn.restore) {
+    sinon.spy(console, 'warn')
+  }
 })
 
 test.afterEach.always(t => {
-  console.warn.restore()
+  if (console.warn.restore) {
+    console.warn.restore()
+  }
   return fs.remove(t.context.workDir)
     .then(() => fs.remove(t.context.tempDir))
 })