Skip to content
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
18 changes: 0 additions & 18 deletions extensions/ql-vscode/src/extension.ts
Original file line number Diff line number Diff line change
Expand Up @@ -1145,15 +1145,6 @@ async function activateWithInstalledDistribution(
),
);

ctx.subscriptions.push(
commandRunner(
"codeQL.cancelVariantAnalysis",
async (variantAnalysisId: number) => {
await variantAnalysisManager.cancelVariantAnalysis(variantAnalysisId);
},
),
);

ctx.subscriptions.push(
commandRunner("codeQL.exportSelectedVariantAnalysisResults", async () => {
await exportSelectedVariantAnalysisResults(qhm);
Expand Down Expand Up @@ -1207,15 +1198,6 @@ async function activateWithInstalledDistribution(
),
);

ctx.subscriptions.push(
commandRunner(
"codeQL.openVariantAnalysisQueryText",
async (variantAnalysisId: number) => {
await variantAnalysisManager.openQueryText(variantAnalysisId);
},
),
);

ctx.subscriptions.push(
commandRunner("codeQL.openReferencedFile", async (selectedQuery: Uri) => {
await openReferencedFile(qs, cliServer, selectedQuery);
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -1008,8 +1008,7 @@ export class QueryHistoryManager extends DisposableObject {
if (item.t === "local") {
item.cancel();
} else if (item.t === "variant-analysis") {
await commands.executeCommand(
"codeQL.cancelVariantAnalysis",
await this.variantAnalysisManager.cancelVariantAnalysis(
item.variantAnalysis.id,
);
} else {
Expand All @@ -1035,8 +1034,7 @@ export class QueryHistoryManager extends DisposableObject {
}

if (finalSingleItem.t === "variant-analysis") {
await commands.executeCommand(
"codeQL.openVariantAnalysisQueryText",
await this.variantAnalysisManager.openQueryText(
finalSingleItem.variantAnalysis.id,
);
return;
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -25,4 +25,6 @@ export interface VariantAnalysisViewManager<
variantAnalysisId: number,
): Promise<VariantAnalysisScannedRepositoryState[]>;
openQueryFile(variantAnalysisId: number): Promise<void>;
openQueryText(variantAnalysisId: number): Promise<void>;
cancelVariantAnalysis(variantAnalysisId: number): Promise<void>;
}
Original file line number Diff line number Diff line change
Expand Up @@ -110,10 +110,7 @@ export class VariantAnalysisView

break;
case "cancelVariantAnalysis":
void commands.executeCommand(
"codeQL.cancelVariantAnalysis",
this.variantAnalysisId,
);
await this.manager.cancelVariantAnalysis(this.variantAnalysisId);
break;
case "requestRepositoryResults":
void commands.executeCommand(
Expand All @@ -126,10 +123,7 @@ export class VariantAnalysisView
await this.manager.openQueryFile(this.variantAnalysisId);
break;
case "openQueryText":
void commands.executeCommand(
"codeQL.openVariantAnalysisQueryText",
this.variantAnalysisId,
);
await this.manager.openQueryText(this.variantAnalysisId);
break;
case "copyRepositoryList":
void commands.executeCommand(
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -32,12 +32,14 @@ const openQueryText = () => {
vscode.postMessage({
t: "openQueryText",
});
sendTelemetry("variant-analysis-open-query-text");
};

const stopQuery = () => {
vscode.postMessage({
t: "cancelVariantAnalysis",
});
sendTelemetry("variant-analysis-cancel");
};

const openLogs = () => {
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -41,6 +41,9 @@ describe("QueryHistoryManager", () => {
let executeCommandSpy: jest.SpiedFunction<
typeof vscode.commands.executeCommand
>;
let cancelVariantAnalysisSpy: jest.SpiedFunction<
typeof variantAnalysisManagerStub.cancelVariantAnalysis
>;
const doCompareCallback = jest.fn();

let queryHistoryManager: QueryHistoryManager;
Expand Down Expand Up @@ -82,9 +85,14 @@ describe("QueryHistoryManager", () => {
onVariantAnalysisStatusUpdated: jest.fn(),
onVariantAnalysisRemoved: jest.fn(),
removeVariantAnalysis: jest.fn(),
cancelVariantAnalysis: jest.fn(),
showView: jest.fn(),
} as any as VariantAnalysisManager;

cancelVariantAnalysisSpy = jest
.spyOn(variantAnalysisManagerStub, "cancelVariantAnalysis")
.mockResolvedValue(undefined);

localQueryHistory = [
// completed
createMockLocalQueryInfo({
Expand Down Expand Up @@ -729,8 +737,7 @@ describe("QueryHistoryManager", () => {
const inProgress1 = variantAnalysisHistory[1];

await queryHistoryManager.handleCancel(inProgress1, [inProgress1]);
expect(executeCommandSpy).toBeCalledWith(
"codeQL.cancelVariantAnalysis",
expect(cancelVariantAnalysisSpy).toBeCalledWith(
inProgress1.variantAnalysis.id,
);
});
Expand All @@ -746,12 +753,10 @@ describe("QueryHistoryManager", () => {
inProgress1,
inProgress2,
]);
expect(executeCommandSpy).toBeCalledWith(
"codeQL.cancelVariantAnalysis",
expect(cancelVariantAnalysisSpy).toBeCalledWith(
inProgress1.variantAnalysis.id,
);
expect(executeCommandSpy).toBeCalledWith(
"codeQL.cancelVariantAnalysis",
expect(cancelVariantAnalysisSpy).toBeCalledWith(
inProgress2.variantAnalysis.id,
);
});
Expand Down Expand Up @@ -793,8 +798,7 @@ describe("QueryHistoryManager", () => {
await queryHistoryManager.handleCancel(completedVariantAnalysis, [
completedVariantAnalysis,
]);
expect(executeCommandSpy).not.toBeCalledWith(
"codeQL.cancelVariantAnalysis",
expect(cancelVariantAnalysisSpy).not.toBeCalledWith(
completedVariantAnalysis.variantAnalysis,
);
});
Expand All @@ -810,12 +814,10 @@ describe("QueryHistoryManager", () => {
completedVariantAnalysis,
failedVariantAnalysis,
]);
expect(executeCommandSpy).not.toBeCalledWith(
"codeQL.cancelVariantAnalysis",
expect(cancelVariantAnalysisSpy).not.toBeCalledWith(
completedVariantAnalysis.variantAnalysis.id,
);
expect(executeCommandSpy).not.toBeCalledWith(
"codeQL.cancelVariantAnalysis",
expect(cancelVariantAnalysisSpy).not.toBeCalledWith(
failedVariantAnalysis.variantAnalysis.id,
);
});
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -8,7 +8,7 @@ import {
} from "fs-extra";
import { join } from "path";

import { commands, ExtensionContext, Uri } from "vscode";
import { ExtensionContext, Uri } from "vscode";
import { DatabaseManager } from "../../../../src/local-databases";
import { tmpDir, walkDirectory } from "../../../../src/helpers";
import { DisposableBucket } from "../../disposable-bucket";
Expand Down Expand Up @@ -54,9 +54,12 @@ describe("Variant Analyses and QueryHistoryManager", () => {
rehydrateVariantAnalysis: rehydrateVariantAnalysisStub,
onVariantAnalysisStatusUpdated: jest.fn(),
showView: showViewStub,
openQueryText: jest.fn(),
} as any as VariantAnalysisManager;

let executeCommandSpy: jest.SpiedFunction<typeof commands.executeCommand>;
let openQueryTextSpy: jest.SpiedFunction<
typeof variantAnalysisManagerStub.openQueryText
>;

beforeEach(async () => {
// Since these tests change the state of the query history manager, we need to copy the original
Expand Down Expand Up @@ -95,8 +98,8 @@ describe("Variant Analyses and QueryHistoryManager", () => {
);
disposables.push(qhm);

executeCommandSpy = jest
.spyOn(commands, "executeCommand")
openQueryTextSpy = jest
.spyOn(variantAnalysisManagerStub, "openQueryText")
.mockResolvedValue(undefined);
});

Expand Down Expand Up @@ -180,8 +183,7 @@ describe("Variant Analyses and QueryHistoryManager", () => {
await qhm.readQueryHistory();
await qhm.handleShowQueryText(qhm.treeDataProvider.allHistory[0], []);

expect(executeCommandSpy).toHaveBeenCalledWith(
"codeQL.openVariantAnalysisQueryText",
expect(openQueryTextSpy).toHaveBeenCalledWith(
rawQueryHistory[0].variantAnalysis.id,
);
});
Expand Down