/** * Task 2 — ensureSystemRoots (TDD). * Task 3 — ensureEntityFolder (TDD). * * Fixture convention: makePort from helpers/factories (async DB insert); * TEST_USER_ID resolved once via beforeAll from a seeded user — same pattern * as document-folders-crud.test.ts and alerts-tenant-isolation.test.ts. */ import { describe, it, expect, beforeAll, beforeEach } from 'vitest'; import { and, eq } from 'drizzle-orm'; import { db } from '@/lib/db'; import { documentFolders } from '@/lib/db/schema/documents'; import { clients } from '@/lib/db/schema/clients'; import { user } from '@/lib/db/schema/users'; import { ensureSystemRoots, ensureEntityFolder } from '@/lib/services/document-folders.service'; import { makePort } from '../helpers/factories'; let TEST_USER_ID = ''; beforeAll(async () => { const [u] = await db.select({ id: user.id }).from(user).limit(1); if (!u) throw new Error('No user available; run pnpm db:seed first'); TEST_USER_ID = u.id; }); describe('document-folders service · ensureSystemRoots', () => { let portId: string; beforeEach(async () => { const port = await makePort(); portId = port.id; await db.delete(documentFolders).where(eq(documentFolders.portId, portId)); }); it('creates Clients, Companies, and Yachts root folders with system_managed=true', async () => { await ensureSystemRoots(portId, TEST_USER_ID); const rows = await db .select() .from(documentFolders) .where(and(eq(documentFolders.portId, portId), eq(documentFolders.entityType, 'root'))); expect(rows.map((r) => r.name).sort()).toEqual(['Clients', 'Companies', 'Yachts']); for (const r of rows) { expect(r.systemManaged).toBe(true); expect(r.parentId).toBeNull(); expect(r.entityId).toBeNull(); } }); it('is idempotent — second call does not create duplicates', async () => { await ensureSystemRoots(portId, TEST_USER_ID); await ensureSystemRoots(portId, TEST_USER_ID); const rows = await db .select() .from(documentFolders) .where(and(eq(documentFolders.portId, portId), eq(documentFolders.entityType, 'root'))); expect(rows).toHaveLength(3); }); it('returns the three root rows in a stable order (Clients, Companies, Yachts)', async () => { const roots = await ensureSystemRoots(portId, TEST_USER_ID); expect(roots.map((r) => r.name)).toEqual(['Clients', 'Companies', 'Yachts']); }); }); describe('document-folders service · ensureEntityFolder', () => { let portId: string; let clientId: string; let rootId: string; beforeEach(async () => { const port = await makePort(); portId = port.id; await db.delete(documentFolders).where(eq(documentFolders.portId, portId)); const roots = await ensureSystemRoots(portId, TEST_USER_ID); rootId = roots.find((r) => r.name === 'Clients')!.id; const [client] = await db .insert(clients) .values({ portId, fullName: `Smith, John ${crypto.randomUUID().slice(0, 8)}`, }) .returning(); clientId = client!.id; }); it('creates a subfolder under the matching system root with system_managed=true', async () => { const folder = await ensureEntityFolder(portId, 'client', clientId, TEST_USER_ID); expect(folder.systemManaged).toBe(true); expect(folder.entityType).toBe('client'); expect(folder.entityId).toBe(clientId); expect(folder.parentId).toBe(rootId); // name is the client's fullName verbatim const [row] = await db.select().from(clients).where(eq(clients.id, clientId)); expect(folder.name).toBe(row!.fullName); }); it('is idempotent — returns the same row on second call', async () => { const a = await ensureEntityFolder(portId, 'client', clientId, TEST_USER_ID); const b = await ensureEntityFolder(portId, 'client', clientId, TEST_USER_ID); expect(a.id).toBe(b.id); const all = await db .select() .from(documentFolders) .where( and(eq(documentFolders.entityType, 'client'), eq(documentFolders.entityId, clientId)), ); expect(all).toHaveLength(1); }); it('appends a numeric suffix on name collision with an existing folder', async () => { // Insert a second client with the exact same fullName as the first. const [firstClient] = await db .select() .from(clients) .where(eq(clients.id, clientId)); const sharedName = firstClient!.fullName; const [collidingClient] = await db .insert(clients) .values({ portId, fullName: sharedName, }) .returning(); await ensureEntityFolder(portId, 'client', clientId, TEST_USER_ID); const second = await ensureEntityFolder(portId, 'client', collidingClient!.id, TEST_USER_ID); expect(second.name).toBe(`${sharedName} (2)`); }); it('rejects unknown entity types', async () => { await expect( // @ts-expect-error -- runtime check ensureEntityFolder(portId, 'boat', clientId, TEST_USER_ID), ).rejects.toThrow(/entity type/i); }); });