gridpilot.gg/apps/api/src/domain/driver/DriverController.test.ts

import { Test, TestingModule } from '@nestjs/testing';
import { vi } from 'vitest';
import { DriverController } from './DriverController';
import { DriverService } from './DriverService';
import type { Request } from 'express';

interface AuthenticatedRequest extends Request {
  user?: { userId: string };
}
import { CompleteOnboardingInputDTO } from './dtos/CompleteOnboardingInputDTO';
import { CompleteOnboardingOutputDTO } from './dtos/CompleteOnboardingOutputDTO';
import { DriversLeaderboardDTO } from './dtos/DriversLeaderboardDTO';
import { DriverStatsDTO } from './dtos/DriverStatsDTO';
import { GetDriverOutputDTO } from './dtos/GetDriverOutputDTO';
import { GetDriverProfileOutputDTO } from './dtos/GetDriverProfileOutputDTO';
import { DriverRegistrationStatusDTO } from './dtos/DriverRegistrationStatusDTO';

describe('DriverController', () => {
  let controller: DriverController;
  let service: ReturnType<typeof vi.mocked<DriverService>>;

  beforeEach(async () => {
    const module: TestingModule = await Test.createTestingModule({
      controllers: [DriverController],
      providers: [
        {
          provide: DriverService,
          useValue: {
            getDriversLeaderboard: vi.fn(),
            getTotalDrivers: vi.fn(),
            getCurrentDriver: vi.fn(),
            completeOnboarding: vi.fn(),
            getDriverRegistrationStatus: vi.fn(),
            getDriver: vi.fn(),
            getDriverProfile: vi.fn(),
            updateDriverProfile: vi.fn(),
          },
        },
      ],
    }).compile();

    controller = module.get<DriverController>(DriverController);
    service = vi.mocked(module.get(DriverService));
  });

  describe('getDriversLeaderboard', () => {
    it('should return drivers leaderboard', async () => {
      const leaderboard: DriversLeaderboardDTO = { items: [] };
      service.getDriversLeaderboard.mockResolvedValue({ viewModel: leaderboard } as never);

      const result = await controller.getDriversLeaderboard();

      expect(service.getDriversLeaderboard).toHaveBeenCalled();
      expect(result).toEqual(leaderboard);
    });
  });

  describe('getTotalDrivers', () => {
    it('should return total drivers stats', async () => {
      const stats: DriverStatsDTO = { totalDrivers: 100 };
      service.getTotalDrivers.mockResolvedValue({ viewModel: stats } as never);

      const result = await controller.getTotalDrivers();

      expect(service.getTotalDrivers).toHaveBeenCalled();
      expect(result).toEqual(stats);
    });
  });

  describe('getCurrentDriver', () => {
    it('should return current driver if userId exists', async () => {
      const userId = 'user-123';
      const driver: GetDriverOutputDTO = { id: 'driver-123', name: 'Driver' } as GetDriverOutputDTO;
      service.getCurrentDriver.mockResolvedValue({ viewModel: driver } as never);

      const mockReq: Partial<AuthenticatedRequest> = { user: { userId } };

      const result = await controller.getCurrentDriver(mockReq as AuthenticatedRequest);

      expect(service.getCurrentDriver).toHaveBeenCalledWith(userId);
      expect(result).toEqual(driver);
    });

    it('should return null if no userId', async () => {
      const mockReq: Partial<AuthenticatedRequest> = {};

      const result = await controller.getCurrentDriver(mockReq as AuthenticatedRequest);

      expect(service.getCurrentDriver).not.toHaveBeenCalled();
      expect(result).toBeNull();
    });
  });

  describe('completeOnboarding', () => {
    it('should complete onboarding', async () => {
      const userId = 'user-123';
      const input: CompleteOnboardingInputDTO = { someField: 'value' } as CompleteOnboardingInputDTO;
      const output: CompleteOnboardingOutputDTO = { success: true };
      service.completeOnboarding.mockResolvedValue({ viewModel: output } as never);

      const mockReq: Partial<AuthenticatedRequest> = { user: { userId } };

      const result = await controller.completeOnboarding(input, mockReq as AuthenticatedRequest);

      expect(service.completeOnboarding).toHaveBeenCalledWith(userId, input);
      expect(result).toEqual(output);
    });
  });

  describe('getDriverRegistrationStatus', () => {
    it('should return registration status', async () => {
      const driverId = 'driver-123';
      const raceId = 'race-456';
      const status: DriverRegistrationStatusDTO = { registered: true } as DriverRegistrationStatusDTO;
      service.getDriverRegistrationStatus.mockResolvedValue({ viewModel: status } as never);

      const result = await controller.getDriverRegistrationStatus(driverId, raceId);

      expect(service.getDriverRegistrationStatus).toHaveBeenCalledWith({ driverId, raceId });
      expect(result).toEqual(status);
    });
  });

  describe('getDriver', () => {
    it('should return driver by id', async () => {
      const driverId = 'driver-123';
      const driver: GetDriverOutputDTO = { id: driverId, name: 'Driver' } as GetDriverOutputDTO;
      service.getDriver.mockResolvedValue({ viewModel: driver } as never);

      const result = await controller.getDriver(driverId);

      expect(service.getDriver).toHaveBeenCalledWith(driverId);
      expect(result).toEqual(driver);
    });
  });

  describe('getDriverProfile', () => {
    it('should return driver profile', async () => {
      const driverId = 'driver-123';
      const profile: GetDriverProfileOutputDTO = { id: driverId, bio: 'Bio' };
      service.getDriverProfile.mockResolvedValue(profile);

      const result = await controller.getDriverProfile(driverId);

      expect(service.getDriverProfile).toHaveBeenCalledWith(driverId);
      expect(result).toEqual(profile);
    });
  });

  describe('updateDriverProfile', () => {
    it('should update driver profile', async () => {
      const driverId = 'driver-123';
      const body = { bio: 'New bio', country: 'US' };
      const updated: GetDriverOutputDTO = { id: driverId, name: 'Driver' };
      service.updateDriverProfile.mockResolvedValue(updated);

      const result = await controller.updateDriverProfile(driverId, body);

      expect(service.updateDriverProfile).toHaveBeenCalledWith(driverId, body.bio, body.country);
      expect(result).toEqual(updated);
    });
  });
});