//

// Note: This example test is leveraging the Mocha test framework.

// Please refer to their documentation on https://mochajs.org/ for help.

//

// Place this right on top

import { initialize } from './../initialize';

// The module 'assert' provides assertion methods from node

import * as assert from 'assert';

// You can import and use all API from the 'vscode' module

// as well as import your extension to test it

import * as vscode from 'vscode';

import { SocketStream } from '../../client/common/net/socket/SocketStream';

import * as net from 'net';

const uint64be = require("uint64be");

class MockSocket {

constructor() {

this._data = '';

}

private _data: string;

private _rawDataWritten: any;

public get dataWritten(): string {

return this._data;

}

public get rawDataWritten(): any {

return this._rawDataWritten;

}

write(data: any) {

this._data = data + '';

this._rawDataWritten = data;

}

}

// Defines a Mocha test suite to group tests of similar kind together

suite('SocketStream', () => {

test('Read Byte', done => {

let buffer = new Buffer("X");

const byteValue = buffer[0];

const socket = new MockSocket();

const stream = new SocketStream((socket as any) as net.Socket, buffer)

assert.equal(stream.ReadByte(), byteValue);

done();

});

test('Read Int32', done => {

const num = 1234;

const socket = new MockSocket();

let buffer = uint64be.encode(num);

const stream = new SocketStream((socket as any) as net.Socket, buffer)

assert.equal(stream.ReadInt32(), num);

done();

});

test('Read Int64', done => {

const num = 9007199254740993;

const socket = new MockSocket();

let buffer = uint64be.encode(num);

const stream = new SocketStream((socket as any) as net.Socket, buffer)

assert.equal(stream.ReadInt64(), num);

done();

});

test('Read Ascii String', done => {

const message = 'Hello World';

const socket = new MockSocket();

let buffer = Buffer.concat([new Buffer('A'), uint64be.encode(message.length), new Buffer(message)]);

const stream = new SocketStream((socket as any) as net.Socket, buffer)

assert.equal(stream.ReadString(), message);

done();

});

test('Read Unicode String', done => {

const message = 'Hello World - Функция проверки ИНН и КПП - 说明';

const socket = new MockSocket();

const stringBuffer = new Buffer(message);

let buffer = Buffer.concat([Buffer.concat([new Buffer('U'), uint64be.encode(stringBuffer.byteLength)]), stringBuffer]);

const stream = new SocketStream((socket as any) as net.Socket, buffer)

assert.equal(stream.ReadString(), message);

done();

});

test('Read RollBackTransaction', done => {

const message = 'Hello World';

const socket = new MockSocket();

let buffer = Buffer.concat([new Buffer('A'), uint64be.encode(message.length), new Buffer(message)]);

// Write part of a second message

const partOfSecondMessage = Buffer.concat([new Buffer('A'), uint64be.encode(message.length)]);

buffer = Buffer.concat([buffer, partOfSecondMessage]);

const stream = new SocketStream((socket as any) as net.Socket, buffer)

stream.BeginTransaction();

assert.equal(stream.ReadString(), message, 'First message not read properly');

const secondMessage = stream.ReadString();

assert.equal(stream.HasInsufficientDataForReading, true, 'Should not have sufficient data for reading');

stream.RollBackTransaction();

assert.equal(stream.ReadString(), message, 'First message not read properly after rolling back transaction');

done();

});

test('Read EndTransaction', done => {

const message = 'Hello World';

const socket = new MockSocket();

let buffer = Buffer.concat([new Buffer('A'), uint64be.encode(message.length), new Buffer(message)]);

// Write part of a second message

const partOfSecondMessage = Buffer.concat([new Buffer('A'), uint64be.encode(message.length)]);

buffer = Buffer.concat([buffer, partOfSecondMessage]);

const stream = new SocketStream((socket as any) as net.Socket, buffer)

stream.BeginTransaction();

assert.equal(stream.ReadString(), message, 'First message not read properly');

const secondMessage = stream.ReadString();

assert.equal(stream.HasInsufficientDataForReading, true, 'Should not have sufficient data for reading');

stream.EndTransaction();

stream.RollBackTransaction();

assert.notEqual(stream.ReadString(), message, 'First message cannot be read after commit transaction');

done();

});

test('Write Buffer', done => {

const message = 'Hello World';

const buffer = new Buffer('');

const socket = new MockSocket();

const stream = new SocketStream((socket as any) as net.Socket, buffer)

stream.Write(new Buffer(message));

assert.equal(socket.dataWritten, message)

done();

});

test('Write Int32', done => {

const num = 1234;

const buffer = new Buffer('');

const socket = new MockSocket();

const stream = new SocketStream((socket as any) as net.Socket, buffer)

stream.WriteInt32(num);

assert.equal(uint64be.decode(socket.rawDataWritten), num)

done();

});

test('Write Int64', done => {

const num = 9007199254740993;

const buffer = new Buffer('');

const socket = new MockSocket();

const stream = new SocketStream((socket as any) as net.Socket, buffer)

stream.WriteInt64(num);

assert.equal(uint64be.decode(socket.rawDataWritten), num)

done();

});

test('Write Ascii String', done => {

const message = 'Hello World';

const buffer = new Buffer('');

const socket = new MockSocket();

const stream = new SocketStream((socket as any) as net.Socket, buffer)

stream.WriteString(message);

assert.equal(socket.dataWritten, message)

done();

});

test('Write Unicode String', done => {

const message = 'Hello World - Функция проверки ИНН и КПП - 说明';

const buffer = new Buffer('');

const socket = new MockSocket();

const stream = new SocketStream((socket as any) as net.Socket, buffer)

stream.WriteString(message);

assert.equal(socket.dataWritten, message)

done();

});

});