import { render, screen } from '@testing-library/react';

import { WizardStep, WizardStepProps } from '../WizardStep';
import * as WizardContext from '../WizardContext';
import { WizardStepType } from '..';

const testStepProps: WizardStepProps = {
  id: 'test-step',
  name: 'Test step',
  footer: <>Step footer</>
};

const testStep: WizardStepType = {
  ...testStepProps,
  index: 1
};

const setStep = jest.fn();

const wizardContext: WizardContext.WizardContextProps = {
  setStep,
  steps: [testStep],
  activeStep: testStep,
  footer: <>Wizard footer</>,
  goToNextStep: jest.fn(),
  goToPrevStep: jest.fn(),
  close: jest.fn(),
  goToStepById: jest.fn(),
  goToStepByName: jest.fn(),
  goToStepByIndex: jest.fn(),
  setFooter: jest.fn(),
  getStep: jest.fn()
};

const useWizardContextSpy = jest.spyOn(WizardContext, 'useWizardContext');

beforeEach(() => {
  useWizardContextSpy.mockReturnValue(wizardContext);
});

test('renders without children', () => {
  const { container } = render(<WizardStep id="test-step" name="Test step" />);
  expect(container).toBeVisible();
});

test('renders with children', () => {
  render(
    <WizardStep id="test-step" name="Test step">
      content
    </WizardStep>
  );

  expect(screen.getByText('content')).toBeVisible();
});

test('updates "isDisabled" in context when the value changes', () => {
  render(<WizardStep {...testStep} isDisabled />);
  expect(setStep).toHaveBeenCalledWith({ ...testStepProps, isDisabled: true, isVisited: true });
});

test('updates "navItem" in context when the value changes', () => {
  const testNavItem = <>Some nav item</>;
  render(<WizardStep {...testStep} navItem={testNavItem} />);

  expect(setStep).toHaveBeenCalledWith({ ...testStepProps, navItem: testNavItem, isVisited: true });
});

test('updates "status" in context when the value changes', () => {
  render(<WizardStep {...testStep} status="error" />);
  expect(setStep).toHaveBeenCalledWith({ ...testStepProps, status: 'error', isVisited: true });
  render(<WizardStep {...testStep} status="success" />);
  expect(setStep).toHaveBeenCalledWith({ ...testStepProps, status: 'success', isVisited: true });
  render(<WizardStep {...testStep} status="warning" />);
  expect(setStep).toHaveBeenCalledWith({ ...testStepProps, status: 'warning', isVisited: true });
});
