refactor dtos to ports

core/racing/application/use-cases/UpdateDriverProfileUseCase.test.ts

@@ -1,32 +1,26 @@
 import { describe, it, expect, vi } from 'vitest';
 import { UpdateDriverProfileUseCase } from './UpdateDriverProfileUseCase';
 import type { IDriverRepository } from '../../domain/repositories/IDriverRepository';
-import { EntityMappers } from '../mappers/EntityMappers';
-
-vi.mock('../mappers/EntityMappers', () => ({
-  EntityMappers: {
-    toDriverDTO: vi.fn(),
-  },
-}));
+import type { Driver } from '../../domain/entities/Driver';
 
 describe('UpdateDriverProfileUseCase', () => {
   it('updates driver profile successfully', async () => {
     const mockDriver = {
       id: 'driver-1',
       update: vi.fn().mockReturnValue({}),
-    };
+    } as unknown as Driver;
 
-    const mockUpdatedDriver = {};
-
-    const mockDTO = { id: 'driver-1', bio: 'New bio' };
+    const mockUpdatedDriver = {
+      id: 'driver-1',
+      bio: 'New bio',
+      country: 'US',
+    } as Driver;
 
     const mockDriverRepository = {
       findById: vi.fn().mockResolvedValue(mockDriver),
       update: vi.fn().mockResolvedValue(mockUpdatedDriver),
     } as unknown as IDriverRepository;
 
-    (EntityMappers.toDriverDTO as any).mockReturnValue(mockDTO);
-
     const useCase = new UpdateDriverProfileUseCase(mockDriverRepository);
 
     const input = {
@@ -38,11 +32,10 @@ describe('UpdateDriverProfileUseCase', () => {
     const result = await useCase.execute(input);
 
     expect(result.isOk()).toBe(true);
-    expect(result.unwrap()).toEqual(mockDTO);
+    expect(result.unwrap()).toEqual(mockUpdatedDriver);
     expect(mockDriverRepository.findById).toHaveBeenCalledWith('driver-1');
     expect(mockDriver.update).toHaveBeenCalledWith({ bio: 'New bio', country: 'US' });
     expect(mockDriverRepository.update).toHaveBeenCalledWith({});
-    expect(EntityMappers.toDriverDTO).toHaveBeenCalledWith(mockUpdatedDriver);
   });
 
   it('returns error when driver not found', async () => {
@@ -67,19 +60,18 @@ describe('UpdateDriverProfileUseCase', () => {
     const mockDriver = {
       id: 'driver-1',
       update: vi.fn().mockReturnValue({}),
-    };
+    } as unknown as Driver;
 
-    const mockUpdatedDriver = {};
-
-    const mockDTO = { id: 'driver-1', country: 'US' };
+    const mockUpdatedDriver = {
+      id: 'driver-1',
+      country: 'US',
+    } as Driver;
 
     const mockDriverRepository = {
       findById: vi.fn().mockResolvedValue(mockDriver),
       update: vi.fn().mockResolvedValue(mockUpdatedDriver),
     } as unknown as IDriverRepository;
 
-    (EntityMappers.toDriverDTO as any).mockReturnValue(mockDTO);
-
     const useCase = new UpdateDriverProfileUseCase(mockDriverRepository);
 
     const input = {