1
0
mirror of https://github.com/node-red/node-red.git synced 2023-10-10 13:36:53 +02:00
node-red/test/nodes/helper.js

165 lines
4.4 KiB
JavaScript
Raw Normal View History

2014-07-29 13:10:20 +02:00
/**
* Copyright 2014, 2016 IBM Corp.
2014-07-29 13:10:20 +02:00
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
**/
var should = require("should");
var sinon = require("sinon");
2014-07-29 13:10:20 +02:00
var when = require("when");
var request = require('supertest');
var express = require("express");
2014-10-01 15:35:51 +02:00
var nock;
if (!process.version.match(/^v0\.[0-9]\./)) {
// only set nock for node >= 0.10
2014-10-01 18:46:00 +02:00
try {
nock = require('nock');
} catch (err) {
// nevermind, will skip nock tests
nock = null;
}
2014-10-01 15:35:51 +02:00
}
2014-07-29 13:10:20 +02:00
var RED = require("../../red/red.js");
var redNodes = require("../../red/runtime/nodes");
var flows = require("../../red/runtime/nodes/flows");
var credentials = require("../../red/runtime/nodes/credentials");
var comms = require("../../red/api/comms.js");
var log = require("../../red/runtime/log.js");
var context = require("../../red/runtime/nodes/context.js");
2014-07-29 13:10:20 +02:00
var http = require('http');
var express = require('express');
var app = express();
var address = '127.0.0.1';
var listenPort = 0; // use ephemeral port
var port;
var url;
var logSpy;
2014-08-01 15:17:32 +02:00
var server;
2014-07-29 13:10:20 +02:00
function helperNode(n) {
RED.nodes.createNode(this, n);
}
module.exports = {
load: function(testNode, testFlows, testCredentials, cb) {
var i;
logSpy = sinon.spy(log,"log");
2015-02-03 23:02:26 +01:00
logSpy.FATAL = log.FATAL;
logSpy.ERROR = log.ERROR;
logSpy.WARN = log.WARN;
logSpy.INFO = log.INFO;
logSpy.DEBUG = log.DEBUG;
logSpy.TRACE = log.TRACE;
logSpy.METRIC = log.METRIC;
if (typeof testCredentials === 'function') {
cb = testCredentials;
testCredentials = {};
}
2014-07-29 13:10:20 +02:00
var storage = {
getFlows: function() {
return when.resolve({flows:testFlows,credentials:testCredentials});
}
2014-07-29 13:10:20 +02:00
};
var settings = {
available: function() { return false; }
2014-10-01 15:35:51 +02:00
};
2015-05-15 11:29:13 +02:00
var red = {};
for (i in RED) {
2015-05-15 11:29:13 +02:00
if (RED.hasOwnProperty(i) && !/^(init|start|stop)$/.test(i)) {
var propDescriptor = Object.getOwnPropertyDescriptor(RED,i);
Object.defineProperty(red,i,propDescriptor);
}
}
2015-05-15 11:29:13 +02:00
red["_"] = function(messageId) {
return messageId;
};
2016-09-23 11:38:30 +02:00
redNodes.init({settings:settings, storage:storage,log:log});
2014-07-29 13:10:20 +02:00
RED.nodes.registerType("helper", helperNode);
if (Array.isArray(testNode)) {
for (i = 0; i < testNode.length; i++) {
2015-05-15 11:29:13 +02:00
testNode[i](red);
}
} else {
2015-05-15 11:29:13 +02:00
testNode(red);
}
2014-07-29 13:10:20 +02:00
flows.load().then(function() {
flows.startFlows();
2016-10-09 23:22:13 +02:00
should.deepEqual(testFlows, flows.getFlows().flows);
2014-07-29 13:10:20 +02:00
cb();
});
},
unload: function() {
// TODO: any other state to remove between tests?
redNodes.clearRegistry();
logSpy.restore();
context.clean({allNodes:[]});
return flows.stopFlows();
},
2014-07-29 13:10:20 +02:00
getNode: function(id) {
return flows.get(id);
},
credentials: credentials,
2014-07-29 13:10:20 +02:00
clearFlows: function() {
2015-01-10 23:09:37 +01:00
return flows.stopFlows();
2014-07-29 13:10:20 +02:00
},
request: function() {
return request(RED.httpAdmin);
},
startServer: function(done) {
server = http.createServer(function(req,res) { app(req,res); });
RED.init(server, {
SKIP_BUILD_CHECK: true,
logging:{console:{level:'off'}}
});
2014-07-29 13:10:20 +02:00
server.listen(listenPort, address);
server.on('listening', function() {
port = server.address().port;
url = 'http://' + address + ':' + port;
comms.start();
done();
});
},
2014-08-01 15:17:32 +02:00
//TODO consider saving TCP handshake/server reinit on start/stop/start sequences
stopServer: function(done) {
if (server) {
try {
server.close(done);
} catch(e) {
done();
}
2014-08-01 15:17:32 +02:00
}
},
2014-07-29 13:10:20 +02:00
url: function() { return url; },
2014-10-01 15:35:51 +02:00
nock: nock,
log: function() { return logSpy;}
2014-07-29 13:10:20 +02:00
};