Files
PaperClipAI/packages/db/src/private-task-draft-assets-migration.test.ts
T
DottaandPaperclip 3a726e676f feat: enforce private task permissions across execution and data (#10633)
Enforce private task and project access across direct reads, search, execution, files, plugins, live delivery, and sharing mutations. Preserve downward-only sharing, current responsible-user authorization, and audited emergency access. Bind historical draft assets with migration 0314.

Co-Authored-By: Paperclip <noreply@paperclip.ing>
2026-10-07 07:06:40 -05:00

36 lines
2.2 KiB
TypeScript

import { createHash, randomUUID } from "node:crypto";
import fs from "node:fs/promises";
import postgres from "postgres";
import { expect, it } from "vitest";
import { applyPendingMigrations } from "./client.js";
import { getEmbeddedPostgresTestSupport, startEmbeddedPostgresTestDatabase } from "./test-embedded-postgres.js";
const embeddedPostgresSupport = await getEmbeddedPostgresTestSupport();
const itEmbeddedPostgres = embeddedPostgresSupport.supported ? it : it.skip;
itEmbeddedPostgres("backfills owned draft images without attaching another uploader's images", async () => {
const database = await startEmbeddedPostgresTestDatabase("paperclip-draft-privacy-");
const sql = postgres(database.connectionString, { max: 1, onnotice: () => {} });
try {
const migration = await fs.readFile(new URL("./migrations/0314_private_task_draft_assets.sql", import.meta.url), "utf8");
const hash = createHash("sha256").update(migration).digest("hex");
const company = randomUUID(), task = randomUUID(), owned = randomUUID(), foreign = randomUUID();
await sql`INSERT INTO companies (id, name, issue_prefix) VALUES (${company}, 'Draft privacy', 'DRAFT')`;
for (const [id, owner] of [[owned, 'owner'], [foreign, 'someone-else']]) {
await sql`INSERT INTO assets (id, company_id, provider, object_key, content_type, byte_size, sha256, created_by_user_id)
VALUES (${id}, ${company}, 'local_disk', ${`${company}/assets/issues/drafts/${id}`}, 'image/png', 1, 'test', ${owner})`;
}
await sql`INSERT INTO issues (id, company_id, title, description, visibility, created_by_user_id)
VALUES (${task}, ${company}, 'Private task', ${`![owned](/api/assets/${owned}/content) ![foreign](/api/assets/${foreign}/content)`}, 'private', 'owner')`;
for (let run = 0; run < 2; run += 1) {
await sql`DELETE FROM drizzle.__drizzle_migrations WHERE hash = ${hash}`;
await applyPendingMigrations(database.connectionString);
expect(await sql`SELECT issue_id, asset_id FROM issue_attachments WHERE company_id = ${company}`)
.toEqual([{ issue_id: task, asset_id: owned }]);
}
} finally {
await sql.end();
await database.cleanup();
}
}, 60_000);