betaflight-configurator/test/tabs/cli.js

class MockAnalytics {
    EVENT_CATEGORIES = {};

    sendEvent() {}
}

var analytics;

describe('TABS.cli', () => {
    function toArrayBuffer(string) {
        var bufferOut = new ArrayBuffer(string.length);
        var bufView = new Uint8Array(bufferOut);

        for (var i = 0; i < string.length; i++) {
            bufView[i] = string.charCodeAt(i);
        }

        return bufferOut;
    }

    describe('output', () => {
        const cliTab = $('<div>').addClass('tab-cli');
        const cliOutput = $('<div>').addClass('wrapper')
        const cliPrompt = $('<textarea name="commands">');

        cliTab.append($('<div>').addClass('window').append(cliOutput));
        cliTab.append(cliPrompt);

        CliAutoComplete.setEnabled(false); // not testing the client-side autocomplete

        before(() => {
            analytics = new MockAnalytics();

            $('body')
                .append(cliTab);

            CONFIGURATOR.cliValid = true;
        });

        after(() => cliTab.remove());

        beforeEach(() => {
            cliOutput.empty();
            cliPrompt.val('');
            TABS.cli.cliBuffer = "";
        });

        it('ambiguous auto-complete results', () => {
            TABS.cli.cliBuffer = 'se';

            TABS.cli.read({
                data: toArrayBuffer('\r\033[Kserialpassthrough\tservo\r\n# ser')
            });

            // Ambigous auto-complete from firmware is preceded with an \r carriage return
            // which only renders a line break on Mac
            const expectedValue = GUI.operating_system !== "Windows" ?
                'se<br>serialpassthrough\tservo<br>' :
                'seserialpassthrough\tservo<br>';
            expect(cliOutput.html()).to.equal(expectedValue);
            expect(cliPrompt.val()).to.equal('ser');
        });

        it('unambiguous auto-complete result', () => {
            TABS.cli.read({
                data: toArrayBuffer('serialpassthrough')
            });

            expect(cliOutput.html()).to.equal('');
            expect(cliPrompt.val()).to.equal('serialpassthrough');
        });

        it('unambiguous auto-complete result with partial buffer', () => {
            TABS.cli.cliBuffer = 'serial';

            TABS.cli.read({
                data: toArrayBuffer('passthrough')
            });

            expect(cliOutput.html()).to.equal('');
            expect(cliPrompt.val()).to.equal('serialpassthrough');
        });

        it("escape characters are skipped", () => {
            TABS.cli.read({
                data: toArrayBuffer('\033[K')
            });

            expect(cliOutput.html()).to.equal('');
            expect(cliPrompt.val()).to.equal('');
        });
    });

    function triggerEnterKey(input) {
        const enterKeycode = 13;
        const event = $.Event("keypress");
        event.which = enterKeycode;
        input.trigger(event);
    }

    function triggerTabKey(input) {
        const tabKeycode = 9;
        const event = $.Event("keydown");
        event.which = tabKeycode;
        input.trigger(event);
    }

    const backspaceCode = String.fromCharCode(127);

    describe('input', () => {
        const content = $('<div>').attr('id', 'content');
        const cliTab = $('<div>').addClass('tab-cli');
        const cliPrompt = $('<textarea name="commands">');
        cliTab.append(cliPrompt);

        beforeEach(() => {
            $('body')
                .append(content);

            // Stub loading of template.
            sinon.stub($.fn, 'load').callsFake((file, callback) => {
                content.append(cliTab);
                callback();
            });
            sinon.stub(TABS.cli, 'send');
            sinon.stub(Promise, 'reduce').callsFake((items, cb, initialValue) => {
                items.forEach((line, idx) => cb(0, line, idx));
            });
            sinon.stub(window, 'Promise').callsFake(resolve => resolve(0));
            sinon.stub(GUI, 'timeout_add').withArgs('CLI_send_slowly')
                .callsFake((name, cb) => {
                    cb();
                });

            TABS.cli.cliBuffer = "";
        });

        afterEach(() => {
            content.remove();
            $.fn.load.restore();
            TABS.cli.send.restore();
            Promise.reduce.restore();
            Promise.restore();
            GUI.timeout_add.restore();
        });

        beforeEach(() => {
            cliPrompt.val('');
            content.empty();
        });

        it('tab key triggers serial message with appended tab char', done => {
            TABS.cli.initialize(() => {
                cliPrompt.val('serial');

                triggerTabKey(cliPrompt);

                expect(TABS.cli.send).to.have.been.calledOnce;
                expect(TABS.cli.send).to.have.been.calledWith('serial\t');
                done()
            });
        });

        it('second auto complete in row', done => {
            TABS.cli.initialize(() => {
                TABS.cli.cliBuffer = '# ser';

                cliPrompt.val('seri');

                triggerTabKey(cliPrompt);

                expect(TABS.cli.send).to.have.been.calledOnce;
                expect(TABS.cli.send).to.have.been.calledWith('i\t');
                done();
            });
        });

        it('auto-complete command with trailing space', done => {
            TABS.cli.initialize(() => {
                TABS.cli.cliBuffer = '# get ';

                cliPrompt.val('get r');

                triggerTabKey(cliPrompt);

                expect(TABS.cli.send).to.have.been.calledOnce;
                expect(TABS.cli.send).to.have.been.calledWith('r\t');
                done();
            });
        });

        it('auto-complete after delete characters', done => {
            TABS.cli.initialize(() => {
                TABS.cli.cliBuffer = '# serial';

                cliPrompt.val('ser');

                triggerTabKey(cliPrompt);

                expect(TABS.cli.send).to.have.been.calledOnce;
                expect(TABS.cli.send).to.have.been.calledWith(backspaceCode.repeat(3) + '\t');
                done();
            });
        });

        it('enter after autocomplete', done => {
            TABS.cli.initialize(() => {
                TABS.cli.cliBuffer = '# servo';

                cliPrompt.val('servo');

                triggerEnterKey(cliPrompt);

                expect(TABS.cli.send).to.have.been.calledOnce;
                expect(TABS.cli.send).to.have.been.calledWith('\n');
                done();
            });
        });

        it('enter after autocomplete', done => {
            TABS.cli.initialize(() => {
                TABS.cli.cliBuffer = '# ser';

                cliPrompt.val('servo');

                triggerEnterKey(cliPrompt);

                expect(TABS.cli.send).to.have.been.calledOnce;
                expect(TABS.cli.send).to.have.been.calledWith('vo\n');
                done();
            });
        });

        it('enter after deleting characters', done => {
            TABS.cli.initialize(() => {
                TABS.cli.cliBuffer = '# serial';

                cliPrompt.val('ser');

                triggerEnterKey(cliPrompt);

                expect(TABS.cli.send).to.have.been.calledOnce;
                expect(TABS.cli.send).to.have.been.calledWith(backspaceCode.repeat(3) + '\n');
                done();
            });
        });

        it('cliBuffer is cleared on startup', done => {
            TABS.cli.cliBuffer = '# serial';

            TABS.cli.initialize(() => {
                expect(TABS.cli.cliBuffer).to.equal('');
                done();
            });
        });

        it('exit upon cleanup clears cliBuffer first', done => {
            CONFIGURATOR.connectionValid = true;
            TABS.cli.cliValid = true;


            TABS.cli.initialize(() => {
                const commandInBuffer = 'resource';

                TABS.cli.cliBuffer = `# ${commandInBuffer}`;

                TABS.cli.cleanup();

                expect(TABS.cli.send).to.have.been.calledOnce;
                expect(TABS.cli.send).to.have.been.calledWith(backspaceCode.repeat(commandInBuffer.length) + 'exit\r');
                done();
            });
        });
    });
});