refactor

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

@@ -1,29 +1,49 @@
-import { describe, it, expect, beforeEach, vi, Mock } from 'vitest';
+import { describe, it, expect, beforeEach, vi } from 'vitest';
 import { GetAllTeamsUseCase } from './GetAllTeamsUseCase';
 import type { ITeamRepository } from '../../domain/repositories/ITeamRepository';
 import type { ITeamMembershipRepository } from '../../domain/repositories/ITeamMembershipRepository';
 import type { Logger } from '@core/shared/application';
 
 describe('GetAllTeamsUseCase', () => {
-  let mockTeamRepo: { findAll: Mock };
-  let mockTeamMembershipRepo: { countByTeamId: Mock };
-  let mockLogger: Logger;
+  const mockTeamFindAll = vi.fn();
+  const mockTeamRepo: ITeamRepository = {
+    findById: vi.fn(),
+    findAll: mockTeamFindAll,
+    findByLeagueId: vi.fn(),
+    create: vi.fn(),
+    update: vi.fn(),
+    delete: vi.fn(),
+    exists: vi.fn(),
+  };
+
+  const mockTeamMembershipCountByTeamId = vi.fn();
+  const mockTeamMembershipRepo: ITeamMembershipRepository = {
+    getMembership: vi.fn(),
+    getActiveMembershipForDriver: vi.fn(),
+    getTeamMembers: vi.fn(),
+    saveMembership: vi.fn(),
+    removeMembership: vi.fn(),
+    countByTeamId: mockTeamMembershipCountByTeamId,
+    getJoinRequests: vi.fn(),
+    saveJoinRequest: vi.fn(),
+    removeJoinRequest: vi.fn(),
+  };
+
+  const mockLogger: Logger = {
+    debug: vi.fn(),
+    info: vi.fn(),
+    warn: vi.fn(),
+    error: vi.fn(),
+  };
 
   beforeEach(() => {
-    mockTeamRepo = { findAll: vi.fn() };
-    mockTeamMembershipRepo = { countByTeamId: vi.fn() };
-    mockLogger = {
-      debug: vi.fn(),
-      info: vi.fn(),
-      warn: vi.fn(),
-      error: vi.fn(),
-    };
+    vi.clearAllMocks();
   });
 
   it('should return teams data', async () => {
     const useCase = new GetAllTeamsUseCase(
-      mockTeamRepo as unknown as ITeamRepository,
-      mockTeamMembershipRepo as unknown as ITeamMembershipRepository,
+      mockTeamRepo,
+      mockTeamMembershipRepo,
       mockLogger,
     );
 
@@ -46,8 +66,8 @@ describe('GetAllTeamsUseCase', () => {
       createdAt: new Date('2023-01-02T00:00:00Z'),
     };
 
-    mockTeamRepo.findAll.mockResolvedValue([team1, team2]);
-    mockTeamMembershipRepo.countByTeamId.mockImplementation((id: string) => Promise.resolve(id === 'team1' ? 5 : 3));
+    mockTeamFindAll.mockResolvedValue([team1, team2]);
+    mockTeamMembershipCountByTeamId.mockImplementation((id: string) => Promise.resolve(id === 'team1' ? 5 : 3));
 
     const result = await useCase.execute();
 
@@ -80,12 +100,12 @@ describe('GetAllTeamsUseCase', () => {
 
   it('should return empty result when no teams', async () => {
     const useCase = new GetAllTeamsUseCase(
-      mockTeamRepo as unknown as ITeamRepository,
-      mockTeamMembershipRepo as unknown as ITeamMembershipRepository,
+      mockTeamRepo,
+      mockTeamMembershipRepo,
       mockLogger,
     );
 
-    mockTeamRepo.findAll.mockResolvedValue([]);
+    mockTeamFindAll.mockResolvedValue([]);
 
     const result = await useCase.execute();
 
@@ -97,17 +117,18 @@ describe('GetAllTeamsUseCase', () => {
 
   it('should return error when repository throws', async () => {
     const useCase = new GetAllTeamsUseCase(
-      mockTeamRepo as unknown as ITeamRepository,
-      mockTeamMembershipRepo as unknown as ITeamMembershipRepository,
+      mockTeamRepo,
+      mockTeamMembershipRepo,
       mockLogger,
     );
 
     const error = new Error('Repository error');
-    mockTeamRepo.findAll.mockRejectedValue(error);
+    mockTeamFindAll.mockRejectedValue(error);
 
     const result = await useCase.execute();
 
     expect(result.isErr()).toBe(true);
-    expect(result.unwrapErr().message).toBe('Repository error');
+    expect(result.unwrapErr().code).toBe('REPOSITORY_ERROR');
+    expect(result.unwrapErr().details.message).toBe('Repository error');
   });
 });