Skip to content

Commit b2f0e2e

Browse files
Convert handleCancel to use createMultiSelectionCommand
1 parent 28819df commit b2f0e2e

2 files changed

Lines changed: 13 additions & 24 deletions

File tree

extensions/ql-vscode/src/query-history/query-history-manager.ts

Lines changed: 5 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -273,7 +273,9 @@ export class QueryHistoryManager extends DisposableObject {
273273
this.handleOpenQueryDirectory.bind(this),
274274
"query",
275275
),
276-
"codeQLQueryHistory.cancel": this.handleCancel.bind(this),
276+
"codeQLQueryHistory.cancel": createMultiSelectionCommand(
277+
this.handleCancel.bind(this),
278+
),
277279
"codeQLQueryHistory.exportResults": this.handleExportResults.bind(this),
278280
"codeQLQueryHistory.viewCsvResults": this.handleViewCsvResults.bind(this),
279281
"codeQLQueryHistory.viewCsvAlerts": this.handleViewCsvAlerts.bind(this),
@@ -791,13 +793,8 @@ export class QueryHistoryManager extends DisposableObject {
791793
}
792794
}
793795

794-
async handleCancel(
795-
singleItem: QueryHistoryInfo,
796-
multiSelect: QueryHistoryInfo[] | undefined,
797-
) {
798-
multiSelect ||= [singleItem];
799-
800-
const results = multiSelect.map(async (item) => {
796+
async handleCancel(items: QueryHistoryInfo[]) {
797+
const results = items.map(async (item) => {
801798
if (item.status === QueryStatus.InProgress) {
802799
if (item.t === "local") {
803800
item.cancel();

extensions/ql-vscode/test/vscode-tests/no-workspace/query-history/query-history-manager.test.ts

Lines changed: 8 additions & 16 deletions
Original file line numberDiff line numberDiff line change
@@ -591,7 +591,7 @@ describe("QueryHistoryManager", () => {
591591
const inProgress1 = localQueryHistory[4];
592592
const cancelSpy = jest.spyOn(inProgress1, "cancel");
593593

594-
await queryHistoryManager.handleCancel(inProgress1, [inProgress1]);
594+
await queryHistoryManager.handleCancel([inProgress1]);
595595
expect(cancelSpy).toBeCalledTimes(1);
596596
});
597597

@@ -605,10 +605,7 @@ describe("QueryHistoryManager", () => {
605605
const cancelSpy1 = jest.spyOn(inProgress1, "cancel");
606606
const cancelSpy2 = jest.spyOn(inProgress2, "cancel");
607607

608-
await queryHistoryManager.handleCancel(inProgress1, [
609-
inProgress1,
610-
inProgress2,
611-
]);
608+
await queryHistoryManager.handleCancel([inProgress1, inProgress2]);
612609
expect(cancelSpy1).toBeCalled();
613610
expect(cancelSpy2).toBeCalled();
614611
});
@@ -619,7 +616,7 @@ describe("QueryHistoryManager", () => {
619616
// cancelling the selected item
620617
const inProgress1 = variantAnalysisHistory[1];
621618

622-
await queryHistoryManager.handleCancel(inProgress1, [inProgress1]);
619+
await queryHistoryManager.handleCancel([inProgress1]);
623620
expect(cancelVariantAnalysisSpy).toBeCalledWith(
624621
inProgress1.variantAnalysis.id,
625622
);
@@ -632,10 +629,7 @@ describe("QueryHistoryManager", () => {
632629
const inProgress1 = variantAnalysisHistory[1];
633630
const inProgress2 = variantAnalysisHistory[3];
634631

635-
await queryHistoryManager.handleCancel(inProgress1, [
636-
inProgress1,
637-
inProgress2,
638-
]);
632+
await queryHistoryManager.handleCancel([inProgress1, inProgress2]);
639633
expect(cancelVariantAnalysisSpy).toBeCalledWith(
640634
inProgress1.variantAnalysis.id,
641635
);
@@ -653,7 +647,7 @@ describe("QueryHistoryManager", () => {
653647
const completed = localQueryHistory[0];
654648
const cancelSpy = jest.spyOn(completed, "cancel");
655649

656-
await queryHistoryManager.handleCancel(completed, [completed]);
650+
await queryHistoryManager.handleCancel([completed]);
657651
expect(cancelSpy).not.toBeCalledTimes(1);
658652
});
659653

@@ -667,7 +661,7 @@ describe("QueryHistoryManager", () => {
667661
const cancelSpy = jest.spyOn(completed, "cancel");
668662
const cancelSpy2 = jest.spyOn(failed, "cancel");
669663

670-
await queryHistoryManager.handleCancel(completed, [completed, failed]);
664+
await queryHistoryManager.handleCancel([completed, failed]);
671665
expect(cancelSpy).not.toBeCalledTimes(1);
672666
expect(cancelSpy2).not.toBeCalledTimes(1);
673667
});
@@ -678,9 +672,7 @@ describe("QueryHistoryManager", () => {
678672
// cancelling the selected item
679673
const completedVariantAnalysis = variantAnalysisHistory[0];
680674

681-
await queryHistoryManager.handleCancel(completedVariantAnalysis, [
682-
completedVariantAnalysis,
683-
]);
675+
await queryHistoryManager.handleCancel([completedVariantAnalysis]);
684676
expect(cancelVariantAnalysisSpy).not.toBeCalledWith(
685677
completedVariantAnalysis.variantAnalysis,
686678
);
@@ -693,7 +685,7 @@ describe("QueryHistoryManager", () => {
693685
const completedVariantAnalysis = variantAnalysisHistory[0];
694686
const failedVariantAnalysis = variantAnalysisHistory[2];
695687

696-
await queryHistoryManager.handleCancel(completedVariantAnalysis, [
688+
await queryHistoryManager.handleCancel([
697689
completedVariantAnalysis,
698690
failedVariantAnalysis,
699691
]);

0 commit comments

Comments
 (0)