Merge VS Code 1.23.1 (#1520)

src/vs/workbench/test/electron-browser/api/extHostWebview.test.ts

@@ -0,0 +1,62 @@
+/*---------------------------------------------------------------------------------------------
+ *  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 { MainThreadWebviews } from 'vs/workbench/api/electron-browser/mainThreadWebview';
+import { ExtHostWebviews } from 'vs/workbench/api/node/extHostWebview';
+import { mock } from 'vs/workbench/test/electron-browser/api/mock';
+import * as vscode from 'vscode';
+import { SingleProxyRPCProtocol } from './testRPCProtocol';
+import { Position as EditorPosition } from 'vs/platform/editor/common/editor';
+
+suite('ExtHostWebview', function () {
+
+	test('Cannot register multiple serializer for the same view type', async () => {
+		const viewType = 'view.type';
+
+		const shape = createNoopMainThreadWebviews();
+		const extHostWebviews = new ExtHostWebviews(SingleProxyRPCProtocol(shape));
+
+		let lastInvokedDeserializer: vscode.WebviewPanelSerializer | undefined = undefined;
+
+		class NoopSerializer implements vscode.WebviewPanelSerializer {
+			async serializeWebviewPanel(webview: vscode.WebviewPanel): Promise<any> { /* noop */ }
+
+			async deserializeWebviewPanel(webview: vscode.WebviewPanel, state: any): Promise<void> {
+				lastInvokedDeserializer = this;
+			}
+		}
+
+		const serializerA = new NoopSerializer();
+		const serializerB = new NoopSerializer();
+
+		const serializerARegistration = extHostWebviews.registerWebviewPanelSerializer(viewType, serializerA);
+
+		await extHostWebviews.$deserializeWebviewPanel('x', viewType, 'title', {}, EditorPosition.ONE, {});
+		assert.strictEqual(lastInvokedDeserializer, serializerA);
+
+		assert.throws(
+			() => extHostWebviews.registerWebviewPanelSerializer(viewType, serializerB),
+			'Should throw when registering two serializers for the same view');
+
+		serializerARegistration.dispose();
+
+		extHostWebviews.registerWebviewPanelSerializer(viewType, serializerB);
+
+		await extHostWebviews.$deserializeWebviewPanel('x', viewType, 'title', {}, EditorPosition.ONE, {});
+		assert.strictEqual(lastInvokedDeserializer, serializerB);
+	});
+});
+
+
+function createNoopMainThreadWebviews() {
+	return new class extends mock<MainThreadWebviews>() {
+		$registerSerializer() { /* noop */ }
+		$unregisterSerializer() { /* noop */ }
+	};
+}
+