import { Button } from '@chakra-ui/react';
import { test, expect } from '@playwright/experimental-ct-react';
import React from 'react';

import TestApp from 'playwright/TestApp';

[
  { variant: 'solid' },
  { variant: 'solid', colorScheme: 'gray', withDarkMode: true },
  { variant: 'outline', colorScheme: 'gray', withDarkMode: true },
  { variant: 'outline', colorScheme: 'gray-dark', withDarkMode: true },
  { variant: 'outline', colorScheme: 'blue', withDarkMode: true },
  { variant: 'simple', withDarkMode: true },
  { variant: 'ghost', withDarkMode: true },
  { variant: 'subtle' },
  { variant: 'subtle', colorScheme: 'gray', withDarkMode: true },
].forEach(({ variant, colorScheme, withDarkMode }) => {
  test.describe(`variant ${ variant }${ colorScheme ? ` with ${ colorScheme } color scheme` : '' }${ withDarkMode ? ' +@dark-mode' : '' }`, () => {
    test('base', async({ mount }) => {
      const component = await mount(
        <TestApp>
          <Button variant={ variant } colorScheme={ colorScheme }>Click me</Button>
        </TestApp>,
      );
      await expect(component.locator('button')).toHaveScreenshot();
    });

    test('disabled', async({ mount }) => {
      const component = await mount(
        <TestApp>
          <Button variant={ variant } colorScheme={ colorScheme } disabled>Click me</Button>
        </TestApp>,
      );
      await expect(component.locator('button')).toHaveScreenshot();
    });

    test('hovered', async({ mount }) => {
      const component = await mount(
        <TestApp>
          <Button variant={ variant } colorScheme={ colorScheme }>Click me</Button>
        </TestApp>,
      );
      await component.getByText(/click/i).hover();
      await expect(component.locator('button')).toHaveScreenshot();
    });

    test('active', async({ mount }) => {
      const component = await mount(
        <TestApp>
          <Button variant={ variant } colorScheme={ colorScheme } isActive>Click me</Button>
        </TestApp>,
      );
      await expect(component.locator('button')).toHaveScreenshot();
    });
  });
});
