14eb6b4f89
The test database is not required for activated extension tests, so we only need to download the test database for CLI integration tests.
90 lines
2.7 KiB
TypeScript
90 lines
2.7 KiB
TypeScript
import { workspace } from "vscode";
|
|
|
|
import {
|
|
beforeAllAction,
|
|
beforeEachAction,
|
|
} from "../jest.activated-extension.setup";
|
|
import * as tmp from "tmp";
|
|
import {
|
|
createWriteStream,
|
|
existsSync,
|
|
mkdirpSync,
|
|
realpathSync,
|
|
} from "fs-extra";
|
|
import { dirname } from "path";
|
|
import { DB_URL, dbLoc, setStoragePath, storagePath } from "../global.helper";
|
|
import fetch from "node-fetch";
|
|
|
|
// create an extension storage location
|
|
let removeStorage: tmp.DirResult["removeCallback"] | undefined;
|
|
|
|
beforeAll(async () => {
|
|
// ensure the test database is downloaded
|
|
mkdirpSync(dirname(dbLoc));
|
|
if (!existsSync(dbLoc)) {
|
|
console.log(`Downloading test database to ${dbLoc}`);
|
|
|
|
await new Promise((resolve, reject) => {
|
|
return fetch(DB_URL).then((response) => {
|
|
const dest = createWriteStream(dbLoc);
|
|
response.body.pipe(dest);
|
|
|
|
response.body.on("error", reject);
|
|
dest.on("error", reject);
|
|
dest.on("close", () => {
|
|
resolve(dbLoc);
|
|
});
|
|
});
|
|
});
|
|
}
|
|
|
|
// Create the temp directory to be used as extension local storage.
|
|
const dir = tmp.dirSync();
|
|
let storagePath = realpathSync(dir.name);
|
|
if (storagePath.substring(0, 2).match(/[A-Z]:/)) {
|
|
storagePath =
|
|
storagePath.substring(0, 1).toLocaleLowerCase() +
|
|
storagePath.substring(1);
|
|
}
|
|
setStoragePath(storagePath);
|
|
|
|
removeStorage = dir.removeCallback;
|
|
|
|
await beforeAllAction();
|
|
});
|
|
|
|
beforeEach(async () => {
|
|
await beforeEachAction();
|
|
});
|
|
|
|
beforeAll(() => {
|
|
// check that the codeql folder is found in the workspace
|
|
const folders = workspace.workspaceFolders;
|
|
if (!folders) {
|
|
throw new Error(
|
|
'No workspace folders found.\nYou will need a local copy of the codeql repo.\nMake sure you specify the path to it in launch.json.\nIt should be something along the lines of "${workspaceRoot}/../codeql" depending on where you have your local copy of the codeql repo.',
|
|
);
|
|
} else {
|
|
const codeqlFolder = folders.find((folder) =>
|
|
["codeql", "ql"].includes(folder.name),
|
|
);
|
|
if (!codeqlFolder) {
|
|
throw new Error(
|
|
'No workspace folders found.\nYou will need a local copy of the codeql repo.\nMake sure you specify the path to it in launch.json.\nIt should be something along the lines of "${workspaceRoot}/../codeql" depending on where you have your local copy of the codeql repo.\n\n\n',
|
|
);
|
|
}
|
|
}
|
|
});
|
|
|
|
// ensure extension is cleaned up.
|
|
afterAll(async () => {
|
|
// ensure temp directory is cleaned up.
|
|
try {
|
|
removeStorage?.();
|
|
} catch (e) {
|
|
// we are exiting anyway so don't worry about it.
|
|
// most likely the directory this is a test on Windows and some files are locked.
|
|
console.warn(`Failed to remove storage directory '${storagePath}': ${e}`);
|
|
}
|
|
});
|