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

describe('GetTeamMembershipUseCase', () => {
  const mockGetMembership = vi.fn();
  const mockMembershipRepo: ITeamMembershipRepository = {
    getMembership: mockGetMembership,
    getActiveMembershipForDriver: vi.fn(),
    getTeamMembers: vi.fn(),
    saveMembership: vi.fn(),
    removeMembership: vi.fn(),
    getJoinRequests: vi.fn(),
    countByTeamId: vi.fn(),
    saveJoinRequest: vi.fn(),
    removeJoinRequest: vi.fn(),
  };

  const mockLogger: Logger = {
    debug: vi.fn(),
    info: vi.fn(),
    warn: vi.fn(),
    error: vi.fn(),
  };

  beforeEach(() => {
    vi.clearAllMocks();
  });

  it('should return membership data when membership exists', async () => {
    const useCase = new GetTeamMembershipUseCase(
      mockMembershipRepo,
      mockLogger,
    );

    const teamId = 'team1';
    const driverId = 'driver1';
    const membership = {
      teamId,
      driverId,
      role: 'manager' as const,
      status: 'active' as const,
      joinedAt: new Date('2023-01-01'),
    };

    mockGetMembership.mockResolvedValue(membership);

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

    expect(result.isOk()).toBe(true);
    expect(result.value).toEqual({
      role: 'manager',
      joinedAt: '2023-01-01T00:00:00.000Z',
      isActive: true,
    });
  });

  it('should return null when no membership exists', async () => {
    const useCase = new GetTeamMembershipUseCase(
      mockMembershipRepo,
      mockLogger,
    );

    const teamId = 'team1';
    const driverId = 'driver1';

    mockGetMembership.mockResolvedValue(null);

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

    expect(result.isOk()).toBe(true);
    expect(result.value).toBe(null);
  });

  it('should map driver role to member', async () => {
    const useCase = new GetTeamMembershipUseCase(
      mockMembershipRepo,
      mockLogger,
    );

    const teamId = 'team1';
    const driverId = 'driver1';
    const membership = {
      teamId,
      driverId,
      role: 'driver' as const,
      status: 'active' as const,
      joinedAt: new Date('2023-01-01'),
    };

    mockGetMembership.mockResolvedValue(membership);

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

    expect(result.isOk()).toBe(true);
    expect(result.value?.role).toBe('member');
  });

  it('should return error when repository throws', async () => {
    const useCase = new GetTeamMembershipUseCase(
      mockMembershipRepo,
      mockLogger,
    );

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

    mockGetMembership.mockRejectedValue(error);

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

    expect(result.isErr()).toBe(true);
    expect(result.unwrapErr().code).toBe('REPOSITORY_ERROR');
    expect(result.unwrapErr().details.message).toBe('Failed to retrieve team membership');
  });
});