SQL Operations Studio Public Preview 1 (0.23) release source code

src/vs/base/parts/ipc/test/node/ipc.net.test.ts

@@ -0,0 +1,90 @@
+/*---------------------------------------------------------------------------------------------
+ *  Copyright (c) Microsoft Corporation. All rights reserved.
+ *  Licensed under the Source EULA. See License.txt in the project root for license information.
+ *--------------------------------------------------------------------------------------------*/
+
+'use strict';
+
+import * as assert from 'assert';
+import { TPromise } from 'vs/base/common/winjs.base';
+import { Socket } from 'net';
+import { EventEmitter } from 'events';
+import { Protocol } from 'vs/base/parts/ipc/node/ipc.net';
+
+class MockDuplex extends EventEmitter {
+
+	private _cache: Buffer[] = [];
+
+	readonly destroyed = false;
+
+	private _deliver(): void {
+		if (this._cache.length) {
+			const data = Buffer.concat(this._cache);
+			this._cache.length = 0;
+			this.emit('data', data);
+		}
+	}
+
+	write(data: Buffer, cb?: Function): boolean {
+		this._cache.push(data);
+		setImmediate(() => this._deliver());
+		return true;
+	}
+}
+
+
+suite('IPC, Socket Protocol', () => {
+
+	let stream: Socket;
+
+	setup(() => {
+		stream = <any>new MockDuplex();
+	});
+
+	test('read/write', () => {
+
+		const a = new Protocol(stream);
+		const b = new Protocol(stream);
+
+		return new TPromise(resolve => {
+			const sub = b.onMessage(data => {
+				sub.dispose();
+				assert.equal(data, 'foobarfarboo');
+				resolve(null);
+			});
+			a.send('foobarfarboo');
+		}).then(() => {
+			return new TPromise(resolve => {
+				const sub = b.onMessage(data => {
+					sub.dispose();
+					assert.equal(data, 123);
+					resolve(null);
+				});
+				a.send(123);
+			});
+		});
+	});
+
+
+	test('read/write, object data', () => {
+
+		const a = new Protocol(stream);
+		const b = new Protocol(stream);
+
+		const data = {
+			pi: Math.PI,
+			foo: 'bar',
+			more: true,
+			data: 'Hello World'.split('')
+		};
+
+		a.send(data);
+
+		return new TPromise(resolve => {
+			b.onMessage(msg => {
+				assert.deepEqual(msg, data);
+				resolve(null);
+			});
+		});
+	});
+});