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

View File

@@ -1,5 +1,5 @@
import type { CancellationToken, ExtensionContext, Range } from "vscode";
import { Uri } from "vscode";
import { CancellationTokenSource, Uri } from "vscode";
import { join, dirname } from "path";
import {
pathExistsSync,
@@ -125,11 +125,7 @@ describeWithCodeQL()("Queries", () => {
safeDel(qlFile);
safeDel(qlpackFile);
token = {
onCancellationRequested: (_) => {
void _;
},
} as CancellationToken;
token = new CancellationTokenSource().token;
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 { QueryOutputDir } from "../../../../../src/local-queries/query-output-dir";
import type { DatabaseInfo } from "../../../../../src/common/interface-types";
import type { CancellationTokenSource } from "vscode";
import { Uri } from "vscode";
import { CancellationTokenSource, Uri } from "vscode";
import { tmpDir } from "../../../../../src/tmp-dir";
import type { QueryHistoryInfo } from "../../../../../src/query-history/query-history-info";
import { createMockVariantAnalysisHistoryItem } from "../../../../factories/query-history/variant-analysis-history-item";
@@ -220,11 +219,7 @@ describe("write and read", () => {
id: `some-id-${dbName}`,
outputDir: outputDir ? outputDir : undefined,
} as InitialQueryInfo,
{
dispose: () => {
/**/
},
} as CancellationTokenSource,
new CancellationTokenSource(),
);
if (queryWithResults) {

View File

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