1c0d6b991a
This will ensure all mocks are restored after every test. This required a significant amount of changes in the tests since `jest.spyOn` now needs to be called in `beforeEach`, rather than in the `describe` block.
113 lines
3.2 KiB
TypeScript
113 lines
3.2 KiB
TypeScript
import * as path from "path";
|
|
import { extensions, CancellationToken, Uri, window } from "vscode";
|
|
|
|
import { CodeQLExtensionInterface } from "../../extension";
|
|
import { CodeQLCliServer } from "../../cli";
|
|
import { DatabaseManager } from "../../databases";
|
|
import {
|
|
promptImportLgtmDatabase,
|
|
importArchiveDatabase,
|
|
promptImportInternetDatabase,
|
|
} from "../../databaseFetcher";
|
|
import { cleanDatabases, dbLoc, DB_URL, storagePath } from "./global.helper";
|
|
|
|
jest.setTimeout(60_000);
|
|
|
|
/**
|
|
* Run various integration tests for databases
|
|
*/
|
|
describe("Databases", () => {
|
|
const LGTM_URL =
|
|
"https://lgtm.com/projects/g/aeisenberg/angular-bind-notifier/";
|
|
|
|
let databaseManager: DatabaseManager;
|
|
let inputBoxStub: jest.SpiedFunction<typeof window.showInputBox>;
|
|
let cli: CodeQLCliServer;
|
|
const progressCallback = jest.fn();
|
|
|
|
beforeEach(async () => {
|
|
inputBoxStub = jest
|
|
.spyOn(window, "showInputBox")
|
|
.mockResolvedValue(undefined);
|
|
|
|
jest.spyOn(window, "showErrorMessage").mockResolvedValue(undefined);
|
|
jest.spyOn(window, "showInformationMessage").mockResolvedValue(undefined);
|
|
|
|
const extension = await extensions
|
|
.getExtension<CodeQLExtensionInterface | Record<string, never>>(
|
|
"GitHub.vscode-codeql",
|
|
)!
|
|
.activate();
|
|
if ("databaseManager" in extension) {
|
|
databaseManager = extension.databaseManager;
|
|
} else {
|
|
throw new Error(
|
|
"Extension not initialized. Make sure cli is downloaded and installed properly.",
|
|
);
|
|
}
|
|
|
|
await cleanDatabases(databaseManager);
|
|
});
|
|
|
|
afterEach(async () => {
|
|
await cleanDatabases(databaseManager);
|
|
});
|
|
|
|
it("should add a database from a folder", async () => {
|
|
const uri = Uri.file(dbLoc);
|
|
let dbItem = await importArchiveDatabase(
|
|
uri.toString(true),
|
|
databaseManager,
|
|
storagePath,
|
|
progressCallback,
|
|
{} as CancellationToken,
|
|
cli,
|
|
);
|
|
expect(dbItem).toBe(databaseManager.currentDatabaseItem);
|
|
expect(dbItem).toBe(databaseManager.databaseItems[0]);
|
|
expect(dbItem).toBeDefined();
|
|
dbItem = dbItem!;
|
|
expect(dbItem.name).toBe("db");
|
|
expect(dbItem.databaseUri.fsPath).toBe(path.join(storagePath, "db", "db"));
|
|
});
|
|
|
|
it("should add a database from lgtm with only one language", async () => {
|
|
inputBoxStub.mockResolvedValue(LGTM_URL);
|
|
let dbItem = await promptImportLgtmDatabase(
|
|
databaseManager,
|
|
storagePath,
|
|
progressCallback,
|
|
{} as CancellationToken,
|
|
cli,
|
|
);
|
|
expect(dbItem).toBeDefined();
|
|
dbItem = dbItem!;
|
|
expect(dbItem.name).toBe("aeisenberg_angular-bind-notifier_106179a");
|
|
expect(dbItem.databaseUri.fsPath).toBe(
|
|
path.join(
|
|
storagePath,
|
|
"javascript",
|
|
"aeisenberg_angular-bind-notifier_106179a",
|
|
),
|
|
);
|
|
});
|
|
|
|
it("should add a database from a url", async () => {
|
|
inputBoxStub.mockResolvedValue(DB_URL);
|
|
|
|
let dbItem = await promptImportInternetDatabase(
|
|
databaseManager,
|
|
storagePath,
|
|
progressCallback,
|
|
{} as CancellationToken,
|
|
cli,
|
|
);
|
|
expect(dbItem).toBeDefined();
|
|
dbItem = dbItem!;
|
|
expect(dbItem.name).toBe("db");
|
|
expect(dbItem.databaseUri.fsPath).toBe(
|
|
path.join(storagePath, "simple-db", "db"),
|
|
);
|
|
});
|
|
});
|