import { describe, it, expect, vi } from 'vitest';
import { UpdateDriverProfileUseCase } from './UpdateDriverProfileUseCase';
import type { IDriverRepository } from '../../domain/repositories/IDriverRepository';
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 = {
      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;

    const useCase = new UpdateDriverProfileUseCase(mockDriverRepository);

    const input = {
      driverId: 'driver-1',
      bio: 'New bio',
      country: 'US',
    };

    const result = await useCase.execute(input);

    expect(result.isOk()).toBe(true);
    expect(result.unwrap()).toEqual(mockUpdatedDriver);
    expect(mockDriverRepository.findById).toHaveBeenCalledWith('driver-1');
    expect(mockDriver.update).toHaveBeenCalledWith({ bio: 'New bio', country: 'US' });
    expect(mockDriverRepository.update).toHaveBeenCalledWith({});
  });

  it('returns error when driver not found', async () => {
    const mockDriverRepository = {
      findById: vi.fn().mockResolvedValue(null),
    } as unknown as IDriverRepository;

    const useCase = new UpdateDriverProfileUseCase(mockDriverRepository);

    const input = {
      driverId: 'driver-1',
      bio: 'New bio',
    };

    const result = await useCase.execute(input);

    expect(result.isErr()).toBe(true);
    expect(result.unwrapErr()).toEqual({ code: 'DRIVER_NOT_FOUND' });
  });

  it('updates only provided fields', async () => {
    const mockDriver = {
      id: 'driver-1',
      update: vi.fn().mockReturnValue({}),
    } as unknown as Driver;

    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;

    const useCase = new UpdateDriverProfileUseCase(mockDriverRepository);

    const input = {
      driverId: 'driver-1',
      country: 'US',
    };

    const result = await useCase.execute(input);

    expect(result.isOk()).toBe(true);
    expect(mockDriver.update).toHaveBeenCalledWith({ country: 'US' });
  });
});