// Copyright (c) .NET Foundation. All rights reserved.
// Licensed under the Apache License, Version 2.0. See License.txt in the project root for license information.

import { Base64EncodedHubProtocol } from "../Microsoft.AspNetCore.SignalR.Client.TS/Base64EncodedHubProtocol"
import { IHubProtocol, HubMessage, ProtocolType } from "../Microsoft.AspNetCore.SignalR.Client.TS/IHubProtocol"

class FakeHubProtocol implements IHubProtocol {
    name: "fakehubprotocol";
    type: ProtocolType;

    parseMessages(input: any): HubMessage[] {
        let s = "";

        new Uint8Array(input).forEach((item: any) => {
            s += String.fromCharCode(item);
        });

        return JSON.parse(s);
    }

    writeMessage(message: HubMessage): any {
        let s = JSON.stringify(message);
        let payload = new Uint8Array(s.length);
        for (let i = 0; i < payload.length; i++) {
            payload[i] = s.charCodeAt(i);
        }
        return payload;
    }
}

describe("Base64EncodedHubProtocol", () => {
    ([
        ["ABC", new Error("Invalid payload.")],
        ["3:ABC", new Error("Invalid payload.")],
        [":;", new Error("Invalid length: ''")],
        ["1.0:A;", new Error("Invalid length: '1.0'")],
        ["2:A;", new Error("Invalid message size.")],
        ["2:ABC;", new Error("Invalid message size.")],
    ] as [[string, Error]]).forEach(([payload, expected_error]) => {
        it(`should fail to parse '${payload}'`, () => {
            expect(() => new Base64EncodedHubProtocol(new FakeHubProtocol()).parseMessages(payload)).toThrow(expected_error);
        });
    });

    ([
        ["2:{};", {}],
    ] as [[string, any]]).forEach(([payload, message]) => {
        it(`should be able to parse '${payload}'`, () => {

            let globalAny: any = global;
            globalAny.atob = function(input: any) { return input };

            let result = new Base64EncodedHubProtocol(new FakeHubProtocol()).parseMessages(payload);
            expect(result).toEqual(message);

            delete globalAny.atob;
        });
    });

    ([
        [{}, "2:{};"],
    ] as [[any, string]]).forEach(([message, payload]) => {
        it(`should be able to write '${JSON.stringify(message)}'`, () => {

            let globalAny: any = global;
            globalAny.btoa = function(input: any) { return input };

            let result = new Base64EncodedHubProtocol(new FakeHubProtocol()).writeMessage(message);
            expect(result).toEqual(payload);

            delete globalAny.btoa;
        });
    });
});