Define "set up" and "tear down" methods for tests

1 files changed, 25 insertions, 11 deletions

diff --git a/js/test/main.js b/js/test/main.js
index 777c020..2ded74e 100644
--- a/js/test/main.js
+++ b/js/test/main.js
@@ -1,19 +1,33 @@
 const ava = require("ava");
 const sinon = require("sinon");
 
-var document = global.document = {};
-document.addEventListener = function(){};
-const bo = require("../backoffice");
-
-ava.test("orders tracking", (t) => {
-  var xhr = sinon.useFakeXMLHttpRequest();
-  var requests = [];
-  xhr.onCreate = function(xhr){
-    requests.push(xhr);
+ava.test.beforeEach(t => {
+  // Global XHR mock
+  t.context.xhr = sinon.useFakeXMLHttpRequest();
+  t.context.requests = [];
+  t.context.xhr.onCreate = function(xhr){
+    t.context.requests.push(xhr);
   };
+  global.XMLHttpRequest = t.context.xhr;
+
+  // Global 'document' mock
+  global.document = {};
+  global.document.addEventListener = function(){};
+
+  // Load back-office code
+  t.context.bo = require("../backoffice");
+});
+
+// Remove mocks from global scope
+ava.test.afterEach(t => {
+  delete global.XMLHttpRequest;
+  delete global.document;
+});
 
-  bo.track_order(22); // Throws: XMLHttpRequest is not defined.
+ava.test("orders tracking", (t) => {
+  t.context.bo.track_order(22);
 
-  xhr.restore();
+  // Clean up
+  t.context.xhr.restore();
   t.pass();
 });