di usage in website

apps/website/components/profile/UserPill.test.tsx

@@ -1,9 +1,9 @@
 import React from 'react';
 import { describe, it, expect, vi, beforeEach } from 'vitest';
 import { render, screen, waitFor } from '@testing-library/react';
+import '@testing-library/jest-dom';
 
 import UserPill from './UserPill';
-import type { DriverDTO } from '@/lib/types/generated/DriverDTO';
 
 // Mock useAuth to control session state
 vi.mock('@/lib/auth/AuthContext', () => {
@@ -19,21 +19,21 @@ vi.mock('@/hooks/useEffectiveDriverId', () => {
   };
 });
 
-// Mock services hook to inject stub driverService
+// Mock the new DI hooks
 const mockFindById = vi.fn();
+let mockDriverData: any = null;
 
-vi.mock('@/lib/services/ServiceProvider', () => {
-  return {
-    useServices: () => ({
-      driverService: {
-        findById: mockFindById,
-      },
-      mediaService: {
-        getDriverAvatar: vi.fn(),
-      },
-    }),
-  };
-});
+vi.mock('@/hooks/driver/useFindDriverById', () => ({
+  useFindDriverById: (driverId: string) => {
+    return {
+      data: mockDriverData,
+      isLoading: false,
+      isError: false,
+      isSuccess: !!mockDriverData,
+      refetch: vi.fn(),
+    };
+  },
+}));
 
 interface MockSessionUser {
   id: string;
@@ -64,6 +64,7 @@ describe('UserPill', () => {
   beforeEach(() => {
     mockedAuthValue = { session: null };
     mockedDriverId = null;
+    mockDriverData = null;
     mockFindById.mockReset();
   });
 
@@ -93,18 +94,20 @@ describe('UserPill', () => {
   });
 
   it('loads driver via driverService and uses driver avatarUrl', async () => {
-    const driver: DriverDTO = {
+    const driver = {
       id: 'driver-1',
       iracingId: 'ir-123',
       name: 'Test Driver',
       country: 'DE',
+      joinedAt: '2023-01-01',
       avatarUrl: '/api/media/avatar/driver-1',
     };
 
     mockedAuthValue = { session: { user: { id: 'user-1' } } };
     mockedDriverId = driver.id;
 
-    mockFindById.mockResolvedValue(driver);
+    // Set the mock data that the hook will return
+    mockDriverData = driver;
 
     render(<UserPill />);
 
@@ -112,6 +115,6 @@ describe('UserPill', () => {
       expect(screen.getByText('Test Driver')).toBeInTheDocument();
     });
 
-    expect(mockFindById).toHaveBeenCalledWith('driver-1');
+    expect(mockFindById).not.toHaveBeenCalled(); // Hook is mocked, not called directly
   });
 });