wehub-resource-sync
108 lines
4.1 KiB
TypeScript
Vendored
108 lines
4.1 KiB
TypeScript
Vendored
import { readFileSync } from "node:fs";
|
|
import { strict as assert } from "node:assert";
|
|
import { test } from "vitest";
|
|
|
|
function readSource(path: string): string {
|
|
return readFileSync(path, "utf8");
|
|
}
|
|
|
|
function functionBody(source: string, name: string): string {
|
|
const signature = `function ${name}(`;
|
|
const asyncSignature = `async ${signature}`;
|
|
const signatureIndex = source.indexOf(asyncSignature) >= 0 ? source.indexOf(asyncSignature) : source.indexOf(signature);
|
|
assert.notEqual(signatureIndex, -1, `Could not find function ${name}`);
|
|
const bodyStart = source.indexOf("{", signatureIndex);
|
|
assert.notEqual(bodyStart, -1, `Could not find body for ${name}`);
|
|
|
|
let depth = 0;
|
|
for (let index = bodyStart; index < source.length; index += 1) {
|
|
const char = source[index];
|
|
if (char === "{") depth += 1;
|
|
if (char === "}") {
|
|
depth -= 1;
|
|
if (depth === 0) return source.slice(bodyStart + 1, index);
|
|
}
|
|
}
|
|
throw new Error(`Could not parse body for ${name}`);
|
|
}
|
|
|
|
test("secondary write entrypoints use the shared production SQL guard", () => {
|
|
const entrypoints = [
|
|
{
|
|
path: "apps/desktop/src/components/diff/SchemaDiffDialog.vue",
|
|
executor: "api.executeScript",
|
|
sourceKey: "production.sourceSchemaDiff",
|
|
},
|
|
{
|
|
path: "apps/desktop/src/components/diff/DataCompareDialog.vue",
|
|
executor: "api.executeBatch",
|
|
sourceKey: "production.sourceDataCompare",
|
|
},
|
|
{
|
|
path: "apps/desktop/src/components/objects/InstallExtensionDialog.vue",
|
|
executor: "api.executeQuery",
|
|
sourceKey: "production.sourceExtension",
|
|
},
|
|
{
|
|
path: "apps/desktop/src/components/sidebar/TreeItem.vue",
|
|
executor: "api.executeQuery",
|
|
sourceKey: "production.sourceSidebar",
|
|
},
|
|
{
|
|
path: "apps/desktop/src/App.vue",
|
|
executor: "executeObjectSourceSave",
|
|
sourceKey: "production.sourceObjectSource",
|
|
},
|
|
{
|
|
path: "apps/desktop/src/components/objects/ObjectSourceDialog.vue",
|
|
executor: "executeObjectSourceSave",
|
|
sourceKey: "production.sourceObjectSource",
|
|
},
|
|
{
|
|
path: "apps/desktop/src/components/objects/ObjectBrowser.vue",
|
|
executor: "executeObjectSourceSave",
|
|
sourceKey: "production.sourceObjectSource",
|
|
},
|
|
{
|
|
path: "apps/desktop/src/components/objects/ObjectBrowser.vue",
|
|
executor: "api.executeQuery",
|
|
sourceKey: "production.sourceObjectBrowser",
|
|
},
|
|
{
|
|
path: "apps/desktop/src/components/generate/DataGenerateDialog.vue",
|
|
executor: "api.executeQuery",
|
|
sourceKey: "production.sourceDataGenerate",
|
|
},
|
|
{
|
|
path: "apps/desktop/src/components/editor/QueryHistory.vue",
|
|
executor: "api.executeScript",
|
|
sourceKey: "production.sourceQueryHistory",
|
|
},
|
|
{
|
|
path: "apps/desktop/src/components/admin/DatabaseUserAdmin.vue",
|
|
executor: "api.executeMulti",
|
|
sourceKey: "production.sourceAdmin",
|
|
},
|
|
{
|
|
path: "apps/desktop/src/components/admin/DamengJobAdmin.vue",
|
|
executor: "api.executeMulti",
|
|
sourceKey: "production.sourceAdmin",
|
|
},
|
|
];
|
|
|
|
for (const entrypoint of entrypoints) {
|
|
const source = readSource(entrypoint.path);
|
|
assert.match(source, /executeWithProductionSqlGuard/, entrypoint.path);
|
|
assert.ok(source.includes(entrypoint.executor), `${entrypoint.path} should still execute SQL through its original backend API`);
|
|
assert.ok(source.includes(entrypoint.sourceKey), `${entrypoint.path} should label the confirmation source`);
|
|
}
|
|
});
|
|
|
|
test("object browser batch empty reviews the frozen SQL plan before executing", () => {
|
|
const source = readSource("apps/desktop/src/components/objects/ObjectBrowser.vue");
|
|
const body = functionBody(source, "confirmBatchEmptyTables");
|
|
assert.match(body, /executeObjectBrowserSqlWithProductionGuard\(\s*reviewSql/, "batch empty must use the Object Browser production guard");
|
|
assert.match(body, /runBatchTableEmpty/, "batch empty should still use the batch executor after confirmation");
|
|
assert.ok(body.indexOf("executeObjectBrowserSqlWithProductionGuard") < body.indexOf("runBatchTableEmpty"), "production guard must be entered before the batch executor");
|
|
});
|