ant-design-vue/components/flex/__tests__/index.test.js

import { mount } from '@vue/test-utils';
import Flex from '..';
import mountTest from '../../../tests/shared/mountTest';

describe('Flex', () => {
  mountTest(Flex);

  it('Flex', () => {
    const wrapper = mount({
      render() {
        return <Flex justify="center">test</Flex>;
      },
    });
    const wrapper2 = mount({
      render() {
        return <Flex gap={100}>test</Flex>;
      },
    });
    expect(wrapper.classes('ant-flex')).toBeTruthy();
    expect(wrapper.find('.ant-flex-justify-center')).toBeTruthy();
    expect(wrapper2.classes('ant-flex')).toBeTruthy();
    expect(wrapper2.element.style.gap).toBe('100px');
  });

  it('Component work', () => {
    const wrapper = mount({
      render() {
        return <Flex>test</Flex>;
      },
    });
    const wrapper2 = mount({
      render() {
        return <Flex component="span">test</Flex>;
      },
    });
    expect(wrapper.find('.ant-flex').element.tagName).toBe('DIV');
    expect(wrapper2.find('.ant-flex').element.tagName).toBe('SPAN');
  });

  it('when vertical=true should stretch work', () => {
    const wrapper = mount({
      render() {
        return <Flex vertical>test</Flex>;
      },
    });
    const wrapper2 = mount({
      render() {
        return (
          <Flex vertical align="center">
            test
          </Flex>
        );
      },
    });
    expect(wrapper.find('.ant-flex-align-stretch')).toBeTruthy();
    expect(wrapper2.find('.ant-flex-align-center')).toBeTruthy();
  });
});