novnc/tests/assertions.js

// some useful assertions for noVNC
chai.use(function (_chai, utils) {
    _chai.Assertion.addMethod('displayed', function (target_data) {
        var obj = this._obj;
        var data_cl = obj._drawCtx.getImageData(0, 0, obj._viewportLoc.w, obj._viewportLoc.h).data;
        // NB(directxman12): PhantomJS 1.x doesn't implement Uint8ClampedArray, so work around that
        var data = new Uint8Array(data_cl);
        var same = true;
        var len = data_cl.length;
        if (len != target_data.length) {
            same = false;
        } else {
            for (var i = 0; i < len; i++) {
                if (data[i] != target_data[i]) {
                    same = false;
                    break;
                }
            }
        }
        if (!same) {
            console.log("expected data: %o, actual data: %o", target_data, data);
        }
        this.assert(same,
            "expected #{this} to have displayed the image #{exp}, but instead it displayed #{act}",
            "expected #{this} not to have displayed the image #{act}",
            target_data,
            data);
    });

    _chai.Assertion.addMethod('sent', function (target_data) {
        var obj = this._obj;
        obj.inspect = function () {
            var res = { _websocket: obj._websocket, rQi: obj._rQi, _rQ: new Uint8Array(obj._rQ.buffer, 0, obj._rQlen),
                        _sQ: new Uint8Array(obj._sQ.buffer, 0, obj._sQlen) };
            res.prototype = obj;
            return res;
        };
        var data = obj._websocket._get_sent_data();
        var same = true;
        for (var i = 0; i < obj.length; i++) {
            if (data[i] != target_data[i]) {
                same = false;
                break;
            }
        }
        if (!same) {
            console.log("expected data: %o, actual data: %o", target_data, data);
        }
        this.assert(same,
            "expected #{this} to have sent the data #{exp}, but it actually sent #{act}",
            "expected #{this} not to have sent the data #{act}",
            Array.prototype.slice.call(target_data),
            Array.prototype.slice.call(data));
    });

    _chai.Assertion.addProperty('array', function () {
        utils.flag(this, 'array', true);
    });

    _chai.Assertion.overwriteMethod('equal', function (_super) {
        return function assertArrayEqual(target) {
            if (utils.flag(this, 'array')) {
                var obj = this._obj;

                var i;
                var same = true;

                if (utils.flag(this, 'deep')) {
                    for (i = 0; i < obj.length; i++) {
                        if (!utils.eql(obj[i], target[i])) {
                            same = false;
                            break;
                        }
                    }

                    this.assert(same,
                        "expected #{this} to have elements deeply equal to #{exp}",
                        "expected #{this} not to have elements deeply equal to #{exp}",
                        Array.prototype.slice.call(target));
                } else {
                    for (i = 0; i < obj.length; i++) {
                        if (obj[i] != target[i]) {
                            same = false;
                            break;
                        }
                    }

                    this.assert(same,
                        "expected #{this} to have elements equal to #{exp}",
                        "expected #{this} not to have elements equal to #{exp}",
                        Array.prototype.slice.call(target));
                }
            } else {
                _super.apply(this, arguments);
            }
        };
    });
});