Add some pinning tests for indent

extensions/typescript-language-features/src/test/onEnter.test.ts

@@ -0,0 +1,58 @@
+/*---------------------------------------------------------------------------------------------
+ *  Copyright (c) Microsoft Corporation. All rights reserved.
+ *  Licensed under the MIT License. See License.txt in the project root for license information.
+ *--------------------------------------------------------------------------------------------*/
+
+import * as assert from 'assert';
+import 'mocha';
+import * as vscode from 'vscode';
+import { CURSOR, withRandomFileEditor } from './testUtils';
+
+const onDocumentChange = (doc: vscode.TextDocument): Promise<vscode.TextDocument> => {
+	return new Promise<vscode.TextDocument>(resolve => {
+		const sub = vscode.workspace.onDidChangeTextDocument(e => {
+			if (e.document !== doc) {
+				return;
+			}
+			sub.dispose();
+			resolve(e.document);
+		});
+	});
+};
+
+const type = async (document: vscode.TextDocument, text: string): Promise<vscode.TextDocument> => {
+	const onChange = onDocumentChange(document);
+	await vscode.commands.executeCommand('type', { text });
+	await onChange;
+	return document;
+};
+
+suite('OnEnter', () => {
+	test('should indent after if block with braces', () => {
+		return withRandomFileEditor(`if (true) {${CURSOR}`, 'js', async (_editor, document) => {
+			await type(document, '\nx');
+			assert.strictEqual(document.getText(), `if (true) {\n    x`);
+		});
+	});
+
+	test('should indent within empty object literal', () => {
+		return withRandomFileEditor(`({${CURSOR}})`, 'js', async (_editor, document) => {
+			await type(document, '\nx');
+			assert.strictEqual(document.getText(), `({\n    x\n})`);
+		});
+	});
+
+	test('should indent after simple jsx tag with attributes', () => {
+		return withRandomFileEditor(`const a = <div onclick={bla}>${CURSOR}`, 'jsx', async (_editor, document) => {
+			await type(document, '\nx');
+			assert.strictEqual(document.getText(), `const a = <div onclick={bla}>\n    x`);
+		});
+	});
+
+	test('should indent after simple jsx tag with attributes', () => {
+		return withRandomFileEditor(`const a = <div onclick={bla}>${CURSOR}`, 'jsx', async (_editor, document) => {
+			await type(document, '\nx');
+			assert.strictEqual(document.getText(), `const a = <div onclick={bla}>\n    x`);
+		});
+	});
+});

extensions/typescript-language-features/src/test/testUtils.ts

@@ -0,0 +1,68 @@
+/*---------------------------------------------------------------------------------------------
+ *  Copyright (c) Microsoft Corporation. All rights reserved.
+ *  Licensed under the MIT License. See License.txt in the project root for license information.
+ *--------------------------------------------------------------------------------------------*/
+
+import * as vscode from 'vscode';
+import * as fs from 'fs';
+import * as os from 'os';
+import { join } from 'path';
+
+function rndName() {
+	return Math.random().toString(36).replace(/[^a-z]+/g, '').substr(0, 10);
+}
+
+export function createRandomFile(contents = '', fileExtension = 'txt'): Thenable<vscode.Uri> {
+	return new Promise((resolve, reject) => {
+		const tmpFile = join(os.tmpdir(), rndName() + '.' + fileExtension);
+		fs.writeFile(tmpFile, contents, (error) => {
+			if (error) {
+				return reject(error);
+			}
+
+			resolve(vscode.Uri.file(tmpFile));
+		});
+	});
+}
+
+
+export function deleteFile(file: vscode.Uri): Thenable<boolean> {
+	return new Promise((resolve, reject) => {
+		fs.unlink(file.fsPath, (err) => {
+			if (err) {
+				reject(err);
+			} else {
+				resolve(true);
+			}
+		});
+	});
+}
+
+export const CURSOR = '$$CURSOR$$';
+
+export function withRandomFileEditor(
+	contents: string,
+	fileExtension: string,
+	run: (editor: vscode.TextEditor, doc: vscode.TextDocument) => Thenable<void>
+): Thenable<boolean> {
+	const cursorIndex = contents.indexOf(CURSOR);
+	return createRandomFile(contents.replace(CURSOR, ''), fileExtension).then(file => {
+		return vscode.workspace.openTextDocument(file).then(doc => {
+			return vscode.window.showTextDocument(doc).then((editor) => {
+				if (cursorIndex >= 0) {
+					const pos = doc.positionAt(cursorIndex);
+					editor.selection = new vscode.Selection(pos, pos);
+				}
+				return run(editor, doc).then(_ => {
+					if (doc.isDirty) {
+						return doc.save().then(() => {
+							return deleteFile(file);
+						});
+					} else {
+						return deleteFile(file);
+					}
+				});
+			});
+		});
+	});
+}