import { describe, it, expect, vi, beforeEach } from 'vitest';
import { DriversPageQuery } from './DriversPageQuery';
import { DriversPageService } from '@/lib/services/drivers/DriversPageService';
import { Result } from '@/lib/contracts/Result';
import { DriversViewDataBuilder } from '@/lib/builders/view-data/DriversViewDataBuilder';

// Mock dependencies
vi.mock('@/lib/services/drivers/DriversPageService', () => ({
  DriversPageService: vi.fn().mockImplementation(function (this: any) {
    this.getLeaderboard = vi.fn();
  }),
}));

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

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

  beforeEach(() => {
    vi.clearAllMocks();
    mockServiceInstance = {
      getLeaderboard: vi.fn(),
    };
    (DriversPageService as any).mockImplementation(function (this: any) {
      return mockServiceInstance;
    });
  });

  describe('execute', () => {
    it('should return view data when service succeeds', async () => {
      const apiDto = { some: 'drivers-data' };
      const viewData = { transformed: 'drivers-view' } as any;

      mockServiceInstance.getLeaderboard.mockResolvedValue(Result.ok(apiDto));
      (DriversViewDataBuilder.build as any).mockReturnValue(viewData);

      const result = await DriversPageQuery.execute();

      expect(result.isOk()).toBe(true);
      expect(result.unwrap()).toEqual(viewData);
      expect(DriversPageService).toHaveBeenCalled();
      expect(mockServiceInstance.getLeaderboard).toHaveBeenCalled();
      expect(DriversViewDataBuilder.build).toHaveBeenCalledWith(apiDto);
    });

    it('should return NotFound when service returns notFound error', async () => {
      mockServiceInstance.getLeaderboard.mockResolvedValue(Result.err('notFound'));

      const result = await DriversPageQuery.execute();

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

    it('should return Error when service returns other error', async () => {
      mockServiceInstance.getLeaderboard.mockResolvedValue(Result.err('some-other-error'));

      const result = await DriversPageQuery.execute();

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

    it('should return Error on exception', async () => {
      mockServiceInstance.getLeaderboard.mockRejectedValue(new Error('Unexpected error'));

      const result = await DriversPageQuery.execute();

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