test.js

var fsp = require('fs-promise');
var Sinon = require('sinon');
var Chai  = require('chai');
var pm2 = require('pm2');
Chai.should();
Chai.use(require('sinon-chai'));

describe("webhooks", function() {
    var oot = require('./index');

    beforeEach(() => {
        Sinon.stub(fsp, "appendFile").resolves();
        Sinon.stub(oot.ssh, "connect").resolves();
        Sinon.stub(oot.ssh, "exec").resolves();
        Sinon.stub(oot.ssh, "putFile").resolves();
        Sinon.stub(pm2, "connect").yields();
        Sinon.stub(pm2, "restart").yields();
        Sinon.stub(pm2, "disconnect");
        oot.slackbot = {
            postMessageToChannel: Sinon.stub().resolves()
        }
    });
    
    afterEach(() => {
        fsp.appendFile.restore();
        oot.ssh.connect.restore();
        oot.ssh.exec.restore();
        oot.ssh.putFile.restore();
        pm2.connect.restore();
        pm2.restart.restore();
        pm2.disconnect.restore();
    });
    
    it("should listen for requests and write to a file", () => {
        return oot.handle({"json": "here"}).then(() => {
            //check for file write here
            return fsp.appendFile.should.have.been.called;
        });
    });

    it("should log the zen", () => {
        return oot.handle({"zen": "What is the sound of one hand clapping?"}).then(() => {
            //check for file write here
            fsp.appendFile.should.have.been.called;
            return fsp.appendFile.firstCall.args[1].should.contain("the sound of one hand clapping?");
        });
    });

    it("should log timestamps", () => {
        var clock = Sinon.useFakeTimers();
        return oot.handle({"zen": "What is the sound of one hand clapping?"}).then(() => {
            clock.restore();
            fsp.appendFile.should.have.been.called;
            return fsp.appendFile.firstCall.args[1].should.contain("[1970-01-01 00:00:00]");
        });
    });
    
    it("should handle log errors", () => {
       Sinon.stub(console, "error");
       const err = new Error("I AM ERROR");
       fsp.appendFile.onFirstCall().rejects(err);
       return oot.handle({"zen": "stuff"}).then(() => {
           console.error.should.have.been.calledWith(err);
           return console.error.restore();
       });
    });
    
    it("should ssh into the staging server", () => {
        return oot.handle({"zen": "What is the sound of one hand clapping?"}).then(() => {
            return oot.ssh.connect.should.have.been.calledWith({
                host: 'sockrpgtest.sockdrawer.io',
                username: 'rover',
                privateKey: '/home/rover/.ssh/id_rsa'
            });
        });
    });

    it("should copy its log to the remote server", () => {
        return oot.handle({"zen": "What is the sound of one hand clapping?"}).then(() => {
            return oot.ssh.putFile.should.have.been.calledWith('/home/rover/hooksreceived.log', '/home/rover/hooksreceived.log');
        });
    });
    
    it("should issue a pull to /usr/local/sockbot/sockbot", () => {
        return oot.handle({"zen": "What is the sound of one hand clapping?"}).then(() => {
            return oot.ssh.exec.should.have.been.calledWith('git pull', [], {'cwd': '/usr/local/sockbot/SockBot'});
        });
    });
    
    it("should handle ssh errors on connect", () => {
       const err = new Error("I AM ANOTHER ERROR");
       oot.ssh.connect.rejects(err);
       return oot.handle({"zen": "stuff"}).then(() => {
           fsp.appendFile.should.have.been.calledTwice;
           fsp.appendFile.secondCall.args[1].should.include('I AM ANOTHER ERROR');
       });
    });
    
    it("should connect to pm2", () => {
        return oot.handle({"zen": "stuff"}).then(() => {
            pm2.connect.should.have.been.called;
       });
    });   
    
    it("should restart zoidberg", () => {
        return oot.handle({"zen": "stuff"}).then(() => {
            pm2.restart.should.have.been.calledWith('zoidberg');
       });
    });
    
    it("should restart sockbot", () => {
        return oot.handle({"zen": "stuff"}).then(() => {
            pm2.restart.should.have.been.calledWith('sockbot');
       });
    });
    
    it("should disconnect from pm2", () => {
        return oot.handle({"zen": "stuff"}).then(() => {
            pm2.disconnect.should.have.been.called;
       });
    }); 
    
    it("should log the restart", () => {
        return oot.handle({"zen": "What is the sound of one hand clapping?"}).then(() => {
            return fsp.appendFile.should.have.been.calledWith(Sinon.match('hooksreceived'), Sinon.match("Restarted zoidberg"));
        });
    });
    
    it("should post to slack", () => {
        return oot.handle({"zen": "What is the sound of one hand clapping?"}).then(() => {
            return oot.slackbot.postMessageToChannel.should.have.been.calledWith("#cd-project", "Sockbot updated!");
        });
    });
});