import { describe, it, expect, vi, beforeEach } from 'vitest';
import { RaceStewardingPageQuery } from './RaceStewardingPageQuery';
import { RaceStewardingService } from '@/lib/services/races/RaceStewardingService';
import { Result } from '@/lib/contracts/Result';
import { RaceStewardingViewDataBuilder } from '@/lib/builders/view-data/RaceStewardingViewDataBuilder';
import { mapToPresentationError } from '@/lib/contracts/page-queries/PresentationError';

// Mock dependencies
vi.mock('@/lib/services/races/RaceStewardingService', () => ({
  RaceStewardingService: vi.fn(class {
    getRaceStewarding = vi.fn();
  }),
}));

vi.mock('@/lib/builders/view-data/RaceStewardingViewDataBuilder', () => ({
  RaceStewardingViewDataBuilder: {
    build: vi.fn(),
  },
}));

vi.mock('@/lib/contracts/page-queries/PresentationError', () => ({
  mapToPresentationError: vi.fn(),
}));

describe('RaceStewardingPageQuery', () => {
  let query: RaceStewardingPageQuery;
  let mockServiceInstance: any;

  beforeEach(() => {
    vi.clearAllMocks();
    query = new RaceStewardingPageQuery();
    mockServiceInstance = {
      getRaceStewarding: vi.fn(),
    };
    (RaceStewardingService as any).mockImplementation(function () {
      return mockServiceInstance;
    });
  });

  it('should return view data when service succeeds', async () => {
    const params = { raceId: 'race-123', driverId: 'driver-456' };
    const apiDto = { race: { id: 'race-123' }, stewarding: [{ incident: 'incident-1' }] };
    const viewData = { race: { id: 'race-123' }, stewarding: [{ incident: 'incident-1' }] };

    mockServiceInstance.getRaceStewarding.mockResolvedValue(Result.ok(apiDto));
    (RaceStewardingViewDataBuilder.build as any).mockReturnValue(viewData);

    const result = await query.execute(params);

    expect(result.isOk()).toBe(true);
    expect(result.unwrap()).toEqual(viewData);
    expect(RaceStewardingService).toHaveBeenCalled();
    expect(mockServiceInstance.getRaceStewarding).toHaveBeenCalledWith('race-123');
    expect(RaceStewardingViewDataBuilder.build).toHaveBeenCalledWith(apiDto);
  });

  it('should return mapped presentation error when service fails', async () => {
    const params = { raceId: 'race-123', driverId: 'driver-456' };
    const serviceError = { type: 'notFound' };
    const presentationError = 'notFound';

    mockServiceInstance.getRaceStewarding.mockResolvedValue(Result.err(serviceError));
    (mapToPresentationError as any).mockReturnValue(presentationError);

    const result = await query.execute(params);

    expect(result.isErr()).toBe(true);
    expect(result.getError()).toBe(presentationError);
    expect(mapToPresentationError).toHaveBeenCalledWith(serviceError);
  });

  it('should return serverError on exception', async () => {
    const params = { raceId: 'race-123', driverId: 'driver-456' };

    mockServiceInstance.getRaceStewarding.mockRejectedValue(new Error('Network error'));

    const result = await query.execute(params);

    expect(result.isErr()).toBe(true);
    expect(result.getError()).toBe('Network error');
  });

  it('should provide a static execute method', async () => {
    const params = { raceId: 'race-123', driverId: 'driver-456' };
    const apiDto = { race: { id: 'race-123' }, stewarding: [{ incident: 'incident-1' }] };
    const viewData = { race: { id: 'race-123' }, stewarding: [{ incident: 'incident-1' }] };

    mockServiceInstance.getRaceStewarding.mockResolvedValue(Result.ok(apiDto));
    (RaceStewardingViewDataBuilder.build as any).mockReturnValue(viewData);

    const result = await RaceStewardingPageQuery.execute(params);

    expect(result.isOk()).toBe(true);
    expect(result.unwrap()).toEqual(viewData);
  });
});