import React from 'react';
import { render, waitFor, screen } from '@testing-library/react';
import RevisionDiff from '../RevisionDiff';
import DJClientContext from '../../../providers/djclient';
import { NodePage } from '../Loadable';
import { MemoryRouter, Route, Routes } from 'react-router-dom';

describe('<RevisionDiff />', () => {
  const mockDjClient = {
    DataJunctionAPI: {
      revisions: vi.fn(),
    },
  };

  const mockNodesWithDimension = [
    {
      node_revision_id: 1,
      node_id: 1,
      type: 'dimension',
      name: 'default.repair_order',
      display_name: 'Repair Orders',
      version: 'v1.0',
      status: 'valid',
      mode: 'published',
      catalog: {
        id: 1,
        uuid: '0fc18295-e1a2-4c3c-b72a-894725c12488',
        created_at: '2023-08-21T16:48:51.146121+00:00',
        updated_at: '2023-08-21T16:48:51.146122+00:00',
        extra_params: {},
        name: 'warehouse',
      },
      description: 'Repair order dimension',
      query:
        'SELECT repair_order_id, municipality_id, hard_hat_id, order_date, ' +
        'dispatcher_id FROM default.repair_orders',
      availability: null,
      columns: [
        {
          name: 'repair_order_id',
          type: 'int',
          attributes: [],
          dimension: {
            name: 'default.repair_order',
          },
        },
        {
          name: 'municipality_id',
          type: 'int',
          attributes: [],
          dimension: null,
        },
        {
          name: 'hard_hat_id',
          type: 'int',
          attributes: [],
          dimension: null,
        },
        {
          name: 'order_date',
          type: 'date',
          attributes: [],
          dimension: null,
        },
        {
          name: 'dispatcher_id',
          type: 'int',
          attributes: [],
          dimension: null,
        },
      ],
      updated_at: '2023-08-21T16:48:52.981201+00:00',
      materializations: [],
      parents: [],
    },
    {
      node_revision_id: 2,
      node_id: 2,
      type: 'dimension',
      name: 'default.repair_order',
      display_name: 'Repair Orders',
      version: 'v2.0',
      status: 'valid',
      mode: 'published',
      catalog: {
        id: 1,
        uuid: '0fc18295-e1a2-4c3c-b72a-894725c12488',
        created_at: '2023-08-21T16:48:51.146121+00:00',
        updated_at: '2023-08-21T16:48:51.146122+00:00',
        extra_params: {},
        name: 'warehouse',
      },
      description: 'Repair order dimension',
      query:
        'SELECT repair_order_id, municipality_id, hard_hat_id, ' +
        'dispatcher_id FROM default.repair_orders',
      availability: null,
      columns: [
        {
          name: 'repair_order_id',
          type: 'int',
          attributes: [],
          dimension: {
            name: 'default.repair_order',
          },
        },
        {
          name: 'municipality_id',
          type: 'int',
          attributes: [],
          dimension: null,
        },
        {
          name: 'hard_hat_id',
          type: 'int',
          attributes: [],
          dimension: null,
        },
        {
          name: 'dispatcher_id',
          type: 'int',
          attributes: [],
          dimension: null,
        },
      ],
      updated_at: '2023-08-21T16:48:52.981201+00:00',
      materializations: [],
      parents: [],
    },
  ];

  beforeEach(() => {
    // Reset the mocks before each test
    mockDjClient.DataJunctionAPI.revisions.mockReset();
  });

  it('renders revision diff', async () => {
    mockDjClient.DataJunctionAPI.revisions.mockReturnValue(
      mockNodesWithDimension,
    );
    const element = (
      <DJClientContext.Provider value={mockDjClient}>
        <RevisionDiff />
      </DJClientContext.Provider>
    );
    const { container } = render(
      <MemoryRouter
        initialEntries={['/nodes/default.repair_orders_cube/revisions/v2.0']}
      >
        <Routes>
          <Route path="nodes/:name/revisions/:revision" element={element} />
        </Routes>
      </MemoryRouter>,
    );
    await waitFor(() => {
      expect(mockDjClient.DataJunctionAPI.revisions).toHaveBeenCalledWith(
        'default.repair_orders_cube',
      );

      const diffViews = screen.getAllByRole('gridcell', 'DiffView');
      diffViews.map(diffView => expect(diffView).toBeInTheDocument());
    });
  });
});