import * as React from 'react';
import { ReactWrapper } from 'enzyme';
import { mountWithProvider as mount, mountWithProvider } from 'test/utils';
import { Props, PropsOf, InstanceOf } from 'src/types';
import { Popper } from 'src/utils/positioner';
import { positioningProps } from 'test/specs/commonTests/implementsPopperProps';

export type ShorthandTestOptions<TProps = any> = {
  mapsValueToProp?: keyof (TProps & React.HTMLProps<HTMLElement>) | false;
  implementsPopper?: boolean;
  requiredProps?: Props;
  requiredShorthandProps?: Props;
};

export const DefaultShorthandTestOptions: ShorthandTestOptions = {
  mapsValueToProp: 'content',
};

export type ShorthandPropTestsRunner<TComponent> = <TShorthandComponent extends React.ComponentType>(
  shorthandProp: keyof PropsOf<InstanceOf<TComponent>>,
  ShorthandComponent: TShorthandComponent,
  options?: ShorthandTestOptions<PropsOf<TShorthandComponent>>,
) => any;

export type ShorthandPropTestsFactory = <TComponent extends React.ComponentType>(
  Component: TComponent,
) => ShorthandPropTestsRunner<TComponent>;

export const implementsShorthandProp = ((Component: React.ComponentType) => {
  return function implementsShorthandProp(
    shorthandProp: string,
    ShorthandComponent: React.ComponentType,
    options: ShorthandTestOptions = DefaultShorthandTestOptions,
  ) {
    const mapsValueToProp = options.mapsValueToProp as string;
    const { displayName } = ShorthandComponent;

    const checkPropsMatch = (props: Props, matchedProps: Props) =>
      Object.keys(matchedProps).every(propName => matchedProps[propName] === props[propName]);

    const expectContainsSingleShorthandElement = (wrapper: ReactWrapper, withProps: Props) =>
      expect(
        wrapper.find(ShorthandComponent).filterWhere(node => checkPropsMatch(node.props(), withProps)).length,
      ).toEqual(1);

    const expectShorthandPropsAreHandled = (withProps: Props | string) => {
      const props = { ...options.requiredProps, [shorthandProp]: withProps };
      const matchedProps = typeof withProps === 'string' ? { [mapsValueToProp]: withProps } : withProps;

      expectContainsSingleShorthandElement(mountWithProvider(<Component {...props} />), matchedProps);
    };

    describe(`shorthand property '${shorthandProp}' with default value of '${displayName}' component`, () => {
      test(`is defined`, () => {
        expect(Component.propTypes[shorthandProp]).toBeTruthy();
      });

      if (options.mapsValueToProp) {
        test(`string value is handled as ${displayName}'s ${mapsValueToProp}`, () => {
          expectShorthandPropsAreHandled('shorthand prop value');
        });
      }

      test(`object value is spread as ${displayName}'s props`, () => {
        expectShorthandPropsAreHandled({
          ...options.requiredShorthandProps,
          foo: 'foo value',
          bar: 'bar value',
        });
      });
    });

    if (options.implementsPopper) {
      describe('implements all positioning props for Popper', () => {
        Object.entries(positioningProps).forEach(([positioningProp, positioningValue]) => {
          test(`"${positioningProp}" is passed to a Popper component`, () => {
            const wrapper = mount(
              React.createElement(Component, {
                ...options.requiredProps,
                [shorthandProp]: {
                  ...options.requiredShorthandProps,
                  popper: { [positioningProp]: positioningValue },
                },
              }),
            );
            // Popper will be a parent of shorthand
            const popper = wrapper.find(ShorthandComponent).closest(Popper);

            expect(popper.prop(positioningProp)).toBe(positioningValue);
          });
        });
      });
    }
  };
}) as ShorthandPropTestsFactory;