import { describe, it, expect, beforeEach, vi, Mock } from 'vitest';
import { GetDriverTeamUseCase } from './GetDriverTeamUseCase';
import type { ITeamRepository } from '../../domain/repositories/ITeamRepository';
import type { ITeamMembershipRepository } from '../../domain/repositories/ITeamMembershipRepository';
import type { Logger } from '@core/shared/application';

describe('GetDriverTeamUseCase', () => {
  let mockTeamRepo: { findById: Mock };
  let mockMembershipRepo: { getActiveMembershipForDriver: Mock };
  let mockLogger: Logger;

  beforeEach(() => {
    mockTeamRepo = { findById: vi.fn() };
    mockMembershipRepo = { getActiveMembershipForDriver: vi.fn() };
    mockLogger = {
      debug: vi.fn(),
      info: vi.fn(),
      warn: vi.fn(),
      error: vi.fn(),
    };
  });

  it('should return driver team data when membership and team exist', async () => {
    const useCase = new GetDriverTeamUseCase(
      mockTeamRepo as unknown as ITeamRepository,
      mockMembershipRepo as unknown as ITeamMembershipRepository,
      mockLogger,
    );

    const driverId = 'driver1';
    const membership = { id: 'membership1', driverId, teamId: 'team1' };
    const team = { id: 'team1', name: 'Team One' };

    mockMembershipRepo.getActiveMembershipForDriver.mockResolvedValue(membership);
    mockTeamRepo.findById.mockResolvedValue(team);

    const result = await useCase.execute({ driverId });

    expect(result.isOk()).toBe(true);
    expect(result.value).toEqual({
      team,
      membership,
      driverId,
    });
  });

  it('should return error when no active membership found', async () => {
    const useCase = new GetDriverTeamUseCase(
      mockTeamRepo as unknown as ITeamRepository,
      mockMembershipRepo as unknown as ITeamMembershipRepository,
      mockLogger,
    );

    const driverId = 'driver1';

    mockMembershipRepo.getActiveMembershipForDriver.mockResolvedValue(null);

    const result = await useCase.execute({ driverId });

    expect(result.isErr()).toBe(true);
    expect(result.unwrapErr().message).toBe('No active membership found for driver driver1');
  });

  it('should return error when team not found', async () => {
    const useCase = new GetDriverTeamUseCase(
      mockTeamRepo as unknown as ITeamRepository,
      mockMembershipRepo as unknown as ITeamMembershipRepository,
      mockLogger,
    );

    const driverId = 'driver1';
    const membership = { id: 'membership1', driverId, teamId: 'team1' };

    mockMembershipRepo.getActiveMembershipForDriver.mockResolvedValue(membership);
    mockTeamRepo.findById.mockResolvedValue(null);

    const result = await useCase.execute({ driverId });

    expect(result.isErr()).toBe(true);
    expect(result.unwrapErr().message).toBe('Team not found for teamId team1');
  });

  it('should return error when repository throws', async () => {
    const useCase = new GetDriverTeamUseCase(
      mockTeamRepo as unknown as ITeamRepository,
      mockMembershipRepo as unknown as ITeamMembershipRepository,
      mockLogger,
    );

    const driverId = 'driver1';
    const error = new Error('Repository error');

    mockMembershipRepo.getActiveMembershipForDriver.mockRejectedValue(error);

    const result = await useCase.execute({ driverId });

    expect(result.isErr()).toBe(true);
    expect(result.unwrapErr().message).toBe('Repository error');
  });
});