view models

apps/website/lib/view-models/CompleteOnboardingViewModel.test.ts

@@ -0,0 +1,56 @@
+import { describe, it, expect } from 'vitest';
+import { CompleteOnboardingViewModel } from './CompleteOnboardingViewModel';
+import type { CompleteOnboardingOutputDTO } from '../types/generated/CompleteOnboardingOutputDTO';
+
+describe('CompleteOnboardingViewModel', () => {
+  it('should create instance with success and driverId', () => {
+    const dto: CompleteOnboardingOutputDTO & { driverId: string } = {
+      success: true,
+      driverId: 'driver-123',
+    };
+
+    const viewModel = new CompleteOnboardingViewModel(dto);
+
+    expect(viewModel.success).toBe(true);
+    expect(viewModel.driverId).toBe('driver-123');
+  });
+
+  it('should return true for isSuccessful when success is true', () => {
+    const dto: CompleteOnboardingOutputDTO & { driverId: string } = {
+      success: true,
+      driverId: 'driver-123',
+    };
+
+    const viewModel = new CompleteOnboardingViewModel(dto);
+
+    expect(viewModel.isSuccessful).toBe(true);
+  });
+
+  it('should return false for isSuccessful when success is false', () => {
+    const dto: CompleteOnboardingOutputDTO & { driverId: string } = {
+      success: false,
+      driverId: 'driver-123',
+    };
+
+    const viewModel = new CompleteOnboardingViewModel(dto);
+
+    expect(viewModel.isSuccessful).toBe(false);
+  });
+
+  it('should preserve driverId regardless of success status', () => {
+    const successDto: CompleteOnboardingOutputDTO & { driverId: string } = {
+      success: true,
+      driverId: 'driver-success',
+    };
+    const failDto: CompleteOnboardingOutputDTO & { driverId: string } = {
+      success: false,
+      driverId: 'driver-fail',
+    };
+
+    const successViewModel = new CompleteOnboardingViewModel(successDto);
+    const failViewModel = new CompleteOnboardingViewModel(failDto);
+
+    expect(successViewModel.driverId).toBe('driver-success');
+    expect(failViewModel.driverId).toBe('driver-fail');
+  });
+});