fix(communication): enforce room and project membership boundaries
This commit is contained in:
50
backend/tests/helpers/collaboration-db.ts
Normal file
50
backend/tests/helpers/collaboration-db.ts
Normal file
@@ -0,0 +1,50 @@
|
||||
import { randomUUID } from "node:crypto"
|
||||
import { readFile } from "node:fs/promises"
|
||||
import { Pool } from "pg"
|
||||
import { drizzle } from "drizzle-orm/node-postgres"
|
||||
import { getTableColumns, getTableName } from "drizzle-orm"
|
||||
import * as schema from "../../db/schema"
|
||||
|
||||
export const testDatabaseUrl = process.env.COLLABORATION_TEST_DATABASE_URL
|
||||
|
||||
/** Always creates and drops a private schema in the explicitly designated disposable test database. */
|
||||
export async function collaborationDatabase(migrations: string[] = []) {
|
||||
if (!testDatabaseUrl || new URL(testDatabaseUrl).pathname !== "/collaboration_test") {
|
||||
throw new Error("Tests require COLLABORATION_TEST_DATABASE_URL ending in /collaboration_test")
|
||||
}
|
||||
const namespace = `collaboration_test_${randomUUID().replaceAll("-", "")}`
|
||||
const admin = new Pool({ connectionString: testDatabaseUrl, max: 1 })
|
||||
await admin.query(`CREATE SCHEMA "${namespace}"`)
|
||||
const pool = new Pool({ connectionString: testDatabaseUrl, options: `-c search_path=${namespace}`, max: 6 })
|
||||
const db = drizzle(pool, { schema })
|
||||
try {
|
||||
await pool.query("CREATE TYPE locked_tenant AS ENUM ('none', 'maintenance_tenant')")
|
||||
for (const table of [schema.authUsers, schema.authTenantUsers, schema.authProfiles, schema.authRoles,
|
||||
schema.authUserRoles, schema.authRolePermissions, schema.projects, schema.communicationRooms]) {
|
||||
const columns = Object.values(getTableColumns(table)).map((column: any) => {
|
||||
let suffix = column.primary ? " PRIMARY KEY" : ""
|
||||
if (column.name === "id" && column.getSQLType() === "uuid") suffix += " DEFAULT gen_random_uuid()"
|
||||
if (column.columnType === "PgTimestamp") suffix += " DEFAULT now()"
|
||||
if (column.getSQLType() === "boolean") suffix += " DEFAULT false"
|
||||
return `"${column.name}" ${column.getSQLType()}${suffix}`
|
||||
})
|
||||
await pool.query(`CREATE TABLE "${getTableName(table)}" (${columns.join(",")})`)
|
||||
}
|
||||
await pool.query(`CREATE TABLE tenants (id bigint PRIMARY KEY, name text, locked text)`)
|
||||
await pool.query(`CREATE UNIQUE INDEX ON communication_rooms (tenant_id, key)`)
|
||||
await pool.query(await readFile(new URL("../../db/migrations/0070_native_communication_chat.sql", import.meta.url), "utf8"))
|
||||
for (const name of migrations) {
|
||||
await pool.query(await readFile(new URL(`../../db/migrations/${name}`, import.meta.url), "utf8"))
|
||||
}
|
||||
return { db, pool, close: async () => {
|
||||
await pool.end()
|
||||
await admin.query(`DROP SCHEMA "${namespace}" CASCADE`)
|
||||
await admin.end()
|
||||
} }
|
||||
} catch (error) {
|
||||
await pool.end()
|
||||
await admin.query(`DROP SCHEMA "${namespace}" CASCADE`)
|
||||
await admin.end()
|
||||
throw error
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user