|
| 1 | +// @vitest-environment node |
| 2 | + |
| 3 | +import { PGlite } from '@electric-sql/pglite'; |
| 4 | +import { Prisma } from '@sourcebot/db'; |
| 5 | +import { afterAll, beforeAll, beforeEach, describe, expect, test, vi } from 'vitest'; |
| 6 | + |
| 7 | +const mocks = vi.hoisted(() => ({ |
| 8 | + queryRaw: vi.fn(), |
| 9 | + findFirst: vi.fn(), |
| 10 | +})); |
| 11 | + |
| 12 | +vi.mock('@/middleware/sew', () => ({ |
| 13 | + sew: (callback: () => unknown) => callback(), |
| 14 | +})); |
| 15 | +vi.mock('@/middleware/withAuth', () => ({ |
| 16 | + withAuth: (callback: (context: unknown) => unknown) => callback({ |
| 17 | + org: { id: 1 }, |
| 18 | + role: 'OWNER', |
| 19 | + prisma: { |
| 20 | + $queryRaw: mocks.queryRaw, |
| 21 | + audit: { findFirst: mocks.findFirst }, |
| 22 | + }, |
| 23 | + }), |
| 24 | +})); |
| 25 | +vi.mock('@/middleware/withMinimumOrgRole', () => ({ |
| 26 | + withMinimumOrgRole: ( |
| 27 | + _role: unknown, |
| 28 | + _minimumRole: unknown, |
| 29 | + callback: () => unknown, |
| 30 | + ) => callback(), |
| 31 | +})); |
| 32 | +vi.mock('@/lib/entitlements', () => ({ |
| 33 | + hasEntitlement: vi.fn().mockResolvedValue(true), |
| 34 | +})); |
| 35 | +vi.mock('@sourcebot/shared', () => ({ |
| 36 | + env: { SOURCEBOT_EE_AUDIT_RETENTION_DAYS: 180 }, |
| 37 | +})); |
| 38 | + |
| 39 | +const { getAnalytics } = await import('./actions'); |
| 40 | + |
| 41 | +const database = new PGlite(); |
| 42 | + |
| 43 | +beforeAll(async () => { |
| 44 | + await database.exec(` |
| 45 | + CREATE TABLE "Audit" ( |
| 46 | + "timestamp" timestamptz NOT NULL DEFAULT CURRENT_TIMESTAMP, |
| 47 | + action text NOT NULL, |
| 48 | + "actorId" text NOT NULL, |
| 49 | + metadata jsonb, |
| 50 | + "orgId" integer NOT NULL |
| 51 | + ); |
| 52 | + `); |
| 53 | +}); |
| 54 | + |
| 55 | +afterAll(async () => { |
| 56 | + await database.close(); |
| 57 | +}); |
| 58 | + |
| 59 | +beforeEach(async () => { |
| 60 | + vi.clearAllMocks(); |
| 61 | + await database.exec(` |
| 62 | + TRUNCATE TABLE "Audit"; |
| 63 | + INSERT INTO "Audit" (action, "actorId", metadata, "orgId") VALUES |
| 64 | + ('user.performed_code_search', 'canonical-user', '{"source":"sourcebot-mcp-server"}', 1), |
| 65 | + ('user.fetched_file_source', 'canonical-user', '{"source":"sourcebot-mcp-server"}', 1), |
| 66 | + ('user.fetched_file_tree', 'legacy-user', '{"source":"mcp"}', 1), |
| 67 | + ('user.performed_code_search', 'api-user', '{}', 1), |
| 68 | + ('user.performed_code_search', 'web-user', '{"source":"sourcebot-web-client"}', 1); |
| 69 | + `); |
| 70 | + mocks.queryRaw.mockImplementation(async (queryParts: TemplateStringsArray, ...parameters: unknown[]) => { |
| 71 | + const query = Prisma.sql(queryParts, ...parameters as never[]); |
| 72 | + const result = await database.query(query.text, query.values as never[]); |
| 73 | + return result.rows; |
| 74 | + }); |
| 75 | + mocks.findFirst.mockResolvedValue(null); |
| 76 | +}); |
| 77 | + |
| 78 | +describe('getAnalytics', () => { |
| 79 | + test('classifies canonical and legacy MCP audit sources as MCP activity', async () => { |
| 80 | + const result = await getAnalytics(); |
| 81 | + |
| 82 | + if ('statusCode' in result) { |
| 83 | + throw new Error(result.message); |
| 84 | + } |
| 85 | + |
| 86 | + const daily = result.rows.find((row) => row.period === 'day'); |
| 87 | + expect(daily).toMatchObject({ |
| 88 | + active_users: 4, |
| 89 | + web_active_users: 1, |
| 90 | + non_web_active_users: 3, |
| 91 | + mcp_requests: 3, |
| 92 | + mcp_active_users: 2, |
| 93 | + api_requests: 1, |
| 94 | + api_active_users: 1, |
| 95 | + }); |
| 96 | + }); |
| 97 | +}); |
0 commit comments