Use CancellationTokenSource instead of casting dummy objects
This commit is contained in:
@@ -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 = () => {
|
||||||
|
|||||||
@@ -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);
|
||||||
});
|
});
|
||||||
|
|||||||
@@ -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) {
|
||||||
|
|||||||
@@ -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) {
|
||||||
|
|||||||
Reference in New Issue
Block a user