From 590506e306b73286c7ee4baa1e9feeb6b1c68611 Mon Sep 17 00:00:00 2001
From: Kazuhito Yokoi <kazuhito.yokoi.nx@hitachi.com>
Date: Fri, 13 Jul 2018 15:26:07 +0900
Subject: [PATCH] Add test cases of persistable context for trigger node

---
 test/nodes/core/core/20-inject_spec.js  |  2 +-
 test/nodes/core/core/89-trigger_spec.js | 37 +++++++++++++++++++++++++
 test/nodes/core/logic/15-change_spec.js |  2 +-
 3 files changed, 39 insertions(+), 2 deletions(-)

diff --git a/test/nodes/core/core/20-inject_spec.js b/test/nodes/core/core/20-inject_spec.js
index 798d59fb3..111feae3a 100644
--- a/test/nodes/core/core/20-inject_spec.js
+++ b/test/nodes/core/core/20-inject_spec.js
@@ -22,7 +22,6 @@ var helper = require("node-red-node-test-helper");
 describe('inject node', function() {
 
     before(function(done) {
-        helper.startServer(done);
         Context.init({
             contextStorage: {
                 memory: {
@@ -31,6 +30,7 @@ describe('inject node', function() {
             }
         });
         Context.load();
+        helper.startServer(done);
     });
 
     after(function(done) {
diff --git a/test/nodes/core/core/89-trigger_spec.js b/test/nodes/core/core/89-trigger_spec.js
index f19bc42fe..1a6fa4ba8 100644
--- a/test/nodes/core/core/89-trigger_spec.js
+++ b/test/nodes/core/core/89-trigger_spec.js
@@ -18,16 +18,27 @@ var should = require("should");
 var sinon = require("sinon");
 var helper = require("node-red-node-test-helper");
 var triggerNode = require("../../../../nodes/core/core/89-trigger.js");
+var Context = require("../../../../red/runtime/nodes/context");
 var RED = require("../../../../red/red.js");
 
 describe('trigger node', function() {
 
     beforeEach(function(done) {
+        Context.init({
+            contextStorage: {
+                memory: {
+                    module: "memory"
+                }
+            }
+        });
+        Context.load();
         helper.startServer(done);
     });
 
     afterEach(function(done) {
         helper.unload().then(function() {
+            Context.clean({allNodes: {}});
+            Context.close();
             helper.stopServer(done);
         });
     });
@@ -312,7 +323,33 @@ describe('trigger node', function() {
             });
             n1.emit("input", {payload:null});
         });
+    });
 
+    it('should be able to return things from persistable flow and global context variables', function (done) {
+        var flow = [{"id": "n1", "type": "trigger", "name": "triggerNode", "op1": "#:(memory)::foo", "op1type": "flow",
+                     "op2": "#:(memory)::bar", "op2type": "global", "duration": "20", "wires": [["n2"]], "z": "flow" },
+                    {"id": "n2", "type": "helper"}];
+        helper.load(triggerNode, flow, function () {
+            var n1 = helper.getNode("n1");
+            var n2 = helper.getNode("n2");
+            var c = 0;
+            n2.on("input", function (msg) {
+                try {
+                    if (c === 0) {
+                        msg.should.have.a.property("payload", "foo");
+                        c += 1;
+                    } else {
+                        msg.should.have.a.property("payload", "bar");
+                        done();
+                    }
+                } catch (err) {
+                    done(err);
+                }
+            });
+            n1.context().flow.set("foo", "foo");
+            n1.context().global.set("bar", "bar");
+            n1.emit("input", { payload: null });
+        });
     });
 
     it('should be able to not output anything on first trigger', function(done) {
diff --git a/test/nodes/core/logic/15-change_spec.js b/test/nodes/core/logic/15-change_spec.js
index ecaa009b5..6b98c697f 100644
--- a/test/nodes/core/logic/15-change_spec.js
+++ b/test/nodes/core/logic/15-change_spec.js
@@ -24,7 +24,6 @@ var helper = require("node-red-node-test-helper");
 describe('change Node', function() {
 
     beforeEach(function(done) {
-        helper.startServer(done);
         Context.init({
             contextStorage: {
                 memory: {
@@ -33,6 +32,7 @@ describe('change Node', function() {
             }
         });
         Context.load();
+        helper.startServer(done);
     });
 
     afterEach(function(done) {