Use CancellationTokenSource instead of casting dummy objects

This commit is contained in:
Robert
2024-03-06 12:28:30 +00:00
parent 83a6b5a054
commit 2bc261eacf
4 changed files with 11 additions and 28 deletions

View File

@@ -6,7 +6,7 @@ import type {
QueryWithResults, QueryWithResults,
} from "../../../src/run-queries-shared"; } from "../../../src/run-queries-shared";
import { QueryOutputDir } from "../../../src/local-queries/query-output-dir"; import { QueryOutputDir } from "../../../src/local-queries/query-output-dir";
import type { CancellationTokenSource } from "vscode"; import { CancellationTokenSource } from "vscode";
import type { QueryMetadata } from "../../../src/common/interface-types"; import type { QueryMetadata } from "../../../src/common/interface-types";
import type { QueryLanguage } from "../../../src/common/query-language"; import type { QueryLanguage } from "../../../src/common/query-language";
@@ -31,12 +31,6 @@ export function createMockLocalQueryInfo({
language?: QueryLanguage; language?: QueryLanguage;
outputDir?: QueryOutputDir | undefined; outputDir?: QueryOutputDir | undefined;
}): LocalQueryInfo { }): LocalQueryInfo {
const cancellationToken = {
dispose: () => {
/**/
},
} as CancellationTokenSource;
const initialQueryInfo = { const initialQueryInfo = {
queryText: "select 1", queryText: "select 1",
isQuickQuery: false, isQuickQuery: false,
@@ -54,7 +48,10 @@ export function createMockLocalQueryInfo({
outputDir, outputDir,
} as InitialQueryInfo; } as InitialQueryInfo;
const localQuery = new LocalQueryInfo(initialQueryInfo, cancellationToken); const localQuery = new LocalQueryInfo(
initialQueryInfo,
new CancellationTokenSource(),
);
localQuery.failureReason = failureReason; localQuery.failureReason = failureReason;
localQuery.cancel = () => { localQuery.cancel = () => {

View File

@@ -1,5 +1,5 @@
import type { CancellationToken, ExtensionContext, Range } from "vscode"; import type { CancellationToken, ExtensionContext, Range } from "vscode";
import { Uri } from "vscode"; import { CancellationTokenSource, Uri } from "vscode";
import { join, dirname } from "path"; import { join, dirname } from "path";
import { import {
pathExistsSync, pathExistsSync,
@@ -125,11 +125,7 @@ describeWithCodeQL()("Queries", () => {
safeDel(qlFile); safeDel(qlFile);
safeDel(qlpackFile); safeDel(qlpackFile);
token = { token = new CancellationTokenSource().token;
onCancellationRequested: (_) => {
void _;
},
} as CancellationToken;
dbItem = await ensureTestDatabase(databaseManager, cli); dbItem = await ensureTestDatabase(databaseManager, cli);
}); });

View File

@@ -10,8 +10,7 @@ import type { QueryWithResults } from "../../../../../src/run-queries-shared";
import { QueryEvaluationInfo } from "../../../../../src/run-queries-shared"; import { QueryEvaluationInfo } from "../../../../../src/run-queries-shared";
import { QueryOutputDir } from "../../../../../src/local-queries/query-output-dir"; import { QueryOutputDir } from "../../../../../src/local-queries/query-output-dir";
import type { DatabaseInfo } from "../../../../../src/common/interface-types"; import type { DatabaseInfo } from "../../../../../src/common/interface-types";
import type { CancellationTokenSource } from "vscode"; import { CancellationTokenSource, Uri } from "vscode";
import { Uri } from "vscode";
import { tmpDir } from "../../../../../src/tmp-dir"; import { tmpDir } from "../../../../../src/tmp-dir";
import type { QueryHistoryInfo } from "../../../../../src/query-history/query-history-info"; import type { QueryHistoryInfo } from "../../../../../src/query-history/query-history-info";
import { createMockVariantAnalysisHistoryItem } from "../../../../factories/query-history/variant-analysis-history-item"; import { createMockVariantAnalysisHistoryItem } from "../../../../factories/query-history/variant-analysis-history-item";
@@ -220,11 +219,7 @@ describe("write and read", () => {
id: `some-id-${dbName}`, id: `some-id-${dbName}`,
outputDir: outputDir ? outputDir : undefined, outputDir: outputDir ? outputDir : undefined,
} as InitialQueryInfo, } as InitialQueryInfo,
{ new CancellationTokenSource(),
dispose: () => {
/**/
},
} as CancellationTokenSource,
); );
if (queryWithResults) { if (queryWithResults) {

View File

@@ -21,8 +21,7 @@ import type {
SortedResultSetInfo, SortedResultSetInfo,
} from "../../../src/common/interface-types"; } from "../../../src/common/interface-types";
import type { CodeQLCliServer, SourceInfo } from "../../../src/codeql-cli/cli"; import type { CodeQLCliServer, SourceInfo } from "../../../src/codeql-cli/cli";
import type { CancellationTokenSource } from "vscode"; import { CancellationTokenSource, Uri } from "vscode";
import { Uri } from "vscode";
import { tmpDir } from "../../../src/tmp-dir"; import { tmpDir } from "../../../src/tmp-dir";
import { sleep } from "../../../src/common/time"; import { sleep } from "../../../src/common/time";
import { mockedObject } from "../utils/mocking.helpers"; import { mockedObject } from "../utils/mocking.helpers";
@@ -462,11 +461,7 @@ describe("query-results", () => {
id: `some-id-${dbName}`, id: `some-id-${dbName}`,
outputDir: new QueryOutputDir("path/to/output/dir"), outputDir: new QueryOutputDir("path/to/output/dir"),
} as InitialQueryInfo, } as InitialQueryInfo,
{ new CancellationTokenSource(),
dispose: () => {
/**/
},
} as CancellationTokenSource,
); );
if (queryWithResults) { if (queryWithResults) {