Fix async problem in test cases

test/nodes/core/core/20-inject_spec.js

@@ -21,7 +21,7 @@ var helper = require("node-red-node-test-helper");
 
 describe('inject node', function() {
 
-    before(function(done) {
+    beforeEach(function(done) {
         Context.init({
             contextStorage: {
                 memory: {
@@ -29,18 +29,19 @@ describe('inject node', function() {
                 }
             }
         });
-        Context.load();
+        Context.load().then(function () {
             helper.startServer(done);
         });
-
-    after(function(done) {
-        helper.stopServer(done);
     });
 
-    afterEach(function() {
+    afterEach(function(done) {
-        helper.unload();
+        helper.unload().then(function () {
-        Context.clean({allNodes: {}});
+            return Context.clean({allNodes: {}});
-        Context.close();
+        }).then(function () {
+            return Context.close();
+        }).then(function () {
+            helper.stopServer(done);
+        });
     });
 
     it('sets the value of flow context property', function (done) {

test/nodes/core/core/89-trigger_spec.js

@@ -31,14 +31,17 @@ describe('trigger node', function() {
                 }
             }
         });
-        Context.load();
+        Context.load().then(function () {
             helper.startServer(done);
         });
+    });
 
     afterEach(function(done) {
-        helper.unload().then(function() {
+        helper.unload().then(function () {
-            Context.clean({allNodes: {}});
+            return Context.clean({allNodes: {}});
-            Context.close();
+        }).then(function () {
+            return Context.close();
+        }).then(function () {
             helper.stopServer(done);
         });
     });

test/nodes/core/logic/15-change_spec.js

@@ -31,15 +31,19 @@ describe('change Node', function() {
                 }
             }
         });
-        Context.load();
+        Context.load().then(function () {
             helper.startServer(done);
         });
+    });
 
     afterEach(function(done) {
-        helper.unload();
+        helper.unload().then(function () {
+            return Context.clean({allNodes: {}});
+        }).then(function () {
+            return Context.close();
+        }).then(function () {
             helper.stopServer(done);
-        Context.clean({allNodes:{}});
+        });
-        Context.close();        
     });
 
     it('should load node with defaults', function(done) {