import { describe, it, expect } from 'vitest'; import { eq } from 'drizzle-orm'; import { search, buildPrefixTsquery, normalizePhoneQuery } from '@/lib/services/search.service'; import { db } from '@/lib/db'; import { yachts, companies } from '@/lib/db/schema'; import { makePort, makeClient, makeYacht, makeCompany } from '../../helpers/factories'; // Default opts — super admin so every bucket runs without per-resource // permission gating getting in the way of the assertions. const ADMIN_OPTS = { permissions: null, isSuperAdmin: true } as const; // ─── Yachts ────────────────────────────────────────────────────────────────── describe('search.service — yachts', () => { it('matches yachts by name (case-insensitive)', async () => { const port = await makePort(); const owner = await makeClient({ portId: port.id }); await makeYacht({ portId: port.id, ownerType: 'client', ownerId: owner.id, name: 'Sea Breeze', }); await makeYacht({ portId: port.id, ownerType: 'client', ownerId: owner.id, name: 'Wind Dancer', }); const results = await search(port.id, 'BREEZE', ADMIN_OPTS); expect(results.yachts.some((y) => y.name === 'Sea Breeze')).toBe(true); expect(results.yachts.some((y) => y.name === 'Wind Dancer')).toBe(false); }); it('matches yachts by hull number', async () => { const port = await makePort(); const owner = await makeClient({ portId: port.id }); await makeYacht({ portId: port.id, ownerType: 'client', ownerId: owner.id, name: 'Nomad', hullNumber: 'HULL-XYZ-999', }); const results = await search(port.id, 'hull-xyz', ADMIN_OPTS); expect(results.yachts.some((y) => y.hullNumber === 'HULL-XYZ-999')).toBe(true); }); it('matches yachts by registration', async () => { const port = await makePort(); const owner = await makeClient({ portId: port.id }); await makeYacht({ portId: port.id, ownerType: 'client', ownerId: owner.id, name: 'Registered One', registration: 'REG-ABC-123', }); const results = await search(port.id, 'reg-abc', ADMIN_OPTS); expect(results.yachts.some((y) => y.registration === 'REG-ABC-123')).toBe(true); }); it('excludes archived yachts', async () => { const port = await makePort(); const owner = await makeClient({ portId: port.id }); const yacht = await makeYacht({ portId: port.id, ownerType: 'client', ownerId: owner.id, name: 'Ghost Ship', }); await db.update(yachts).set({ archivedAt: new Date() }).where(eq(yachts.id, yacht.id)); const results = await search(port.id, 'ghost ship', ADMIN_OPTS); expect(results.yachts.some((y) => y.id === yacht.id)).toBe(false); }); it('is tenant-scoped', async () => { const portA = await makePort(); const portB = await makePort(); const ownerA = await makeClient({ portId: portA.id }); const ownerB = await makeClient({ portId: portB.id }); await makeYacht({ portId: portA.id, ownerType: 'client', ownerId: ownerA.id, name: 'UniqueYachtNameA', }); await makeYacht({ portId: portB.id, ownerType: 'client', ownerId: ownerB.id, name: 'UniqueYachtNameB', }); const resultsA = await search(portA.id, 'UniqueYachtName', ADMIN_OPTS); expect(resultsA.yachts.some((y) => y.name === 'UniqueYachtNameA')).toBe(true); expect(resultsA.yachts.some((y) => y.name === 'UniqueYachtNameB')).toBe(false); }); }); // ─── Companies ─────────────────────────────────────────────────────────────── describe('search.service — companies', () => { it('matches companies by name', async () => { const port = await makePort(); await makeCompany({ portId: port.id, overrides: { name: 'Poseidon Maritime Ltd' } }); await makeCompany({ portId: port.id, overrides: { name: 'Neptune Holdings' } }); const results = await search(port.id, 'poseidon', ADMIN_OPTS); expect(results.companies.some((c) => c.name === 'Poseidon Maritime Ltd')).toBe(true); expect(results.companies.some((c) => c.name === 'Neptune Holdings')).toBe(false); }); it('matches companies by legal name', async () => { const port = await makePort(); await makeCompany({ portId: port.id, overrides: { name: 'AcmeShort', legalName: 'Acme Legal Holdings Inc.' }, }); const results = await search(port.id, 'Acme Legal', ADMIN_OPTS); expect(results.companies.some((c) => c.legalName === 'Acme Legal Holdings Inc.')).toBe(true); }); it('matches companies by tax ID', async () => { const port = await makePort(); await makeCompany({ portId: port.id, overrides: { name: 'TaxyCo', taxId: 'VAT-112233445' }, }); const results = await search(port.id, 'vat-112233', ADMIN_OPTS); expect(results.companies.some((c) => c.taxId === 'VAT-112233445')).toBe(true); }); it('excludes archived companies', async () => { const port = await makePort(); const company = await makeCompany({ portId: port.id, overrides: { name: 'ArchivedCompanyXyz' }, }); await db.update(companies).set({ archivedAt: new Date() }).where(eq(companies.id, company.id)); const results = await search(port.id, 'ArchivedCompanyXyz', ADMIN_OPTS); expect(results.companies.some((c) => c.id === company.id)).toBe(false); }); it('is tenant-scoped', async () => { const portA = await makePort(); const portB = await makePort(); await makeCompany({ portId: portA.id, overrides: { name: 'UniqueCompanyA' } }); await makeCompany({ portId: portB.id, overrides: { name: 'UniqueCompanyB' } }); const resultsA = await search(portA.id, 'UniqueCompany', ADMIN_OPTS); expect(resultsA.companies.some((c) => c.name === 'UniqueCompanyA')).toBe(true); expect(resultsA.companies.some((c) => c.name === 'UniqueCompanyB')).toBe(false); }); }); // ─── Combined ──────────────────────────────────────────────────────────────── describe('search.service — combined', () => { it('returns clients, yachts, and companies for a query that matches multiple', async () => { const port = await makePort(); const client = await makeClient({ portId: port.id, overrides: { fullName: 'Alpha Person' }, }); await makeYacht({ portId: port.id, ownerType: 'client', ownerId: client.id, name: 'Alpha Yacht', }); await makeCompany({ portId: port.id, overrides: { name: 'Alpha Holdings' } }); const results = await search(port.id, 'alpha', ADMIN_OPTS); expect(results.clients.some((c) => c.fullName === 'Alpha Person')).toBe(true); expect(results.yachts.some((y) => y.name === 'Alpha Yacht')).toBe(true); expect(results.companies.some((c) => c.name === 'Alpha Holdings')).toBe(true); }); it('returns every bucket as an empty array when no matches exist', async () => { const port = await makePort(); const results = await search(port.id, 'zzz-nothing-matches-zzz', ADMIN_OPTS); // Shape check (no rows in any bucket) expect(results.clients).toEqual([]); expect(results.residentialClients).toEqual([]); expect(results.yachts).toEqual([]); expect(results.companies).toEqual([]); expect(results.interests).toEqual([]); expect(results.residentialInterests).toEqual([]); expect(results.berths).toEqual([]); expect(results.invoices).toEqual([]); expect(results.expenses).toEqual([]); expect(results.documents).toEqual([]); expect(results.files).toEqual([]); expect(results.reminders).toEqual([]); expect(results.brochures).toEqual([]); expect(results.tags).toEqual([]); expect(results.navigation).toEqual([]); // Totals are present and zero across the board. expect(Object.values(results.totals).every((n) => n === 0)).toBe(true); }); }); // ─── Helper purity tests ───────────────────────────────────────────────────── describe('buildPrefixTsquery', () => { it('builds a tokenized prefix tsquery', () => { expect(buildPrefixTsquery('joh smi')).toBe('joh:* & smi:*'); }); it('strips regex / tsquery meta-characters', () => { expect(buildPrefixTsquery('a&b!c')).toBe('abc:*'); }); it('returns null for empty / whitespace-only input', () => { expect(buildPrefixTsquery('')).toBeNull(); expect(buildPrefixTsquery(' ')).toBeNull(); }); it('returns null when every token reduces to nothing after sanitization', () => { expect(buildPrefixTsquery('!@#$ %^&*')).toBeNull(); }); }); describe('normalizePhoneQuery', () => { it('strips non-digit / non-plus characters', () => { expect(normalizePhoneQuery('+44 7700 900 123')).toBe('+447700900123'); }); it('preserves leading +', () => { expect(normalizePhoneQuery('+1 (555) 123-4567')).toBe('+15551234567'); }); it('returns null when fewer than 3 digits remain (too short to be unique)', () => { expect(normalizePhoneQuery('a-b')).toBeNull(); }); }); // ─── Partial name matching ─────────────────────────────────────────────────── describe('search.service — partial name matching', () => { it('matches "joh smi" against "John Smith" via tokenized prefix tsquery', async () => { const port = await makePort(); await makeClient({ portId: port.id, overrides: { fullName: 'John Smith' } }); await makeClient({ portId: port.id, overrides: { fullName: 'Joanna Smithers' } }); await makeClient({ portId: port.id, overrides: { fullName: 'Bob Jones' } }); const results = await search(port.id, 'joh smi', ADMIN_OPTS); expect(results.clients.some((c) => c.fullName === 'John Smith')).toBe(true); expect(results.clients.some((c) => c.fullName === 'Bob Jones')).toBe(false); }); it('matches a single name fragment ("joh" → "John …")', async () => { const port = await makePort(); await makeClient({ portId: port.id, overrides: { fullName: 'Johnathan Doe' } }); const results = await search(port.id, 'joh', ADMIN_OPTS); expect(results.clients.some((c) => c.fullName === 'Johnathan Doe')).toBe(true); }); }); describe('search.service — bucket totals', () => { it('emits per-bucket totals so the UI can render "show more" links', async () => { const port = await makePort(); await makeClient({ portId: port.id, overrides: { fullName: 'TotalsCheck One' } }); const results = await search(port.id, 'TotalsCheck', ADMIN_OPTS); expect(results.totals.clients).toBeGreaterThanOrEqual(1); expect(typeof results.totals.invoices).toBe('number'); }); });