view data fixes

apps/website/lib/page-queries/ProfilePageQuery.test.ts

@@ -0,0 +1,100 @@
+import { describe, it, expect, vi, beforeEach } from 'vitest';
+import { ProfilePageQuery } from './ProfilePageQuery';
+import { SessionGateway } from '@/lib/gateways/SessionGateway';
+import { DriverProfileService } from '@/lib/services/drivers/DriverProfileService';
+import { ProfileViewDataBuilder } from '@/lib/builders/view-data/ProfileViewDataBuilder';
+import { Result } from '@/lib/contracts/Result';
+
+// Mock dependencies
+vi.mock('@/lib/gateways/SessionGateway', () => ({
+  SessionGateway: vi.fn().mockImplementation(function() {
+    return { getSession: vi.fn() };
+  }),
+}));
+
+vi.mock('@/lib/services/drivers/DriverProfileService', () => ({
+  DriverProfileService: vi.fn().mockImplementation(function() {
+    return { getDriverProfile: vi.fn() };
+  }),
+}));
+
+vi.mock('@/lib/builders/view-data/ProfileViewDataBuilder', () => ({
+  ProfileViewDataBuilder: {
+    build: vi.fn(),
+  },
+}));
+
+describe('ProfilePageQuery', () => {
+  let query: ProfilePageQuery;
+  let mockSessionGateway: any;
+  let mockService: any;
+
+  beforeEach(() => {
+    vi.clearAllMocks();
+    query = new ProfilePageQuery();
+
+    mockSessionGateway = {
+      getSession: vi.fn(),
+    };
+    (SessionGateway as any).mockImplementation(function() { return mockSessionGateway; });
+
+    mockService = {
+      getDriverProfile: vi.fn(),
+    };
+    (DriverProfileService as any).mockImplementation(function() { return mockService; });
+  });
+
+  it('should return notFound if no session exists', async () => {
+    mockSessionGateway.getSession.mockResolvedValue(null);
+
+    const result = await query.execute();
+
+    expect(result.isErr()).toBe(true);
+    expect(result.getError()).toBe('notFound');
+  });
+
+  it('should return notFound if session has no primaryDriverId', async () => {
+    mockSessionGateway.getSession.mockResolvedValue({ user: {} });
+
+    const result = await query.execute();
+
+    expect(result.isErr()).toBe(true);
+    expect(result.getError()).toBe('notFound');
+  });
+
+  it('should return view data when service succeeds', async () => {
+    const driverId = 'driver-123';
+    const apiDto = { id: driverId, name: 'Test Driver' };
+    const viewData = { id: driverId, name: 'Test Driver' };
+
+    mockSessionGateway.getSession.mockResolvedValue({ user: { primaryDriverId: driverId } });
+    mockService.getDriverProfile.mockResolvedValue(Result.ok(apiDto));
+    (ProfileViewDataBuilder.build as any).mockReturnValue(viewData);
+
+    const result = await query.execute();
+
+    expect(result.isOk()).toBe(true);
+    expect(result.unwrap()).toEqual(viewData);
+    expect(mockService.getDriverProfile).toHaveBeenCalledWith(driverId);
+    expect(ProfileViewDataBuilder.build).toHaveBeenCalledWith(apiDto);
+  });
+
+  it('should map service errors correctly', async () => {
+    const driverId = 'driver-123';
+    mockSessionGateway.getSession.mockResolvedValue({ user: { primaryDriverId: driverId } });
+
+    const errorMappings = [
+      { service: 'notFound', expected: 'notFound' },
+      { service: 'unauthorized', expected: 'unauthorized' },
+      { service: 'serverError', expected: 'serverError' },
+      { service: 'other', expected: 'unknown' },
+    ];
+
+    for (const mapping of errorMappings) {
+      mockService.getDriverProfile.mockResolvedValue(Result.err(mapping.service));
+      const result = await query.execute();
+      expect(result.isErr()).toBe(true);
+      expect(result.getError()).toBe(mapping.expected);
+    }
+  });
+});