node-red-nodes/test/analysis/mlsentiment/mlsentiment_spec.js

/**
 * Copyright JS Foundation and other contributors, http://js.foundation
 *
 * 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 sentimentNode = require("../../../analysis/mlsentiment/mlsentiment.js");
var helper = require("node-red-node-test-helper");

describe('mlsentiment Node', function() {

    before(function(done) {
        helper.startServer(done);
    });

    after(function(done) {
        helper.stopServer(done);
    });

    afterEach(function() {
        helper.unload();
    });

    it('should be loaded', function(done) {
        var flow = [{id:"sentimentNode1", type:"mlsentiment", name: "sentimentNode" }];
        helper.load(sentimentNode, flow, function() {
            var sentimentNode1 = helper.getNode("sentimentNode1");
            sentimentNode1.should.have.property('name', 'sentimentNode');
            done();
        });
    });

    it('should pass on msg if no payload', function(done) {
        var flow = [{id:"jn1",type:"mlsentiment",wires:[["jn2"]]},
                    {id:"jn2", type:"helper"}];
        helper.load(sentimentNode, flow, function() {
            var jn1 = helper.getNode("jn1");
            var jn2 = helper.getNode("jn2");
            jn2.on("input", function(msg) {
                msg.should.not.have.property('sentiment');
                msg.topic.should.equal("pass on");
                done();
            });
            var testString = 'good, great, best, brilliant';
            jn1.receive({topic:"pass on"});
        });
    });

    it('should add a positive score for good words', function(done) {
        var flow = [{id:"jn1",type:"mlsentiment",wires:[["jn2"]]},
                    {id:"jn2", type:"helper"}];
        helper.load(sentimentNode, flow, function() {
            var jn1 = helper.getNode("jn1");
            var jn2 = helper.getNode("jn2");
            jn2.on("input", function(msg) {
                try {
                    msg.should.have.property('sentiment');
                    msg.sentiment.should.have.property('score');
                    msg.sentiment.score.should.be.a.Number();
                    msg.sentiment.score.should.be.above(10);
                    done();
                } catch(err) {
                    done(err);
                }
            });
            var testString = 'good, great, best, brilliant';
            jn1.receive({payload:testString});
        });
    });

    it('should add a positive score for good words (in French)', function(done) {
        var flow = [{id:"jn1",type:"mlsentiment",wires:[["jn2"]],lang:"fr"},
                    {id:"jn2", type:"helper"}];
        helper.load(sentimentNode, flow, function() {
            var jn1 = helper.getNode("jn1");
            var jn2 = helper.getNode("jn2");
            jn2.on("input", function(msg) {
                try {
                    msg.should.have.property('sentiment');
                    msg.sentiment.should.have.property('score');
                    msg.sentiment.score.should.be.a.Number();
                    msg.sentiment.score.should.be.above(5);
                    done();
                } catch(err) {
                    done(err);
                }
            });
            var testString = 'bon, belle, don du ciel, brillant';
            jn1.receive({payload:testString});
        });
    });

    it('should add a positive score for good words - alternative property', function(done) {
        var flow = [{id:"jn1",type:"mlsentiment",property:"foo",wires:[["jn2"]]},
                    {id:"jn2", type:"helper"}];
        helper.load(sentimentNode, flow, function() {
            var jn1 = helper.getNode("jn1");
            var jn2 = helper.getNode("jn2");
            jn2.on("input", function(msg) {
                try {
                    msg.should.have.property('sentiment');
                    msg.sentiment.should.have.property('score');
                    msg.sentiment.score.should.be.a.Number();
                    msg.sentiment.score.should.be.above(10);
                    done();
                } catch(err) {
                    done(err);
                }
            });
            var testString = 'good, great, best, brilliant';
            jn1.receive({foo:testString});
        });
    });

    it('should add a negative score for bad words', function(done) {
        var flow = [{id:"jn1",type:"mlsentiment",wires:[["jn2"]]},
                    {id:"jn2", type:"helper"}];
        helper.load(sentimentNode, flow, function() {
            var jn1 = helper.getNode("jn1");
            var jn2 = helper.getNode("jn2");
            jn2.on("input", function(msg) {
                msg.should.have.property('sentiment');
                msg.sentiment.should.have.property('score');
                msg.sentiment.score.should.be.a.Number();
                msg.sentiment.score.should.be.below(-10);
                done();
            });
            var testString = 'bad, horrible, negative, awful';
            jn1.receive({payload:testString});
        });
    });

    it('should add a negative score for bad words - alternative property', function(done) {
        var flow = [{id:"jn1",type:"mlsentiment",property:"foo",wires:[["jn2"]]},
                    {id:"jn2", type:"helper"}];
        helper.load(sentimentNode, flow, function() {
            var jn1 = helper.getNode("jn1");
            var jn2 = helper.getNode("jn2");
            jn2.on("input", function(msg) {
                msg.should.have.property('sentiment');
                msg.sentiment.should.have.property('score');
                msg.sentiment.score.should.be.a.Number();
                msg.sentiment.score.should.be.below(-10);
                done();
            });
            var testString = 'bad, horrible, negative, awful';
            jn1.receive({foo:testString});
        });
    });

    it('should allow you to override word scoring', function(done) {
        var flow = [{id:"jn1",type:"mlsentiment",wires:[["jn2"]]},
                    {id:"jn2", type:"helper"}];
        helper.load(sentimentNode, flow, function() {
            var jn1 = helper.getNode("jn1");
            var jn2 = helper.getNode("jn2");
            jn2.on("input", function(msg) {
                msg.should.have.property('sentiment');
                msg.sentiment.should.have.property('score');
                msg.sentiment.score.should.be.a.Number();
                msg.sentiment.score.should.equal(20);
                done();
            });
            var testString = 'sick, wicked';
            var overrides = {'sick': 10, 'wicked': 10 };
            jn1.receive({payload:testString,overrides:overrides});
        });
    });

    it('should allow you to override word scoring - alternative property', function(done) {
        var flow = [{id:"jn1",type:"mlsentiment",property:"foo",wires:[["jn2"]]},
                    {id:"jn2", type:"helper"}];
        helper.load(sentimentNode, flow, function() {
            var jn1 = helper.getNode("jn1");
            var jn2 = helper.getNode("jn2");
            jn2.on("input", function(msg) {
                msg.should.have.property('sentiment');
                msg.sentiment.should.have.property('score');
                msg.sentiment.score.should.be.a.Number();
                msg.sentiment.score.should.equal(20);
                done();
            });
            var testString = 'sick, wicked';
            var overrides = {'sick': 10, 'wicked': 10 };
            jn1.receive({foo:testString,overrides:overrides});
        });
    });

});