Add test for opening existing SQL file and typing text into it (#5816)

* Add test for opening existing SQL file and typing text into it

* Clean up

* More cleanup, remove unneeded queryEditor and add smoke test scripts

* Update comments to be clearer

test/smoke/src/sql/queryEditor/queryEditor.test.ts

@@ -0,0 +1,28 @@
+/*---------------------------------------------------------------------------------------------
+ *  Copyright (c) Microsoft Corporation. All rights reserved.
+ *  Licensed under the Source EULA. See License.txt in the project root for license information.
+ *--------------------------------------------------------------------------------------------*/
+
+import { Application } from '../../application';
+import * as fs from 'fs';
+import * as os from 'os';
+import * as path from 'path';
+
+export function setup() {
+	describe('Query Editor Test Suite', () => {
+
+		it('Can open and edit existing file', async function () {
+			const testFilePath = path.join(os.tmpdir(), 'QueryEditorSmokeTest.sql');
+			fs.writeFileSync(testFilePath, '');
+			try {
+				const app = this.app as Application;
+				await app.workbench.queryEditors.openFile(testFilePath);
+				const fileBaseName = path.basename(testFilePath);
+				await app.workbench.editor.waitForTypeInEditor(fileBaseName, 'SELECT * FROM sys.tables');
+			}
+			finally {
+				fs.unlinkSync(testFilePath);
+			}
+		});
+	});
+}

test/smoke/src/sql/queryEditor/queryEditors.ts

@@ -0,0 +1,54 @@
+/*---------------------------------------------------------------------------------------------
+ *  Copyright (c) Microsoft Corporation. All rights reserved.
+ *  Licensed under the Source EULA. See License.txt in the project root for license information.
+ *--------------------------------------------------------------------------------------------*/
+
+import { Editors } from '../../areas/editor/editors';
+import { QuickOpen } from '../../areas/quickopen/quickopen';
+import { Code } from '../../vscode/code';
+import * as path from 'path';
+
+export class QueryEditors extends Editors {
+
+	constructor(
+		private vsCode: Code,
+		private quickopen: QuickOpen
+	) {
+		super(vsCode);
+	}
+
+	/**
+	 * Opens the specified file - this correctly handles SQL files which are opened in a Query Editor window
+	 * @param filePath The full path of the file to open.
+	 */
+	async openFile(filePath: string): Promise<void> {
+		await this.quickopen.openQuickOpen(filePath);
+
+		const fileBaseName = path.basename(filePath);
+		await this.quickopen.waitForQuickOpenElements(names => names[0] === fileBaseName);
+		await this.vsCode.dispatchKeybinding('enter');
+		await this.waitForEditorFocus(fileBaseName);
+	}
+
+	/**
+	 * Waits for an active SQL Query Editor tab for the specified file. This is a modification of the editors.waitForActiveTab that
+	 * takes into account the connected status displayed in the title of Query Editors.
+	 * @param fileName The name of the file opened in the editor
+	 * @param isDirty Whether the file is dirty or not
+	 */
+	async waitForActiveTab(fileName: string, isDirty: boolean = false): Promise<void> {
+		// For now assume all opened tabs are disconnected until we have a need to open connected tabs
+		await this.vsCode.waitForElement(`.tabs-container div.tab.active${isDirty ? '.dirty' : ''}[aria-selected="true"][aria-label="${fileName} - disconnected, tab"]`);
+	}
+
+
+	/**
+	 * Waits for an active Query Editor for the specified file to have focus. This is a modification of the editors.waitForEditorFocus
+	 * that takes into account the connected status displayed in the title of Query Editors.
+	 * @param fileName The name of the file opened in the editor
+	 */
+	async waitForEditorFocus(fileName: string): Promise<void> {
+		await this.waitForActiveTab(fileName);
+		await super.waitForActiveEditor(fileName);
+	}
+}