ckaczor/azuredatastudio
1
0
Fork 0
mirror of https://github.com/ckaczor/azuredatastudio.git synced 2026-01-20 01:25:37 -05:00
Code Issues Packages Projects Releases Wiki Activity

Query Runner Tests (#10252)

Browse Source 
* rework some code and write an inital test

* fix strict

* add more to standard test

* add to existing workflow test

* fix tests

* simplify the code

* add more tests

* remove bad import

* fix compile

* fix timestampiong
This commit is contained in:
Anthony Dresser
2020-05-06 13:38:12 -07:00
committed by GitHub
parent 4199cec393
commit df5df38a55
25 changed files with 856 additions and 430 deletions
Download Patch File Download Diff File Expand all files Collapse all files

21
src/sql/workbench/common/editor/query/queryEditorInput.ts
View File

@@ -14,8 +14,9 @@ import { IConnectionManagementService, IConnectableInput, INewConnectionParams,
import { QueryResultsInput } from 'sql/workbench/common/editor/query/queryResultsInput';
import { IQueryModelService } from 'sql/workbench/services/query/common/queryModel';
import { ISelectionData, ExecutionPlanOptions } from 'azdata';
import { ExecutionPlanOptions } from 'azdata';
import { startsWith } from 'vs/base/common/strings';
import { IRange } from 'vs/editor/common/core/range';
const MAX_SIZE = 13;
@@ -229,13 +230,13 @@ export abstract class QueryEditorInput extends EditorInput implements IConnectab
}
// State update funtions
public runQuery(selection?: ISelectionData, executePlanOptions?: ExecutionPlanOptions): void {
this.queryModelService.runQuery(this.uri, selection, executePlanOptions);
public runQuery(range?: IRange, executePlanOptions?: ExecutionPlanOptions): void {
this.queryModelService.runQuery(this.uri, range, executePlanOptions);
this.state.executing = true;
}
public runQueryStatement(selection?: ISelectionData): void {
this.queryModelService.runQueryStatement(this.uri, selection);
public runQueryStatement(range?: IRange): void {
this.queryModelService.runQueryStatement(this.uri, range);
this.state.executing = true;
}
@@ -269,15 +270,15 @@ export abstract class QueryEditorInput extends EditorInput implements IConnectab
let isRunningQuery = this.queryModelService.isRunningQuery(this.uri);
if (!isRunningQuery && params && params.runQueryOnCompletion) {
let selection: ISelectionData | undefined = params ? params.querySelection : undefined;
let range: IRange | undefined = params ? params.queryRange : undefined;
if (params.runQueryOnCompletion === RunQueryOnConnectionMode.executeCurrentQuery) {
this.runQueryStatement(selection);
this.runQueryStatement(range);
} else if (params.runQueryOnCompletion === RunQueryOnConnectionMode.executeQuery) {
this.runQuery(selection);
this.runQuery(range);
} else if (params.runQueryOnCompletion === RunQueryOnConnectionMode.estimatedQueryPlan) {
this.runQuery(selection, { displayEstimatedQueryPlan: true });
this.runQuery(range, { displayEstimatedQueryPlan: true });
} else if (params.runQueryOnCompletion === RunQueryOnConnectionMode.actualQueryPlan) {
this.runQuery(selection, { displayActualQueryPlan: true });
this.runQuery(range, { displayActualQueryPlan: true });
}
}
this._onDidChangeLabel.fire();