@tickernelz/wardmote-db 2026.1112.1 → 2026.1114.0
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- package/dist/backup-lib.d.ts.map +1 -1
- package/dist/backup-lib.js +10 -7
- package/dist/backup-lib.js.map +1 -1
- package/dist/backup-lib.test.js +43 -32
- package/dist/backup-lib.test.js.map +1 -1
- package/dist/client.d.ts.map +1 -1
- package/dist/client.js +14 -4
- package/dist/client.js.map +1 -1
- package/dist/client.test.js +1 -1
- package/dist/heartbeat-run-issue-privacy-migration.test.js +1 -1
- package/dist/heartbeat-run-issue-privacy-migration.test.js.map +1 -1
- package/dist/migration-lock.test.d.ts +2 -0
- package/dist/migration-lock.test.d.ts.map +1 -0
- package/dist/migration-lock.test.js +37 -0
- package/dist/migration-lock.test.js.map +1 -0
- package/dist/migrations/0374_wardmote_runner_question_key.sql +3 -0
- package/dist/migrations/0375_wardmote_railway_tool_names.sql +8 -0
- package/dist/migrations/0376_wardmote_ai_model_prefix.sql +11 -0
- package/dist/migrations/0377_wardmote_managed_archive_token.sql +8 -0
- package/dist/migrations/0378_wardmote_tool_template_keys.sql +82 -0
- package/dist/migrations/0379_wardmote_tool_selector_names.sql +84 -0
- package/dist/migrations/0380_wardmote_self_tool_names.sql +16 -0
- package/dist/migrations/0381_wardmote_plugin_skill_keys.sql +73 -0
- package/dist/migrations/0382_wardmote_work_product_provider.sql +6 -0
- package/dist/migrations/0383_wardmote_feedback_versions.sql +3 -0
- package/dist/migrations/0384_wardmote_stored_keys.sql +21 -0
- package/dist/migrations/0385_wardmote_package_metadata.sql +12 -0
- package/dist/migrations/0386_wardmote_acpx_permission_mode.sql +9 -0
- package/dist/migrations/0387_wardmote_omp_session_dir.sql +7 -0
- package/dist/migrations/0388_wardmote_identity_link_user_column.sql +8 -0
- package/dist/migrations/meta/_journal.json +105 -0
- package/dist/nested-skill-folders-migration.test.js +1 -1
- package/dist/nested-skill-folders-migration.test.js.map +1 -1
- package/dist/retired-backup-names.d.ts +3 -0
- package/dist/retired-backup-names.d.ts.map +1 -0
- package/dist/retired-backup-names.js +3 -0
- package/dist/retired-backup-names.js.map +1 -0
- package/dist/retired-migration-lock.d.ts +2 -0
- package/dist/retired-migration-lock.d.ts.map +1 -0
- package/dist/retired-migration-lock.js +2 -0
- package/dist/retired-migration-lock.js.map +1 -0
- package/dist/runtime-config.d.ts +1 -1
- package/dist/runtime-config.d.ts.map +1 -1
- package/dist/runtime-config.js +1 -1
- package/dist/runtime-config.js.map +1 -1
- package/dist/runtime-config.test.js +5 -5
- package/dist/runtime-config.test.js.map +1 -1
- package/dist/schema/agent_api_keys.d.ts +4 -4
- package/dist/schema/chat_channels.d.ts +1 -1
- package/dist/schema/chat_channels.js +1 -1
- package/dist/schema/chat_channels.js.map +1 -1
- package/dist/schema/feedback_exports.js +3 -3
- package/dist/schema/feedback_exports.js.map +1 -1
- package/dist/wardmote-acpx-permission-mode-migration.test.d.ts +2 -0
- package/dist/wardmote-acpx-permission-mode-migration.test.d.ts.map +1 -0
- package/dist/wardmote-acpx-permission-mode-migration.test.js +60 -0
- package/dist/wardmote-acpx-permission-mode-migration.test.js.map +1 -0
- package/dist/wardmote-ai-model-prefix-migration.test.d.ts +2 -0
- package/dist/wardmote-ai-model-prefix-migration.test.d.ts.map +1 -0
- package/dist/wardmote-ai-model-prefix-migration.test.js +58 -0
- package/dist/wardmote-ai-model-prefix-migration.test.js.map +1 -0
- package/dist/wardmote-feedback-versions-migration.test.d.ts +2 -0
- package/dist/wardmote-feedback-versions-migration.test.d.ts.map +1 -0
- package/dist/wardmote-feedback-versions-migration.test.js +43 -0
- package/dist/wardmote-feedback-versions-migration.test.js.map +1 -0
- package/dist/wardmote-identity-link-user-column-migration.test.d.ts +2 -0
- package/dist/wardmote-identity-link-user-column-migration.test.d.ts.map +1 -0
- package/dist/wardmote-identity-link-user-column-migration.test.js +40 -0
- package/dist/wardmote-identity-link-user-column-migration.test.js.map +1 -0
- package/dist/wardmote-managed-archive-token-migration.test.d.ts +2 -0
- package/dist/wardmote-managed-archive-token-migration.test.d.ts.map +1 -0
- package/dist/wardmote-managed-archive-token-migration.test.js +64 -0
- package/dist/wardmote-managed-archive-token-migration.test.js.map +1 -0
- package/dist/wardmote-omp-session-dir-migration.test.d.ts +2 -0
- package/dist/wardmote-omp-session-dir-migration.test.d.ts.map +1 -0
- package/dist/wardmote-omp-session-dir-migration.test.js +49 -0
- package/dist/wardmote-omp-session-dir-migration.test.js.map +1 -0
- package/dist/wardmote-package-metadata-migration.test.d.ts +2 -0
- package/dist/wardmote-package-metadata-migration.test.d.ts.map +1 -0
- package/dist/wardmote-package-metadata-migration.test.js +64 -0
- package/dist/wardmote-package-metadata-migration.test.js.map +1 -0
- package/dist/wardmote-plugin-skill-keys-migration.test.d.ts +2 -0
- package/dist/wardmote-plugin-skill-keys-migration.test.d.ts.map +1 -0
- package/dist/wardmote-plugin-skill-keys-migration.test.js +88 -0
- package/dist/wardmote-plugin-skill-keys-migration.test.js.map +1 -0
- package/dist/wardmote-railway-tool-names-migration.test.d.ts +2 -0
- package/dist/wardmote-railway-tool-names-migration.test.d.ts.map +1 -0
- package/dist/wardmote-railway-tool-names-migration.test.js +63 -0
- package/dist/wardmote-railway-tool-names-migration.test.js.map +1 -0
- package/dist/wardmote-runner-question-key-migration.test.d.ts +2 -0
- package/dist/wardmote-runner-question-key-migration.test.d.ts.map +1 -0
- package/dist/wardmote-runner-question-key-migration.test.js +54 -0
- package/dist/wardmote-runner-question-key-migration.test.js.map +1 -0
- package/dist/wardmote-self-tool-names-migration.test.d.ts +2 -0
- package/dist/wardmote-self-tool-names-migration.test.d.ts.map +1 -0
- package/dist/wardmote-self-tool-names-migration.test.js +67 -0
- package/dist/wardmote-self-tool-names-migration.test.js.map +1 -0
- package/dist/wardmote-stored-keys-migration.test.d.ts +2 -0
- package/dist/wardmote-stored-keys-migration.test.d.ts.map +1 -0
- package/dist/wardmote-stored-keys-migration.test.js +69 -0
- package/dist/wardmote-stored-keys-migration.test.js.map +1 -0
- package/dist/wardmote-tool-selector-names-migration.test.d.ts +2 -0
- package/dist/wardmote-tool-selector-names-migration.test.d.ts.map +1 -0
- package/dist/wardmote-tool-selector-names-migration.test.js +91 -0
- package/dist/wardmote-tool-selector-names-migration.test.js.map +1 -0
- package/dist/wardmote-tool-template-keys-migration.test.d.ts +2 -0
- package/dist/wardmote-tool-template-keys-migration.test.d.ts.map +1 -0
- package/dist/wardmote-tool-template-keys-migration.test.js +117 -0
- package/dist/wardmote-tool-template-keys-migration.test.js.map +1 -0
- package/dist/wardmote-work-product-provider-migration.test.d.ts +2 -0
- package/dist/wardmote-work-product-provider-migration.test.d.ts.map +1 -0
- package/dist/wardmote-work-product-provider-migration.test.js +65 -0
- package/dist/wardmote-work-product-provider-migration.test.js.map +1 -0
- package/package.json +2 -2
|
@@ -0,0 +1,69 @@
|
|
|
1
|
+
import { createHash, randomUUID } from "node:crypto";
|
|
2
|
+
import fs from "node:fs";
|
|
3
|
+
import { afterEach, describe, expect, it } from "vitest";
|
|
4
|
+
import postgres from "postgres";
|
|
5
|
+
import { applyPendingMigrations } from "./client.js";
|
|
6
|
+
import { EMBEDDED_POSTGRES_TEST_TIMEOUT_MS, getEmbeddedPostgresTestSupport, startEmbeddedPostgresTestDatabase, } from "./test-embedded-postgres.js";
|
|
7
|
+
const MIGRATION_FILE = "0384_wardmote_stored_keys.sql";
|
|
8
|
+
const RETIRED_CREATE_TASK_PREFIX = "paperclip-runner:create-task:";
|
|
9
|
+
const RETIRED_SMOKE_LAB_PREFIX = "paperclip.smoke-lab.";
|
|
10
|
+
const cleanups = [];
|
|
11
|
+
const embeddedPostgresSupport = await getEmbeddedPostgresTestSupport();
|
|
12
|
+
const describeEmbeddedPostgres = embeddedPostgresSupport.supported ? describe : describe.skip;
|
|
13
|
+
async function migrationHash() {
|
|
14
|
+
const content = await fs.promises.readFile(new URL(`./migrations/${MIGRATION_FILE}`, import.meta.url), "utf8");
|
|
15
|
+
return createHash("sha256").update(content).digest("hex");
|
|
16
|
+
}
|
|
17
|
+
describeEmbeddedPostgres("wardmote stored keys migration", () => {
|
|
18
|
+
afterEach(async () => {
|
|
19
|
+
await Promise.all(cleanups.splice(0).map((cleanup) => cleanup()));
|
|
20
|
+
});
|
|
21
|
+
it("moves runner create-task origins and smoke-lab keys to wardmote and leaves other values", async () => {
|
|
22
|
+
const database = await startEmbeddedPostgresTestDatabase("wardmote-stored-keys-migration-");
|
|
23
|
+
cleanups.push(database.cleanup);
|
|
24
|
+
const sql = postgres(database.connectionString, { max: 1, onnotice: () => { } });
|
|
25
|
+
cleanups.push(async () => sql.end());
|
|
26
|
+
const companyId = randomUUID();
|
|
27
|
+
const runnerChild = randomUUID();
|
|
28
|
+
const manualIssue = randomUUID();
|
|
29
|
+
const smokeApp = randomUUID();
|
|
30
|
+
const otherApp = randomUUID();
|
|
31
|
+
const smokeProfile = randomUUID();
|
|
32
|
+
await sql `SET session_replication_role = replica`;
|
|
33
|
+
await sql `
|
|
34
|
+
INSERT INTO "issues" ("id", "company_id", "title", "origin_id")
|
|
35
|
+
VALUES
|
|
36
|
+
(${runnerChild}, ${companyId}, 'child', ${`${RETIRED_CREATE_TASK_PREFIX}issue-1:key-1`}),
|
|
37
|
+
(${manualIssue}, ${companyId}, 'manual', 'github:acme/repo:issue:7')
|
|
38
|
+
`;
|
|
39
|
+
await sql `
|
|
40
|
+
INSERT INTO "tool_applications" ("id", "company_id", "name", "type", "application_key")
|
|
41
|
+
VALUES
|
|
42
|
+
(${smokeApp}, ${companyId}, 'smoke', 'mcp', ${`${RETIRED_SMOKE_LAB_PREFIX}http-fixture`}),
|
|
43
|
+
(${otherApp}, ${companyId}, 'other', 'mcp', 'acme.tools')
|
|
44
|
+
`;
|
|
45
|
+
await sql `
|
|
46
|
+
INSERT INTO "tool_profiles" ("id", "company_id", "profile_key", "name")
|
|
47
|
+
VALUES (${smokeProfile}, ${companyId}, ${`${RETIRED_SMOKE_LAB_PREFIX}profile`}, 'smoke')
|
|
48
|
+
`;
|
|
49
|
+
await sql `SET session_replication_role = DEFAULT`;
|
|
50
|
+
const hash = await migrationHash();
|
|
51
|
+
await sql `DELETE FROM "drizzle"."__drizzle_migrations" WHERE "hash" = ${hash}`;
|
|
52
|
+
await applyPendingMigrations(database.connectionString);
|
|
53
|
+
const snapshot = async () => ({
|
|
54
|
+
issues: Object.fromEntries((await sql `SELECT "id", "origin_id" FROM "issues" WHERE "company_id" = ${companyId}`).map((row) => [row.id, row.origin_id])),
|
|
55
|
+
apps: Object.fromEntries((await sql `SELECT "id", "application_key" FROM "tool_applications" WHERE "company_id" = ${companyId}`).map((row) => [row.id, row.application_key])),
|
|
56
|
+
profiles: Object.fromEntries((await sql `SELECT "id", "profile_key" FROM "tool_profiles" WHERE "company_id" = ${companyId}`).map((row) => [row.id, row.profile_key])),
|
|
57
|
+
});
|
|
58
|
+
const after = await snapshot();
|
|
59
|
+
expect(after.issues[runnerChild]).toBe("wardmote-runner:create-task:issue-1:key-1");
|
|
60
|
+
expect(after.issues[manualIssue]).toBe("github:acme/repo:issue:7");
|
|
61
|
+
expect(after.apps[smokeApp]).toBe("wardmote.smoke-lab.http-fixture");
|
|
62
|
+
expect(after.apps[otherApp]).toBe("acme.tools");
|
|
63
|
+
expect(after.profiles[smokeProfile]).toBe("wardmote.smoke-lab.profile");
|
|
64
|
+
await sql `DELETE FROM "drizzle"."__drizzle_migrations" WHERE "hash" = ${hash}`;
|
|
65
|
+
await applyPendingMigrations(database.connectionString);
|
|
66
|
+
expect(await snapshot()).toEqual(after);
|
|
67
|
+
}, EMBEDDED_POSTGRES_TEST_TIMEOUT_MS);
|
|
68
|
+
});
|
|
69
|
+
//# sourceMappingURL=wardmote-stored-keys-migration.test.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"wardmote-stored-keys-migration.test.js","sourceRoot":"","sources":["../src/wardmote-stored-keys-migration.test.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,UAAU,EAAE,UAAU,EAAE,MAAM,aAAa,CAAC;AACrD,OAAO,EAAE,MAAM,SAAS,CAAC;AACzB,OAAO,EAAE,SAAS,EAAE,QAAQ,EAAE,MAAM,EAAE,EAAE,EAAE,MAAM,QAAQ,CAAC;AACzD,OAAO,QAAQ,MAAM,UAAU,CAAC;AAChC,OAAO,EAAE,sBAAsB,EAAE,MAAM,aAAa,CAAC;AACrD,OAAO,EACL,iCAAiC,EACjC,8BAA8B,EAC9B,iCAAiC,GAClC,MAAM,6BAA6B,CAAC;AAErC,MAAM,cAAc,GAAG,+BAA+B,CAAC;AACvD,MAAM,0BAA0B,GAAG,+BAA+B,CAAC;AACnE,MAAM,wBAAwB,GAAG,sBAAsB,CAAC;AACxD,MAAM,QAAQ,GAA+B,EAAE,CAAC;AAChD,MAAM,uBAAuB,GAAG,MAAM,8BAA8B,EAAE,CAAC;AACvE,MAAM,wBAAwB,GAAG,uBAAuB,CAAC,SAAS,CAAC,CAAC,CAAC,QAAQ,CAAC,CAAC,CAAC,QAAQ,CAAC,IAAI,CAAC;AAE9F,KAAK,UAAU,aAAa;IAC1B,MAAM,OAAO,GAAG,MAAM,EAAE,CAAC,QAAQ,CAAC,QAAQ,CAAC,IAAI,GAAG,CAAC,gBAAgB,cAAc,EAAE,EAAE,OAAO,IAAI,CAAC,GAAG,CAAC,EAAE,MAAM,CAAC,CAAC;IAC/G,OAAO,UAAU,CAAC,QAAQ,CAAC,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;AAC5D,CAAC;AAED,wBAAwB,CAAC,gCAAgC,EAAE,GAAG,EAAE;IAC9D,SAAS,CAAC,KAAK,IAAI,EAAE;QACnB,MAAM,OAAO,CAAC,GAAG,CAAC,QAAQ,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,OAAO,EAAE,EAAE,CAAC,OAAO,EAAE,CAAC,CAAC,CAAC;IACpE,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,yFAAyF,EAAE,KAAK,IAAI,EAAE;QACvG,MAAM,QAAQ,GAAG,MAAM,iCAAiC,CAAC,iCAAiC,CAAC,CAAC;QAC5F,QAAQ,CAAC,IAAI,CAAC,QAAQ,CAAC,OAAO,CAAC,CAAC;QAChC,MAAM,GAAG,GAAG,QAAQ,CAAC,QAAQ,CAAC,gBAAgB,EAAE,EAAE,GAAG,EAAE,CAAC,EAAE,QAAQ,EAAE,GAAG,EAAE,GAAE,CAAC,EAAE,CAAC,CAAC;QAChF,QAAQ,CAAC,IAAI,CAAC,KAAK,IAAI,EAAE,CAAC,GAAG,CAAC,GAAG,EAAE,CAAC,CAAC;QAErC,MAAM,SAAS,GAAG,UAAU,EAAE,CAAC;QAC/B,MAAM,WAAW,GAAG,UAAU,EAAE,CAAC;QACjC,MAAM,WAAW,GAAG,UAAU,EAAE,CAAC;QACjC,MAAM,QAAQ,GAAG,UAAU,EAAE,CAAC;QAC9B,MAAM,QAAQ,GAAG,UAAU,EAAE,CAAC;QAC9B,MAAM,YAAY,GAAG,UAAU,EAAE,CAAC;QAClC,MAAM,GAAG,CAAA,wCAAwC,CAAC;QAClD,MAAM,GAAG,CAAA;;;WAGF,WAAW,KAAK,SAAS,cAAc,GAAG,0BAA0B,eAAe;WACnF,WAAW,KAAK,SAAS;KAC/B,CAAC;QACF,MAAM,GAAG,CAAA;;;WAGF,QAAQ,KAAK,SAAS,qBAAqB,GAAG,wBAAwB,cAAc;WACpF,QAAQ,KAAK,SAAS;KAC5B,CAAC;QACF,MAAM,GAAG,CAAA;;gBAEG,YAAY,KAAK,SAAS,KAAK,GAAG,wBAAwB,SAAS;KAC9E,CAAC;QACF,MAAM,GAAG,CAAA,wCAAwC,CAAC;QAElD,MAAM,IAAI,GAAG,MAAM,aAAa,EAAE,CAAC;QACnC,MAAM,GAAG,CAAA,+DAA+D,IAAI,EAAE,CAAC;QAC/E,MAAM,sBAAsB,CAAC,QAAQ,CAAC,gBAAgB,CAAC,CAAC;QAExD,MAAM,QAAQ,GAAG,KAAK,IAAI,EAAE,CAAC,CAAC;YAC5B,MAAM,EAAE,MAAM,CAAC,WAAW,CAAC,CAAC,MAAM,GAAG,CAAA,+DAA+D,SAAS,EAAE,CAAC,CAAC,GAAG,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,CAAC,GAAG,CAAC,EAAE,EAAE,GAAG,CAAC,SAAS,CAAC,CAAC,CAAC;YACvJ,IAAI,EAAE,MAAM,CAAC,WAAW,CAAC,CAAC,MAAM,GAAG,CAAA,gFAAgF,SAAS,EAAE,CAAC,CAAC,GAAG,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,CAAC,GAAG,CAAC,EAAE,EAAE,GAAG,CAAC,eAAe,CAAC,CAAC,CAAC;YAC5K,QAAQ,EAAE,MAAM,CAAC,WAAW,CAAC,CAAC,MAAM,GAAG,CAAA,wEAAwE,SAAS,EAAE,CAAC,CAAC,GAAG,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,CAAC,GAAG,CAAC,EAAE,EAAE,GAAG,CAAC,WAAW,CAAC,CAAC,CAAC;SACrK,CAAC,CAAC;QACH,MAAM,KAAK,GAAG,MAAM,QAAQ,EAAE,CAAC;QAC/B,MAAM,CAAC,KAAK,CAAC,MAAM,CAAC,WAAW,CAAC,CAAC,CAAC,IAAI,CAAC,2CAA2C,CAAC,CAAC;QACpF,MAAM,CAAC,KAAK,CAAC,MAAM,CAAC,WAAW,CAAC,CAAC,CAAC,IAAI,CAAC,0BAA0B,CAAC,CAAC;QACnE,MAAM,CAAC,KAAK,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC,CAAC,IAAI,CAAC,iCAAiC,CAAC,CAAC;QACrE,MAAM,CAAC,KAAK,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC;QAChD,MAAM,CAAC,KAAK,CAAC,QAAQ,CAAC,YAAY,CAAC,CAAC,CAAC,IAAI,CAAC,4BAA4B,CAAC,CAAC;QAExE,MAAM,GAAG,CAAA,+DAA+D,IAAI,EAAE,CAAC;QAC/E,MAAM,sBAAsB,CAAC,QAAQ,CAAC,gBAAgB,CAAC,CAAC;QACxD,MAAM,CAAC,MAAM,QAAQ,EAAE,CAAC,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC;IAC1C,CAAC,EAAE,iCAAiC,CAAC,CAAC;AACxC,CAAC,CAAC,CAAC"}
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"wardmote-tool-selector-names-migration.test.d.ts","sourceRoot":"","sources":["../src/wardmote-tool-selector-names-migration.test.ts"],"names":[],"mappings":""}
|
|
@@ -0,0 +1,91 @@
|
|
|
1
|
+
import { createHash, randomUUID } from "node:crypto";
|
|
2
|
+
import fs from "node:fs";
|
|
3
|
+
import { afterEach, describe, expect, it } from "vitest";
|
|
4
|
+
import postgres from "postgres";
|
|
5
|
+
import { applyPendingMigrations } from "./client.js";
|
|
6
|
+
import { EMBEDDED_POSTGRES_TEST_TIMEOUT_MS, getEmbeddedPostgresTestSupport, startEmbeddedPostgresTestDatabase, } from "./test-embedded-postgres.js";
|
|
7
|
+
const MIGRATION_FILE = "0379_wardmote_tool_selector_names.sql";
|
|
8
|
+
const RETIRED = "paperclip";
|
|
9
|
+
const cleanups = [];
|
|
10
|
+
const embeddedPostgresSupport = await getEmbeddedPostgresTestSupport();
|
|
11
|
+
const describeEmbeddedPostgres = embeddedPostgresSupport.supported ? describe : describe.skip;
|
|
12
|
+
async function migrationHash() {
|
|
13
|
+
const content = await fs.promises.readFile(new URL(`./migrations/${MIGRATION_FILE}`, import.meta.url), "utf8");
|
|
14
|
+
return createHash("sha256").update(content).digest("hex");
|
|
15
|
+
}
|
|
16
|
+
describeEmbeddedPostgres("wardmote tool selector names migration", () => {
|
|
17
|
+
afterEach(async () => {
|
|
18
|
+
await Promise.all(cleanups.splice(0).map((cleanup) => cleanup()));
|
|
19
|
+
});
|
|
20
|
+
it("keeps policies, profile conditions and tool grants pointing at renamed tools and applications", async () => {
|
|
21
|
+
const database = await startEmbeddedPostgresTestDatabase("wardmote-tool-selector-names-migration-");
|
|
22
|
+
cleanups.push(database.cleanup);
|
|
23
|
+
const sql = postgres(database.connectionString, { max: 1, onnotice: () => { } });
|
|
24
|
+
cleanups.push(async () => sql.end());
|
|
25
|
+
const companyId = randomUUID();
|
|
26
|
+
const keptPagesCompanyId = randomUUID();
|
|
27
|
+
const denyRailway = randomUUID();
|
|
28
|
+
const pagesPolicy = randomUUID();
|
|
29
|
+
const keptPagesPolicy = randomUUID();
|
|
30
|
+
const unrelatedPolicy = randomUUID();
|
|
31
|
+
const profile = randomUUID();
|
|
32
|
+
const entry = randomUUID();
|
|
33
|
+
const toolGrant = randomUUID();
|
|
34
|
+
const otherGrant = randomUUID();
|
|
35
|
+
await sql `
|
|
36
|
+
INSERT INTO "companies" ("id", "name", "issue_prefix")
|
|
37
|
+
VALUES (${companyId}, 'Selectors', 'TSA'), (${keptPagesCompanyId}, 'Kept pages', 'TSB')
|
|
38
|
+
`;
|
|
39
|
+
await sql `SET session_replication_role = replica`;
|
|
40
|
+
await sql `
|
|
41
|
+
INSERT INTO "tool_applications" ("id", "company_id", "name", "type", "application_key")
|
|
42
|
+
VALUES (${randomUUID()}, ${keptPagesCompanyId}, 'Pages', 'mcp', ${`pages.${RETIRED}`})
|
|
43
|
+
`;
|
|
44
|
+
await sql `
|
|
45
|
+
INSERT INTO "tool_policies" ("id", "company_id", "name", "policy_type", "selectors", "conditions")
|
|
46
|
+
VALUES
|
|
47
|
+
(${denyRailway}, ${companyId}, 'deny railway', 'deny', ${sql.json({ toolNames: [`${RETIRED}-railway-restart`, "list-services"] })}, ${sql.json({ context: { toolName: `${RETIRED}-railway-read-logs` } })}),
|
|
48
|
+
(${pagesPolicy}, ${companyId}, 'pages', 'deny', ${sql.json({ applicationKeys: [`${RETIRED}.pages`, `${RETIRED}.examples.safe-read-only-todo-kv`] })}, NULL),
|
|
49
|
+
(${keptPagesPolicy}, ${keptPagesCompanyId}, 'kept pages', 'deny', ${sql.json({ applicationKey: `pages.${RETIRED}` })}, NULL),
|
|
50
|
+
(${unrelatedPolicy}, ${companyId}, 'unrelated', 'deny', ${sql.json({ toolName: "railway-restart" })}, NULL)
|
|
51
|
+
`;
|
|
52
|
+
await sql `
|
|
53
|
+
INSERT INTO "tool_profiles" ("id", "company_id", "profile_key", "name")
|
|
54
|
+
VALUES (${profile}, ${companyId}, 'profile', 'Profile')
|
|
55
|
+
`;
|
|
56
|
+
await sql `
|
|
57
|
+
INSERT INTO "tool_profile_entries" ("id", "company_id", "profile_id", "selector_type", "conditions")
|
|
58
|
+
VALUES (${entry}, ${companyId}, ${profile}, 'tool_name', ${sql.json({ actor: { toolName: `${RETIRED}-railway-restart` } })})
|
|
59
|
+
`;
|
|
60
|
+
await sql `
|
|
61
|
+
INSERT INTO "principal_permission_grants" ("id", "company_id", "principal_type", "principal_id", "permission_key", "scope")
|
|
62
|
+
VALUES
|
|
63
|
+
(${toolGrant}, ${companyId}, 'agent', 'agent-1', 'tools:use', ${sql.json({ allow: [`tool:${RETIRED}-railway-restart`], applicationKey: `${RETIRED}.pages` })}),
|
|
64
|
+
(${otherGrant}, ${companyId}, 'agent', 'agent-1', 'issues:read', ${sql.json({ note: `${RETIRED}-railway-restart` })})
|
|
65
|
+
`;
|
|
66
|
+
await sql `SET session_replication_role = DEFAULT`;
|
|
67
|
+
const hash = await migrationHash();
|
|
68
|
+
await sql `DELETE FROM "drizzle"."__drizzle_migrations" WHERE "hash" = ${hash}`;
|
|
69
|
+
await applyPendingMigrations(database.connectionString);
|
|
70
|
+
const snapshot = async () => ({
|
|
71
|
+
policies: new Map((await sql `SELECT "id", "selectors", "conditions" FROM "tool_policies" WHERE "company_id" IN ${sql([companyId, keptPagesCompanyId])}`).map((row) => [row.id, [row.selectors, row.conditions]])),
|
|
72
|
+
entry: (await sql `SELECT "conditions" FROM "tool_profile_entries" WHERE "id" = ${entry}`)[0]?.conditions,
|
|
73
|
+
grants: new Map((await sql `SELECT "id", "scope" FROM "principal_permission_grants" WHERE "company_id" = ${companyId}`).map((row) => [row.id, row.scope])),
|
|
74
|
+
});
|
|
75
|
+
const after = await snapshot();
|
|
76
|
+
expect(after.policies.get(denyRailway)).toEqual([
|
|
77
|
+
{ toolNames: ["wardmote-railway-restart", "list-services"] },
|
|
78
|
+
{ context: { toolName: "wardmote-railway-read-logs" } },
|
|
79
|
+
]);
|
|
80
|
+
expect(after.policies.get(pagesPolicy)).toEqual([{ applicationKeys: ["wardmote-pages", "wardmote.examples.safe-read-only-todo-kv"] }, null]);
|
|
81
|
+
expect(after.policies.get(keptPagesPolicy)).toEqual([{ applicationKey: `pages.${RETIRED}` }, null]);
|
|
82
|
+
expect(after.policies.get(unrelatedPolicy)).toEqual([{ toolName: "railway-restart" }, null]);
|
|
83
|
+
expect(after.entry).toEqual({ actor: { toolName: "wardmote-railway-restart" } });
|
|
84
|
+
expect(after.grants.get(toolGrant)).toEqual({ allow: ["tool:wardmote-railway-restart"], applicationKey: "wardmote-pages" });
|
|
85
|
+
expect(after.grants.get(otherGrant)).toEqual({ note: `${RETIRED}-railway-restart` });
|
|
86
|
+
await sql `DELETE FROM "drizzle"."__drizzle_migrations" WHERE "hash" = ${hash}`;
|
|
87
|
+
await applyPendingMigrations(database.connectionString);
|
|
88
|
+
expect(await snapshot()).toEqual(after);
|
|
89
|
+
}, EMBEDDED_POSTGRES_TEST_TIMEOUT_MS);
|
|
90
|
+
});
|
|
91
|
+
//# sourceMappingURL=wardmote-tool-selector-names-migration.test.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"wardmote-tool-selector-names-migration.test.js","sourceRoot":"","sources":["../src/wardmote-tool-selector-names-migration.test.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,UAAU,EAAE,UAAU,EAAE,MAAM,aAAa,CAAC;AACrD,OAAO,EAAE,MAAM,SAAS,CAAC;AACzB,OAAO,EAAE,SAAS,EAAE,QAAQ,EAAE,MAAM,EAAE,EAAE,EAAE,MAAM,QAAQ,CAAC;AACzD,OAAO,QAAQ,MAAM,UAAU,CAAC;AAChC,OAAO,EAAE,sBAAsB,EAAE,MAAM,aAAa,CAAC;AACrD,OAAO,EACL,iCAAiC,EACjC,8BAA8B,EAC9B,iCAAiC,GAClC,MAAM,6BAA6B,CAAC;AAErC,MAAM,cAAc,GAAG,uCAAuC,CAAC;AAC/D,MAAM,OAAO,GAAG,WAAW,CAAC;AAC5B,MAAM,QAAQ,GAA+B,EAAE,CAAC;AAChD,MAAM,uBAAuB,GAAG,MAAM,8BAA8B,EAAE,CAAC;AACvE,MAAM,wBAAwB,GAAG,uBAAuB,CAAC,SAAS,CAAC,CAAC,CAAC,QAAQ,CAAC,CAAC,CAAC,QAAQ,CAAC,IAAI,CAAC;AAE9F,KAAK,UAAU,aAAa;IAC1B,MAAM,OAAO,GAAG,MAAM,EAAE,CAAC,QAAQ,CAAC,QAAQ,CAAC,IAAI,GAAG,CAAC,gBAAgB,cAAc,EAAE,EAAE,OAAO,IAAI,CAAC,GAAG,CAAC,EAAE,MAAM,CAAC,CAAC;IAC/G,OAAO,UAAU,CAAC,QAAQ,CAAC,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;AAC5D,CAAC;AAED,wBAAwB,CAAC,wCAAwC,EAAE,GAAG,EAAE;IACtE,SAAS,CAAC,KAAK,IAAI,EAAE;QACnB,MAAM,OAAO,CAAC,GAAG,CAAC,QAAQ,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,OAAO,EAAE,EAAE,CAAC,OAAO,EAAE,CAAC,CAAC,CAAC;IACpE,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,+FAA+F,EAAE,KAAK,IAAI,EAAE;QAC7G,MAAM,QAAQ,GAAG,MAAM,iCAAiC,CAAC,yCAAyC,CAAC,CAAC;QACpG,QAAQ,CAAC,IAAI,CAAC,QAAQ,CAAC,OAAO,CAAC,CAAC;QAChC,MAAM,GAAG,GAAG,QAAQ,CAAC,QAAQ,CAAC,gBAAgB,EAAE,EAAE,GAAG,EAAE,CAAC,EAAE,QAAQ,EAAE,GAAG,EAAE,GAAE,CAAC,EAAE,CAAC,CAAC;QAChF,QAAQ,CAAC,IAAI,CAAC,KAAK,IAAI,EAAE,CAAC,GAAG,CAAC,GAAG,EAAE,CAAC,CAAC;QAErC,MAAM,SAAS,GAAG,UAAU,EAAE,CAAC;QAC/B,MAAM,kBAAkB,GAAG,UAAU,EAAE,CAAC;QACxC,MAAM,WAAW,GAAG,UAAU,EAAE,CAAC;QACjC,MAAM,WAAW,GAAG,UAAU,EAAE,CAAC;QACjC,MAAM,eAAe,GAAG,UAAU,EAAE,CAAC;QACrC,MAAM,eAAe,GAAG,UAAU,EAAE,CAAC;QACrC,MAAM,OAAO,GAAG,UAAU,EAAE,CAAC;QAC7B,MAAM,KAAK,GAAG,UAAU,EAAE,CAAC;QAC3B,MAAM,SAAS,GAAG,UAAU,EAAE,CAAC;QAC/B,MAAM,UAAU,GAAG,UAAU,EAAE,CAAC;QAChC,MAAM,GAAG,CAAA;;gBAEG,SAAS,2BAA2B,kBAAkB;KACjE,CAAC;QACF,MAAM,GAAG,CAAA,wCAAwC,CAAC;QAClD,MAAM,GAAG,CAAA;;gBAEG,UAAU,EAAE,KAAK,kBAAkB,qBAAqB,SAAS,OAAO,EAAE;KACrF,CAAC;QACF,MAAM,GAAG,CAAA;;;WAGF,WAAW,KAAK,SAAS,6BAA6B,GAAG,CAAC,IAAI,CAAC,EAAE,SAAS,EAAE,CAAC,GAAG,OAAO,kBAAkB,EAAE,eAAe,CAAC,EAAE,CAAC,KAAK,GAAG,CAAC,IAAI,CAAC,EAAE,OAAO,EAAE,EAAE,QAAQ,EAAE,GAAG,OAAO,oBAAoB,EAAE,EAAE,CAAC;WACtM,WAAW,KAAK,SAAS,sBAAsB,GAAG,CAAC,IAAI,CAAC,EAAE,eAAe,EAAE,CAAC,GAAG,OAAO,QAAQ,EAAE,GAAG,OAAO,kCAAkC,CAAC,EAAE,CAAC;WAChJ,eAAe,KAAK,kBAAkB,2BAA2B,GAAG,CAAC,IAAI,CAAC,EAAE,cAAc,EAAE,SAAS,OAAO,EAAE,EAAE,CAAC;WACjH,eAAe,KAAK,SAAS,0BAA0B,GAAG,CAAC,IAAI,CAAC,EAAE,QAAQ,EAAE,iBAAiB,EAAE,CAAC;KACtG,CAAC;QACF,MAAM,GAAG,CAAA;;gBAEG,OAAO,KAAK,SAAS;KAChC,CAAC;QACF,MAAM,GAAG,CAAA;;gBAEG,KAAK,KAAK,SAAS,KAAK,OAAO,kBAAkB,GAAG,CAAC,IAAI,CAAC,EAAE,KAAK,EAAE,EAAE,QAAQ,EAAE,GAAG,OAAO,kBAAkB,EAAE,EAAE,CAAC;KAC3H,CAAC;QACF,MAAM,GAAG,CAAA;;;WAGF,SAAS,KAAK,SAAS,sCAAsC,GAAG,CAAC,IAAI,CAAC,EAAE,KAAK,EAAE,CAAC,QAAQ,OAAO,kBAAkB,CAAC,EAAE,cAAc,EAAE,GAAG,OAAO,QAAQ,EAAE,CAAC;WACzJ,UAAU,KAAK,SAAS,wCAAwC,GAAG,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,GAAG,OAAO,kBAAkB,EAAE,CAAC;KACtH,CAAC;QACF,MAAM,GAAG,CAAA,wCAAwC,CAAC;QAElD,MAAM,IAAI,GAAG,MAAM,aAAa,EAAE,CAAC;QACnC,MAAM,GAAG,CAAA,+DAA+D,IAAI,EAAE,CAAC;QAC/E,MAAM,sBAAsB,CAAC,QAAQ,CAAC,gBAAgB,CAAC,CAAC;QAExD,MAAM,QAAQ,GAAG,KAAK,IAAI,EAAE,CAAC,CAAC;YAC5B,QAAQ,EAAE,IAAI,GAAG,CAAC,CAAC,MAAM,GAAG,CAAA,qFAAqF,GAAG,CAAC,CAAC,SAAS,EAAE,kBAAkB,CAAC,CAAC,EAAE,CAAC,CAAC,GAAG,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,CAAC,GAAG,CAAC,EAAE,EAAE,CAAC,GAAG,CAAC,SAAS,EAAE,GAAG,CAAC,UAAU,CAAC,CAAC,CAAC,CAAC;YACjN,KAAK,EAAE,CAAC,MAAM,GAAG,CAAA,gEAAgE,KAAK,EAAE,CAAC,CAAC,CAAC,CAAC,EAAE,UAAU;YACxG,MAAM,EAAE,IAAI,GAAG,CAAC,CAAC,MAAM,GAAG,CAAA,gFAAgF,SAAS,EAAE,CAAC,CAAC,GAAG,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,CAAC,GAAG,CAAC,EAAE,EAAE,GAAG,CAAC,KAAK,CAAC,CAAC,CAAC;SAC1J,CAAC,CAAC;QACH,MAAM,KAAK,GAAG,MAAM,QAAQ,EAAE,CAAC;QAC/B,MAAM,CAAC,KAAK,CAAC,QAAQ,CAAC,GAAG,CAAC,WAAW,CAAC,CAAC,CAAC,OAAO,CAAC;YAC9C,EAAE,SAAS,EAAE,CAAC,0BAA0B,EAAE,eAAe,CAAC,EAAE;YAC5D,EAAE,OAAO,EAAE,EAAE,QAAQ,EAAE,4BAA4B,EAAE,EAAE;SACxD,CAAC,CAAC;QACH,MAAM,CAAC,KAAK,CAAC,QAAQ,CAAC,GAAG,CAAC,WAAW,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,EAAE,eAAe,EAAE,CAAC,gBAAgB,EAAE,0CAA0C,CAAC,EAAE,EAAE,IAAI,CAAC,CAAC,CAAC;QAC7I,MAAM,CAAC,KAAK,CAAC,QAAQ,CAAC,GAAG,CAAC,eAAe,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,EAAE,cAAc,EAAE,SAAS,OAAO,EAAE,EAAE,EAAE,IAAI,CAAC,CAAC,CAAC;QACpG,MAAM,CAAC,KAAK,CAAC,QAAQ,CAAC,GAAG,CAAC,eAAe,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,EAAE,QAAQ,EAAE,iBAAiB,EAAE,EAAE,IAAI,CAAC,CAAC,CAAC;QAC7F,MAAM,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC,OAAO,CAAC,EAAE,KAAK,EAAE,EAAE,QAAQ,EAAE,0BAA0B,EAAE,EAAE,CAAC,CAAC;QACjF,MAAM,CAAC,KAAK,CAAC,MAAM,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC,CAAC,OAAO,CAAC,EAAE,KAAK,EAAE,CAAC,+BAA+B,CAAC,EAAE,cAAc,EAAE,gBAAgB,EAAE,CAAC,CAAC;QAC5H,MAAM,CAAC,KAAK,CAAC,MAAM,CAAC,GAAG,CAAC,UAAU,CAAC,CAAC,CAAC,OAAO,CAAC,EAAE,IAAI,EAAE,GAAG,OAAO,kBAAkB,EAAE,CAAC,CAAC;QAErF,MAAM,GAAG,CAAA,+DAA+D,IAAI,EAAE,CAAC;QAC/E,MAAM,sBAAsB,CAAC,QAAQ,CAAC,gBAAgB,CAAC,CAAC;QACxD,MAAM,CAAC,MAAM,QAAQ,EAAE,CAAC,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC;IAC1C,CAAC,EAAE,iCAAiC,CAAC,CAAC;AACxC,CAAC,CAAC,CAAC"}
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"wardmote-tool-template-keys-migration.test.d.ts","sourceRoot":"","sources":["../src/wardmote-tool-template-keys-migration.test.ts"],"names":[],"mappings":""}
|
|
@@ -0,0 +1,117 @@
|
|
|
1
|
+
import { createHash, randomUUID } from "node:crypto";
|
|
2
|
+
import fs from "node:fs";
|
|
3
|
+
import { afterEach, describe, expect, it } from "vitest";
|
|
4
|
+
import postgres from "postgres";
|
|
5
|
+
import { applyPendingMigrations } from "./client.js";
|
|
6
|
+
import { EMBEDDED_POSTGRES_TEST_TIMEOUT_MS, getEmbeddedPostgresTestSupport, startEmbeddedPostgresTestDatabase, } from "./test-embedded-postgres.js";
|
|
7
|
+
const MIGRATION_FILE = "0378_wardmote_tool_template_keys.sql";
|
|
8
|
+
const RETIRED = "paperclip";
|
|
9
|
+
const cleanups = [];
|
|
10
|
+
const embeddedPostgresSupport = await getEmbeddedPostgresTestSupport();
|
|
11
|
+
const describeEmbeddedPostgres = embeddedPostgresSupport.supported ? describe : describe.skip;
|
|
12
|
+
async function migrationHash() {
|
|
13
|
+
const content = await fs.promises.readFile(new URL(`./migrations/${MIGRATION_FILE}`, import.meta.url), "utf8");
|
|
14
|
+
return createHash("sha256").update(content).digest("hex");
|
|
15
|
+
}
|
|
16
|
+
describeEmbeddedPostgres("wardmote tool template keys migration", () => {
|
|
17
|
+
afterEach(async () => {
|
|
18
|
+
await Promise.all(cleanups.splice(0).map((cleanup) => cleanup()));
|
|
19
|
+
});
|
|
20
|
+
it("renames built-in template ids, example keys and pages keys, and moves colliding custom templates aside", async () => {
|
|
21
|
+
const database = await startEmbeddedPostgresTestDatabase("wardmote-tool-template-keys-migration-");
|
|
22
|
+
cleanups.push(database.cleanup);
|
|
23
|
+
const sql = postgres(database.connectionString, { max: 1, onnotice: () => { } });
|
|
24
|
+
cleanups.push(async () => sql.end());
|
|
25
|
+
const companyId = randomUUID();
|
|
26
|
+
const otherCompanyId = randomUUID();
|
|
27
|
+
const id = () => randomUUID();
|
|
28
|
+
const app = { gallery: id(), example: id(), pages: id(), legacyPages: id(), otherPages: id(), otherWardmotePages: id() };
|
|
29
|
+
const conn = { cognee: id(), custom: id(), example: id(), pages: id(), otherPages: id() };
|
|
30
|
+
const slot = { builtIn: id(), custom: id() };
|
|
31
|
+
const profile = id();
|
|
32
|
+
const binding = id();
|
|
33
|
+
const entry = id();
|
|
34
|
+
const customTemplate = id();
|
|
35
|
+
await sql `
|
|
36
|
+
INSERT INTO "companies" ("id", "name", "issue_prefix")
|
|
37
|
+
VALUES (${companyId}, 'Template keys', 'TKA'), (${otherCompanyId}, 'Other', 'TKB')
|
|
38
|
+
`;
|
|
39
|
+
await sql `SET session_replication_role = replica`;
|
|
40
|
+
await sql `
|
|
41
|
+
INSERT INTO "tool_applications" ("id", "company_id", "name", "type", "application_key", "metadata")
|
|
42
|
+
VALUES
|
|
43
|
+
(${app.gallery}, ${companyId}, 'Cognee', 'mcp', 'cognee', '{}'),
|
|
44
|
+
(${app.example}, ${companyId}, 'Example', 'mcp', ${`${RETIRED}.examples.safe-read-only-todo-kv`}, ${sql.json({ source: `${RETIRED}_example`, exampleId: "safe-read-only-todo-kv" })}),
|
|
45
|
+
(${app.pages}, ${companyId}, 'Pages', 'mcp', ${`${RETIRED}.pages`}, '{}'),
|
|
46
|
+
(${app.legacyPages}, ${companyId}, 'Pages legacy', 'mcp', ${`pages.${RETIRED}`}, '{}'),
|
|
47
|
+
(${app.otherPages}, ${otherCompanyId}, 'Pages', 'mcp', ${`pages.${RETIRED}`}, '{}'),
|
|
48
|
+
(${app.otherWardmotePages}, ${otherCompanyId}, 'Wardmote pages', 'mcp', 'wardmote-pages', '{}')
|
|
49
|
+
`;
|
|
50
|
+
await sql `
|
|
51
|
+
INSERT INTO "tool_stdio_command_templates" ("id", "company_id", "template_key", "name", "command")
|
|
52
|
+
VALUES (${customTemplate}, ${companyId}, 'wardmote.cognee-cloud', 'Custom', 'node')
|
|
53
|
+
`;
|
|
54
|
+
await sql `
|
|
55
|
+
INSERT INTO "tool_connections" ("id", "company_id", "application_id", "name", "uid", "transport", "config", "transport_config")
|
|
56
|
+
VALUES
|
|
57
|
+
(${conn.cognee}, ${companyId}, ${app.gallery}, 'cognee', 'cognee/cognee-1', 'local_stdio', ${sql.json({ templateId: `${RETIRED}.cognee-cloud` })}, ${sql.json({ templateId: `${RETIRED}.cognee-cloud` })}),
|
|
58
|
+
(${conn.custom}, ${companyId}, ${app.gallery}, 'custom', 'custom/custom-1', 'local_stdio', ${sql.json({ templateId: "wardmote.cognee-cloud" })}, ${sql.json({ templateId: "wardmote.cognee-cloud" })}),
|
|
59
|
+
(${conn.example}, ${companyId}, ${app.example}, 'example', ${`${RETIRED}/wardmote-example-12345678`}, 'local_stdio', ${sql.json({ templateId: `${RETIRED}.synthetic-todo-kv` })}, ${sql.json({ templateId: `${RETIRED}.synthetic-todo-kv` })}),
|
|
60
|
+
(${conn.pages}, ${companyId}, ${app.legacyPages}, 'pages', 'pages/pages-1', 'mcp_remote', '{}', '{}'),
|
|
61
|
+
(${conn.otherPages}, ${otherCompanyId}, ${app.otherPages}, 'pages', 'pages/pages-2', 'mcp_remote', ${sql.json({ connectionType: "custom" })}, '{}')
|
|
62
|
+
`;
|
|
63
|
+
await sql `
|
|
64
|
+
INSERT INTO "tool_runtime_slots" ("id", "company_id", "slot_key", "provider", "provider_ref", "command_template_key", "metadata")
|
|
65
|
+
VALUES
|
|
66
|
+
(${slot.builtIn}, ${companyId}, 'slot-1', ${RETIRED}, ${`template:${RETIRED}.cognee-cloud`}, ${`${RETIRED}.cognee-cloud`}, ${sql.json({ templateId: `${RETIRED}.cognee-cloud` })}),
|
|
67
|
+
(${slot.custom}, ${companyId}, 'slot-2', 'local', 'template:wardmote.cognee-cloud', 'wardmote.cognee-cloud', '{}')
|
|
68
|
+
`;
|
|
69
|
+
await sql `
|
|
70
|
+
INSERT INTO "tool_profiles" ("id", "company_id", "profile_key", "name", "metadata")
|
|
71
|
+
VALUES (${profile}, ${companyId}, ${`${RETIRED}.examples.safe-read-only-todo-kv.profile`}, 'Example', ${sql.json({ source: `${RETIRED}_example` })})
|
|
72
|
+
`;
|
|
73
|
+
await sql `
|
|
74
|
+
INSERT INTO "tool_profile_bindings" ("id", "company_id", "profile_id", "target_type", "target_id", "metadata")
|
|
75
|
+
VALUES (${binding}, ${companyId}, ${profile}, 'company', ${companyId}, ${sql.json({ source: `${RETIRED}_example`, safeDefault: true })})
|
|
76
|
+
`;
|
|
77
|
+
await sql `
|
|
78
|
+
INSERT INTO "tool_profile_entries" ("id", "company_id", "profile_id", "selector_type", "conditions")
|
|
79
|
+
VALUES (${entry}, ${companyId}, ${profile}, 'catalog_entry', ${sql.json({ source: `${RETIRED}_example` })})
|
|
80
|
+
`;
|
|
81
|
+
await sql `SET session_replication_role = DEFAULT`;
|
|
82
|
+
const hash = await migrationHash();
|
|
83
|
+
await sql `DELETE FROM "drizzle"."__drizzle_migrations" WHERE "hash" = ${hash}`;
|
|
84
|
+
await applyPendingMigrations(database.connectionString);
|
|
85
|
+
const companies = [companyId, otherCompanyId];
|
|
86
|
+
const snapshot = async () => ({
|
|
87
|
+
applications: new Map((await sql `SELECT "id", "application_key", "metadata" FROM "tool_applications" WHERE "company_id" IN ${sql(companies)}`).map((row) => [row.id, [row.application_key, row.metadata]])),
|
|
88
|
+
connections: new Map((await sql `SELECT "id", "uid", "config", "transport_config" FROM "tool_connections" WHERE "company_id" IN ${sql(companies)}`).map((row) => [row.id, [row.uid, row.config, row.transport_config]])),
|
|
89
|
+
slots: new Map((await sql `SELECT "id", "provider", "provider_ref", "command_template_key", "metadata" FROM "tool_runtime_slots" WHERE "company_id" = ${companyId}`).map((row) => [row.id, [row.provider, row.provider_ref, row.command_template_key, row.metadata]])),
|
|
90
|
+
templates: (await sql `SELECT "template_key" FROM "tool_stdio_command_templates" WHERE "company_id" = ${companyId}`).map((row) => row.template_key),
|
|
91
|
+
profileKey: (await sql `SELECT "profile_key", "metadata" FROM "tool_profiles" WHERE "id" = ${profile}`)[0],
|
|
92
|
+
binding: (await sql `SELECT "metadata" FROM "tool_profile_bindings" WHERE "id" = ${binding}`)[0]?.metadata,
|
|
93
|
+
entry: (await sql `SELECT "conditions" FROM "tool_profile_entries" WHERE "id" = ${entry}`)[0]?.conditions,
|
|
94
|
+
});
|
|
95
|
+
const after = await snapshot();
|
|
96
|
+
expect(after.templates).toEqual(["wardmote.cognee-cloud.custom"]);
|
|
97
|
+
expect(after.connections.get(conn.custom)).toEqual(["custom/custom-1", { templateId: "wardmote.cognee-cloud.custom" }, { templateId: "wardmote.cognee-cloud.custom" }]);
|
|
98
|
+
expect(after.connections.get(conn.cognee)).toEqual(["cognee/cognee-1", { templateId: "wardmote.cognee-cloud" }, { templateId: "wardmote.cognee-cloud" }]);
|
|
99
|
+
expect(after.slots.get(slot.custom)).toEqual(["local", "template:wardmote.cognee-cloud", "wardmote.cognee-cloud.custom", {}]);
|
|
100
|
+
expect(after.slots.get(slot.builtIn)).toEqual(["wardmote", "template:wardmote.cognee-cloud", "wardmote.cognee-cloud", { templateId: "wardmote.cognee-cloud" }]);
|
|
101
|
+
expect(after.applications.get(app.example)).toEqual(["wardmote.examples.safe-read-only-todo-kv", { source: "wardmote_example", exampleId: "safe-read-only-todo-kv" }]);
|
|
102
|
+
expect(after.connections.get(conn.example)).toEqual(["wardmote/wardmote-example-12345678", { templateId: "wardmote.synthetic-todo-kv" }, { templateId: "wardmote.synthetic-todo-kv" }]);
|
|
103
|
+
expect(after.profileKey).toEqual({ profile_key: "wardmote.examples.safe-read-only-todo-kv.profile", metadata: { source: "wardmote_example" } });
|
|
104
|
+
expect(after.binding).toEqual({ source: "wardmote_example", safeDefault: true });
|
|
105
|
+
expect(after.entry).toEqual({ source: "wardmote_example" });
|
|
106
|
+
expect(after.applications.get(app.pages)?.[0]).toBe("wardmote-pages");
|
|
107
|
+
expect(after.applications.get(app.legacyPages)?.[0]).toBe(`pages.${RETIRED}`);
|
|
108
|
+
expect(after.connections.get(conn.pages)?.[1]).toEqual({ connectionType: "pages" });
|
|
109
|
+
expect(after.applications.get(app.otherPages)?.[0]).toBe(`pages.${RETIRED}`);
|
|
110
|
+
expect(after.applications.get(app.otherWardmotePages)?.[0]).toBe("wardmote-pages");
|
|
111
|
+
expect(after.connections.get(conn.otherPages)?.[1]).toEqual({ connectionType: "custom" });
|
|
112
|
+
await sql `DELETE FROM "drizzle"."__drizzle_migrations" WHERE "hash" = ${hash}`;
|
|
113
|
+
await applyPendingMigrations(database.connectionString);
|
|
114
|
+
expect(await snapshot()).toEqual(after);
|
|
115
|
+
}, EMBEDDED_POSTGRES_TEST_TIMEOUT_MS);
|
|
116
|
+
});
|
|
117
|
+
//# sourceMappingURL=wardmote-tool-template-keys-migration.test.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"wardmote-tool-template-keys-migration.test.js","sourceRoot":"","sources":["../src/wardmote-tool-template-keys-migration.test.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,UAAU,EAAE,UAAU,EAAE,MAAM,aAAa,CAAC;AACrD,OAAO,EAAE,MAAM,SAAS,CAAC;AACzB,OAAO,EAAE,SAAS,EAAE,QAAQ,EAAE,MAAM,EAAE,EAAE,EAAE,MAAM,QAAQ,CAAC;AACzD,OAAO,QAAQ,MAAM,UAAU,CAAC;AAChC,OAAO,EAAE,sBAAsB,EAAE,MAAM,aAAa,CAAC;AACrD,OAAO,EACL,iCAAiC,EACjC,8BAA8B,EAC9B,iCAAiC,GAClC,MAAM,6BAA6B,CAAC;AAErC,MAAM,cAAc,GAAG,sCAAsC,CAAC;AAC9D,MAAM,OAAO,GAAG,WAAW,CAAC;AAC5B,MAAM,QAAQ,GAA+B,EAAE,CAAC;AAChD,MAAM,uBAAuB,GAAG,MAAM,8BAA8B,EAAE,CAAC;AACvE,MAAM,wBAAwB,GAAG,uBAAuB,CAAC,SAAS,CAAC,CAAC,CAAC,QAAQ,CAAC,CAAC,CAAC,QAAQ,CAAC,IAAI,CAAC;AAE9F,KAAK,UAAU,aAAa;IAC1B,MAAM,OAAO,GAAG,MAAM,EAAE,CAAC,QAAQ,CAAC,QAAQ,CAAC,IAAI,GAAG,CAAC,gBAAgB,cAAc,EAAE,EAAE,OAAO,IAAI,CAAC,GAAG,CAAC,EAAE,MAAM,CAAC,CAAC;IAC/G,OAAO,UAAU,CAAC,QAAQ,CAAC,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;AAC5D,CAAC;AAED,wBAAwB,CAAC,uCAAuC,EAAE,GAAG,EAAE;IACrE,SAAS,CAAC,KAAK,IAAI,EAAE;QACnB,MAAM,OAAO,CAAC,GAAG,CAAC,QAAQ,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,OAAO,EAAE,EAAE,CAAC,OAAO,EAAE,CAAC,CAAC,CAAC;IACpE,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,wGAAwG,EAAE,KAAK,IAAI,EAAE;QACtH,MAAM,QAAQ,GAAG,MAAM,iCAAiC,CAAC,wCAAwC,CAAC,CAAC;QACnG,QAAQ,CAAC,IAAI,CAAC,QAAQ,CAAC,OAAO,CAAC,CAAC;QAChC,MAAM,GAAG,GAAG,QAAQ,CAAC,QAAQ,CAAC,gBAAgB,EAAE,EAAE,GAAG,EAAE,CAAC,EAAE,QAAQ,EAAE,GAAG,EAAE,GAAE,CAAC,EAAE,CAAC,CAAC;QAChF,QAAQ,CAAC,IAAI,CAAC,KAAK,IAAI,EAAE,CAAC,GAAG,CAAC,GAAG,EAAE,CAAC,CAAC;QAErC,MAAM,SAAS,GAAG,UAAU,EAAE,CAAC;QAC/B,MAAM,cAAc,GAAG,UAAU,EAAE,CAAC;QACpC,MAAM,EAAE,GAAG,GAAG,EAAE,CAAC,UAAU,EAAE,CAAC;QAC9B,MAAM,GAAG,GAAG,EAAE,OAAO,EAAE,EAAE,EAAE,EAAE,OAAO,EAAE,EAAE,EAAE,EAAE,KAAK,EAAE,EAAE,EAAE,EAAE,WAAW,EAAE,EAAE,EAAE,EAAE,UAAU,EAAE,EAAE,EAAE,EAAE,kBAAkB,EAAE,EAAE,EAAE,EAAE,CAAC;QACzH,MAAM,IAAI,GAAG,EAAE,MAAM,EAAE,EAAE,EAAE,EAAE,MAAM,EAAE,EAAE,EAAE,EAAE,OAAO,EAAE,EAAE,EAAE,EAAE,KAAK,EAAE,EAAE,EAAE,EAAE,UAAU,EAAE,EAAE,EAAE,EAAE,CAAC;QAC1F,MAAM,IAAI,GAAG,EAAE,OAAO,EAAE,EAAE,EAAE,EAAE,MAAM,EAAE,EAAE,EAAE,EAAE,CAAC;QAC7C,MAAM,OAAO,GAAG,EAAE,EAAE,CAAC;QACrB,MAAM,OAAO,GAAG,EAAE,EAAE,CAAC;QACrB,MAAM,KAAK,GAAG,EAAE,EAAE,CAAC;QACnB,MAAM,cAAc,GAAG,EAAE,EAAE,CAAC;QAE5B,MAAM,GAAG,CAAA;;gBAEG,SAAS,+BAA+B,cAAc;KACjE,CAAC;QACF,MAAM,GAAG,CAAA,wCAAwC,CAAC;QAClD,MAAM,GAAG,CAAA;;;WAGF,GAAG,CAAC,OAAO,KAAK,SAAS;WACzB,GAAG,CAAC,OAAO,KAAK,SAAS,uBAAuB,GAAG,OAAO,kCAAkC,KAAK,GAAG,CAAC,IAAI,CAAC,EAAE,MAAM,EAAE,GAAG,OAAO,UAAU,EAAE,SAAS,EAAE,wBAAwB,EAAE,CAAC;WAChL,GAAG,CAAC,KAAK,KAAK,SAAS,qBAAqB,GAAG,OAAO,QAAQ;WAC9D,GAAG,CAAC,WAAW,KAAK,SAAS,4BAA4B,SAAS,OAAO,EAAE;WAC3E,GAAG,CAAC,UAAU,KAAK,cAAc,qBAAqB,SAAS,OAAO,EAAE;WACxE,GAAG,CAAC,kBAAkB,KAAK,cAAc;KAC/C,CAAC;QACF,MAAM,GAAG,CAAA;;gBAEG,cAAc,KAAK,SAAS;KACvC,CAAC;QACF,MAAM,GAAG,CAAA;;;WAGF,IAAI,CAAC,MAAM,KAAK,SAAS,KAAK,GAAG,CAAC,OAAO,iDAAiD,GAAG,CAAC,IAAI,CAAC,EAAE,UAAU,EAAE,GAAG,OAAO,eAAe,EAAE,CAAC,KAAK,GAAG,CAAC,IAAI,CAAC,EAAE,UAAU,EAAE,GAAG,OAAO,eAAe,EAAE,CAAC;WACrM,IAAI,CAAC,MAAM,KAAK,SAAS,KAAK,GAAG,CAAC,OAAO,iDAAiD,GAAG,CAAC,IAAI,CAAC,EAAE,UAAU,EAAE,uBAAuB,EAAE,CAAC,KAAK,GAAG,CAAC,IAAI,CAAC,EAAE,UAAU,EAAE,uBAAuB,EAAE,CAAC;WACjM,IAAI,CAAC,OAAO,KAAK,SAAS,KAAK,GAAG,CAAC,OAAO,gBAAgB,GAAG,OAAO,4BAA4B,oBAAoB,GAAG,CAAC,IAAI,CAAC,EAAE,UAAU,EAAE,GAAG,OAAO,oBAAoB,EAAE,CAAC,KAAK,GAAG,CAAC,IAAI,CAAC,EAAE,UAAU,EAAE,GAAG,OAAO,oBAAoB,EAAE,CAAC;WACzO,IAAI,CAAC,KAAK,KAAK,SAAS,KAAK,GAAG,CAAC,WAAW;WAC5C,IAAI,CAAC,UAAU,KAAK,cAAc,KAAK,GAAG,CAAC,UAAU,6CAA6C,GAAG,CAAC,IAAI,CAAC,EAAE,cAAc,EAAE,QAAQ,EAAE,CAAC;KAC9I,CAAC;QACF,MAAM,GAAG,CAAA;;;WAGF,IAAI,CAAC,OAAO,KAAK,SAAS,eAAe,OAAO,KAAK,YAAY,OAAO,eAAe,KAAK,GAAG,OAAO,eAAe,KAAK,GAAG,CAAC,IAAI,CAAC,EAAE,UAAU,EAAE,GAAG,OAAO,eAAe,EAAE,CAAC;WAC7K,IAAI,CAAC,MAAM,KAAK,SAAS;KAC/B,CAAC;QACF,MAAM,GAAG,CAAA;;gBAEG,OAAO,KAAK,SAAS,KAAK,GAAG,OAAO,0CAA0C,gBAAgB,GAAG,CAAC,IAAI,CAAC,EAAE,MAAM,EAAE,GAAG,OAAO,UAAU,EAAE,CAAC;KACnJ,CAAC;QACF,MAAM,GAAG,CAAA;;gBAEG,OAAO,KAAK,SAAS,KAAK,OAAO,gBAAgB,SAAS,KAAK,GAAG,CAAC,IAAI,CAAC,EAAE,MAAM,EAAE,GAAG,OAAO,UAAU,EAAE,WAAW,EAAE,IAAI,EAAE,CAAC;KACvI,CAAC;QACF,MAAM,GAAG,CAAA;;gBAEG,KAAK,KAAK,SAAS,KAAK,OAAO,sBAAsB,GAAG,CAAC,IAAI,CAAC,EAAE,MAAM,EAAE,GAAG,OAAO,UAAU,EAAE,CAAC;KAC1G,CAAC;QACF,MAAM,GAAG,CAAA,wCAAwC,CAAC;QAElD,MAAM,IAAI,GAAG,MAAM,aAAa,EAAE,CAAC;QACnC,MAAM,GAAG,CAAA,+DAA+D,IAAI,EAAE,CAAC;QAC/E,MAAM,sBAAsB,CAAC,QAAQ,CAAC,gBAAgB,CAAC,CAAC;QAExD,MAAM,SAAS,GAAG,CAAC,SAAS,EAAE,cAAc,CAAC,CAAC;QAC9C,MAAM,QAAQ,GAAG,KAAK,IAAI,EAAE,CAAC,CAAC;YAC5B,YAAY,EAAE,IAAI,GAAG,CAAC,CAAC,MAAM,GAAG,CAAA,6FAA6F,GAAG,CAAC,SAAS,CAAC,EAAE,CAAC,CAAC,GAAG,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,CAAC,GAAG,CAAC,EAAE,EAAE,CAAC,GAAG,CAAC,eAAe,EAAE,GAAG,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAC;YAC3M,WAAW,EAAE,IAAI,GAAG,CAAC,CAAC,MAAM,GAAG,CAAA,kGAAkG,GAAG,CAAC,SAAS,CAAC,EAAE,CAAC,CAAC,GAAG,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,CAAC,GAAG,CAAC,EAAE,EAAE,CAAC,GAAG,CAAC,GAAG,EAAE,GAAG,CAAC,MAAM,EAAE,GAAG,CAAC,gBAAgB,CAAC,CAAC,CAAC,CAAC;YACvN,KAAK,EAAE,IAAI,GAAG,CAAC,CAAC,MAAM,GAAG,CAAA,8HAA8H,SAAS,EAAE,CAAC,CAAC,GAAG,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,CAAC,GAAG,CAAC,EAAE,EAAE,CAAC,GAAG,CAAC,QAAQ,EAAE,GAAG,CAAC,YAAY,EAAE,GAAG,CAAC,oBAAoB,EAAE,GAAG,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAC;YACrQ,SAAS,EAAE,CAAC,MAAM,GAAG,CAAA,kFAAkF,SAAS,EAAE,CAAC,CAAC,GAAG,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,GAAG,CAAC,YAAY,CAAC;YAClJ,UAAU,EAAE,CAAC,MAAM,GAAG,CAAA,sEAAsE,OAAO,EAAE,CAAC,CAAC,CAAC,CAAC;YACzG,OAAO,EAAE,CAAC,MAAM,GAAG,CAAA,+DAA+D,OAAO,EAAE,CAAC,CAAC,CAAC,CAAC,EAAE,QAAQ;YACzG,KAAK,EAAE,CAAC,MAAM,GAAG,CAAA,gEAAgE,KAAK,EAAE,CAAC,CAAC,CAAC,CAAC,EAAE,UAAU;SACzG,CAAC,CAAC;QACH,MAAM,KAAK,GAAG,MAAM,QAAQ,EAAE,CAAC;QAE/B,MAAM,CAAC,KAAK,CAAC,SAAS,CAAC,CAAC,OAAO,CAAC,CAAC,8BAA8B,CAAC,CAAC,CAAC;QAClE,MAAM,CAAC,KAAK,CAAC,WAAW,CAAC,GAAG,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,iBAAiB,EAAE,EAAE,UAAU,EAAE,8BAA8B,EAAE,EAAE,EAAE,UAAU,EAAE,8BAA8B,EAAE,CAAC,CAAC,CAAC;QACxK,MAAM,CAAC,KAAK,CAAC,WAAW,CAAC,GAAG,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,iBAAiB,EAAE,EAAE,UAAU,EAAE,uBAAuB,EAAE,EAAE,EAAE,UAAU,EAAE,uBAAuB,EAAE,CAAC,CAAC,CAAC;QAC1J,MAAM,CAAC,KAAK,CAAC,KAAK,CAAC,GAAG,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,OAAO,EAAE,gCAAgC,EAAE,8BAA8B,EAAE,EAAE,CAAC,CAAC,CAAC;QAC9H,MAAM,CAAC,KAAK,CAAC,KAAK,CAAC,GAAG,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,UAAU,EAAE,gCAAgC,EAAE,uBAAuB,EAAE,EAAE,UAAU,EAAE,uBAAuB,EAAE,CAAC,CAAC,CAAC;QAEhK,MAAM,CAAC,KAAK,CAAC,YAAY,CAAC,GAAG,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,0CAA0C,EAAE,EAAE,MAAM,EAAE,kBAAkB,EAAE,SAAS,EAAE,wBAAwB,EAAE,CAAC,CAAC,CAAC;QACvK,MAAM,CAAC,KAAK,CAAC,WAAW,CAAC,GAAG,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,oCAAoC,EAAE,EAAE,UAAU,EAAE,4BAA4B,EAAE,EAAE,EAAE,UAAU,EAAE,4BAA4B,EAAE,CAAC,CAAC,CAAC;QACxL,MAAM,CAAC,KAAK,CAAC,UAAU,CAAC,CAAC,OAAO,CAAC,EAAE,WAAW,EAAE,kDAAkD,EAAE,QAAQ,EAAE,EAAE,MAAM,EAAE,kBAAkB,EAAE,EAAE,CAAC,CAAC;QAChJ,MAAM,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,OAAO,CAAC,EAAE,MAAM,EAAE,kBAAkB,EAAE,WAAW,EAAE,IAAI,EAAE,CAAC,CAAC;QACjF,MAAM,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC,OAAO,CAAC,EAAE,MAAM,EAAE,kBAAkB,EAAE,CAAC,CAAC;QAE5D,MAAM,CAAC,KAAK,CAAC,YAAY,CAAC,GAAG,CAAC,GAAG,CAAC,KAAK,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,gBAAgB,CAAC,CAAC;QACtE,MAAM,CAAC,KAAK,CAAC,YAAY,CAAC,GAAG,CAAC,GAAG,CAAC,WAAW,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,SAAS,OAAO,EAAE,CAAC,CAAC;QAC9E,MAAM,CAAC,KAAK,CAAC,WAAW,CAAC,GAAG,CAAC,IAAI,CAAC,KAAK,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,EAAE,cAAc,EAAE,OAAO,EAAE,CAAC,CAAC;QACpF,MAAM,CAAC,KAAK,CAAC,YAAY,CAAC,GAAG,CAAC,GAAG,CAAC,UAAU,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,SAAS,OAAO,EAAE,CAAC,CAAC;QAC7E,MAAM,CAAC,KAAK,CAAC,YAAY,CAAC,GAAG,CAAC,GAAG,CAAC,kBAAkB,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,gBAAgB,CAAC,CAAC;QACnF,MAAM,CAAC,KAAK,CAAC,WAAW,CAAC,GAAG,CAAC,IAAI,CAAC,UAAU,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,EAAE,cAAc,EAAE,QAAQ,EAAE,CAAC,CAAC;QAE1F,MAAM,GAAG,CAAA,+DAA+D,IAAI,EAAE,CAAC;QAC/E,MAAM,sBAAsB,CAAC,QAAQ,CAAC,gBAAgB,CAAC,CAAC;QACxD,MAAM,CAAC,MAAM,QAAQ,EAAE,CAAC,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC;IAC1C,CAAC,EAAE,iCAAiC,CAAC,CAAC;AACxC,CAAC,CAAC,CAAC"}
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"wardmote-work-product-provider-migration.test.d.ts","sourceRoot":"","sources":["../src/wardmote-work-product-provider-migration.test.ts"],"names":[],"mappings":""}
|
|
@@ -0,0 +1,65 @@
|
|
|
1
|
+
import { createHash, randomUUID } from "node:crypto";
|
|
2
|
+
import fs from "node:fs";
|
|
3
|
+
import { afterEach, describe, expect, it } from "vitest";
|
|
4
|
+
import postgres from "postgres";
|
|
5
|
+
import { applyPendingMigrations } from "./client.js";
|
|
6
|
+
import { EMBEDDED_POSTGRES_TEST_TIMEOUT_MS, getEmbeddedPostgresTestSupport, startEmbeddedPostgresTestDatabase, } from "./test-embedded-postgres.js";
|
|
7
|
+
const MIGRATION_FILE = "0382_wardmote_work_product_provider.sql";
|
|
8
|
+
const RETIRED = "paperclip";
|
|
9
|
+
const cleanups = [];
|
|
10
|
+
const embeddedPostgresSupport = await getEmbeddedPostgresTestSupport();
|
|
11
|
+
const describeEmbeddedPostgres = embeddedPostgresSupport.supported ? describe : describe.skip;
|
|
12
|
+
async function migrationHash() {
|
|
13
|
+
const content = await fs.promises.readFile(new URL(`./migrations/${MIGRATION_FILE}`, import.meta.url), "utf8");
|
|
14
|
+
return createHash("sha256").update(content).digest("hex");
|
|
15
|
+
}
|
|
16
|
+
describeEmbeddedPostgres("wardmote work product provider migration", () => {
|
|
17
|
+
afterEach(async () => {
|
|
18
|
+
await Promise.all(cleanups.splice(0).map((cleanup) => cleanup()));
|
|
19
|
+
});
|
|
20
|
+
it("renames attachment artifact providers and smoke screenshot refs and leaves other providers", async () => {
|
|
21
|
+
const database = await startEmbeddedPostgresTestDatabase("wardmote-work-product-provider-migration-");
|
|
22
|
+
cleanups.push(database.cleanup);
|
|
23
|
+
const sql = postgres(database.connectionString, { max: 1, onnotice: () => { } });
|
|
24
|
+
cleanups.push(async () => sql.end());
|
|
25
|
+
const companyId = randomUUID();
|
|
26
|
+
const issueId = randomUUID();
|
|
27
|
+
const artifact = randomUUID();
|
|
28
|
+
const pullRequest = randomUUID();
|
|
29
|
+
const runId = randomUUID();
|
|
30
|
+
const step = randomUUID();
|
|
31
|
+
const otherStep = randomUUID();
|
|
32
|
+
const attachmentId = randomUUID();
|
|
33
|
+
await sql `INSERT INTO "companies" ("id", "name", "issue_prefix") VALUES (${companyId}, 'Work products', 'WPP')`;
|
|
34
|
+
await sql `SET session_replication_role = replica`;
|
|
35
|
+
await sql `
|
|
36
|
+
INSERT INTO "issue_work_products" ("id", "company_id", "issue_id", "type", "provider", "external_id", "title", "status")
|
|
37
|
+
VALUES
|
|
38
|
+
(${artifact}, ${companyId}, ${issueId}, 'artifact', ${RETIRED}, ${attachmentId}, 'report.pdf', 'active'),
|
|
39
|
+
(${pullRequest}, ${companyId}, ${issueId}, 'pull_request', 'github', '42', 'PR', 'active')
|
|
40
|
+
`;
|
|
41
|
+
await sql `
|
|
42
|
+
INSERT INTO "smoke_run_steps" ("id", "company_id", "run_id", "path", "scenario_step", "status", "screenshot_artifact_ref")
|
|
43
|
+
VALUES
|
|
44
|
+
(${step}, ${companyId}, ${runId}, 'browser', 'open', 'passed', ${sql.json({ provider: RETIRED, attachmentId })}),
|
|
45
|
+
(${otherStep}, ${companyId}, ${runId}, 'browser', 'close', 'passed', NULL)
|
|
46
|
+
`;
|
|
47
|
+
await sql `SET session_replication_role = DEFAULT`;
|
|
48
|
+
const hash = await migrationHash();
|
|
49
|
+
await sql `DELETE FROM "drizzle"."__drizzle_migrations" WHERE "hash" = ${hash}`;
|
|
50
|
+
await applyPendingMigrations(database.connectionString);
|
|
51
|
+
const snapshot = async () => ({
|
|
52
|
+
products: new Map((await sql `SELECT "id", "provider" FROM "issue_work_products" WHERE "company_id" = ${companyId}`).map((row) => [row.id, row.provider])),
|
|
53
|
+
steps: new Map((await sql `SELECT "id", "screenshot_artifact_ref" FROM "smoke_run_steps" WHERE "company_id" = ${companyId}`).map((row) => [row.id, row.screenshot_artifact_ref])),
|
|
54
|
+
});
|
|
55
|
+
const after = await snapshot();
|
|
56
|
+
expect(after.products.get(artifact)).toBe("wardmote");
|
|
57
|
+
expect(after.products.get(pullRequest)).toBe("github");
|
|
58
|
+
expect(after.steps.get(step)).toEqual({ provider: "wardmote", attachmentId });
|
|
59
|
+
expect(after.steps.get(otherStep)).toBeNull();
|
|
60
|
+
await sql `DELETE FROM "drizzle"."__drizzle_migrations" WHERE "hash" = ${hash}`;
|
|
61
|
+
await applyPendingMigrations(database.connectionString);
|
|
62
|
+
expect(await snapshot()).toEqual(after);
|
|
63
|
+
}, EMBEDDED_POSTGRES_TEST_TIMEOUT_MS);
|
|
64
|
+
});
|
|
65
|
+
//# sourceMappingURL=wardmote-work-product-provider-migration.test.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"wardmote-work-product-provider-migration.test.js","sourceRoot":"","sources":["../src/wardmote-work-product-provider-migration.test.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,UAAU,EAAE,UAAU,EAAE,MAAM,aAAa,CAAC;AACrD,OAAO,EAAE,MAAM,SAAS,CAAC;AACzB,OAAO,EAAE,SAAS,EAAE,QAAQ,EAAE,MAAM,EAAE,EAAE,EAAE,MAAM,QAAQ,CAAC;AACzD,OAAO,QAAQ,MAAM,UAAU,CAAC;AAChC,OAAO,EAAE,sBAAsB,EAAE,MAAM,aAAa,CAAC;AACrD,OAAO,EACL,iCAAiC,EACjC,8BAA8B,EAC9B,iCAAiC,GAClC,MAAM,6BAA6B,CAAC;AAErC,MAAM,cAAc,GAAG,yCAAyC,CAAC;AACjE,MAAM,OAAO,GAAG,WAAW,CAAC;AAC5B,MAAM,QAAQ,GAA+B,EAAE,CAAC;AAChD,MAAM,uBAAuB,GAAG,MAAM,8BAA8B,EAAE,CAAC;AACvE,MAAM,wBAAwB,GAAG,uBAAuB,CAAC,SAAS,CAAC,CAAC,CAAC,QAAQ,CAAC,CAAC,CAAC,QAAQ,CAAC,IAAI,CAAC;AAE9F,KAAK,UAAU,aAAa;IAC1B,MAAM,OAAO,GAAG,MAAM,EAAE,CAAC,QAAQ,CAAC,QAAQ,CAAC,IAAI,GAAG,CAAC,gBAAgB,cAAc,EAAE,EAAE,OAAO,IAAI,CAAC,GAAG,CAAC,EAAE,MAAM,CAAC,CAAC;IAC/G,OAAO,UAAU,CAAC,QAAQ,CAAC,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;AAC5D,CAAC;AAED,wBAAwB,CAAC,0CAA0C,EAAE,GAAG,EAAE;IACxE,SAAS,CAAC,KAAK,IAAI,EAAE;QACnB,MAAM,OAAO,CAAC,GAAG,CAAC,QAAQ,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,OAAO,EAAE,EAAE,CAAC,OAAO,EAAE,CAAC,CAAC,CAAC;IACpE,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,4FAA4F,EAAE,KAAK,IAAI,EAAE;QAC1G,MAAM,QAAQ,GAAG,MAAM,iCAAiC,CAAC,2CAA2C,CAAC,CAAC;QACtG,QAAQ,CAAC,IAAI,CAAC,QAAQ,CAAC,OAAO,CAAC,CAAC;QAChC,MAAM,GAAG,GAAG,QAAQ,CAAC,QAAQ,CAAC,gBAAgB,EAAE,EAAE,GAAG,EAAE,CAAC,EAAE,QAAQ,EAAE,GAAG,EAAE,GAAE,CAAC,EAAE,CAAC,CAAC;QAChF,QAAQ,CAAC,IAAI,CAAC,KAAK,IAAI,EAAE,CAAC,GAAG,CAAC,GAAG,EAAE,CAAC,CAAC;QAErC,MAAM,SAAS,GAAG,UAAU,EAAE,CAAC;QAC/B,MAAM,OAAO,GAAG,UAAU,EAAE,CAAC;QAC7B,MAAM,QAAQ,GAAG,UAAU,EAAE,CAAC;QAC9B,MAAM,WAAW,GAAG,UAAU,EAAE,CAAC;QACjC,MAAM,KAAK,GAAG,UAAU,EAAE,CAAC;QAC3B,MAAM,IAAI,GAAG,UAAU,EAAE,CAAC;QAC1B,MAAM,SAAS,GAAG,UAAU,EAAE,CAAC;QAC/B,MAAM,YAAY,GAAG,UAAU,EAAE,CAAC;QAClC,MAAM,GAAG,CAAA,kEAAkE,SAAS,2BAA2B,CAAC;QAChH,MAAM,GAAG,CAAA,wCAAwC,CAAC;QAClD,MAAM,GAAG,CAAA;;;WAGF,QAAQ,KAAK,SAAS,KAAK,OAAO,iBAAiB,OAAO,KAAK,YAAY;WAC3E,WAAW,KAAK,SAAS,KAAK,OAAO;KAC3C,CAAC;QACF,MAAM,GAAG,CAAA;;;WAGF,IAAI,KAAK,SAAS,KAAK,KAAK,kCAAkC,GAAG,CAAC,IAAI,CAAC,EAAE,QAAQ,EAAE,OAAO,EAAE,YAAY,EAAE,CAAC;WAC3G,SAAS,KAAK,SAAS,KAAK,KAAK;KACvC,CAAC;QACF,MAAM,GAAG,CAAA,wCAAwC,CAAC;QAElD,MAAM,IAAI,GAAG,MAAM,aAAa,EAAE,CAAC;QACnC,MAAM,GAAG,CAAA,+DAA+D,IAAI,EAAE,CAAC;QAC/E,MAAM,sBAAsB,CAAC,QAAQ,CAAC,gBAAgB,CAAC,CAAC;QAExD,MAAM,QAAQ,GAAG,KAAK,IAAI,EAAE,CAAC,CAAC;YAC5B,QAAQ,EAAE,IAAI,GAAG,CAAC,CAAC,MAAM,GAAG,CAAA,2EAA2E,SAAS,EAAE,CAAC,CAAC,GAAG,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,CAAC,GAAG,CAAC,EAAE,EAAE,GAAG,CAAC,QAAQ,CAAC,CAAC,CAAC;YACzJ,KAAK,EAAE,IAAI,GAAG,CAAC,CAAC,MAAM,GAAG,CAAA,sFAAsF,SAAS,EAAE,CAAC,CAAC,GAAG,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,CAAC,GAAG,CAAC,EAAE,EAAE,GAAG,CAAC,uBAAuB,CAAC,CAAC,CAAC;SACjL,CAAC,CAAC;QACH,MAAM,KAAK,GAAG,MAAM,QAAQ,EAAE,CAAC;QAC/B,MAAM,CAAC,KAAK,CAAC,QAAQ,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC;QACtD,MAAM,CAAC,KAAK,CAAC,QAAQ,CAAC,GAAG,CAAC,WAAW,CAAC,CAAC,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;QACvD,MAAM,CAAC,KAAK,CAAC,KAAK,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC,CAAC,OAAO,CAAC,EAAE,QAAQ,EAAE,UAAU,EAAE,YAAY,EAAE,CAAC,CAAC;QAC9E,MAAM,CAAC,KAAK,CAAC,KAAK,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC,CAAC,QAAQ,EAAE,CAAC;QAE9C,MAAM,GAAG,CAAA,+DAA+D,IAAI,EAAE,CAAC;QAC/E,MAAM,sBAAsB,CAAC,QAAQ,CAAC,gBAAgB,CAAC,CAAC;QACxD,MAAM,CAAC,MAAM,QAAQ,EAAE,CAAC,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC;IAC1C,CAAC,EAAE,iCAAiC,CAAC,CAAC;AACxC,CAAC,CAAC,CAAC"}
|
package/package.json
CHANGED
|
@@ -1,6 +1,6 @@
|
|
|
1
1
|
{
|
|
2
2
|
"name": "@tickernelz/wardmote-db",
|
|
3
|
-
"version": "2026.
|
|
3
|
+
"version": "2026.1114.0",
|
|
4
4
|
"license": "MIT",
|
|
5
5
|
"homepage": "https://github.com/tickernelz/wardmote",
|
|
6
6
|
"bugs": {
|
|
@@ -39,7 +39,7 @@
|
|
|
39
39
|
"seed": "pnpm --filter @tickernelz/wardmote-server exec tsx scripts/seed.ts"
|
|
40
40
|
},
|
|
41
41
|
"dependencies": {
|
|
42
|
-
"@tickernelz/wardmote-shared": "2026.
|
|
42
|
+
"@tickernelz/wardmote-shared": "2026.1114.0",
|
|
43
43
|
"drizzle-orm": "^0.45.2",
|
|
44
44
|
"embedded-postgres": "^18.1.0-beta.16",
|
|
45
45
|
"postgres": "^3.4.9"
|