ant-design/components/app/__tests__/index.test.tsx

import React, { useEffect } from 'react';
import App from '..';
import mountTest from '../../../tests/shared/mountTest';
import rtlTest from '../../../tests/shared/rtlTest';
import { render, waitFakeTimer } from '../../../tests/utils';
import type { AppConfig } from '../context';
import { AppConfigContext } from '../context';

describe('App', () => {
  mountTest(App);
  rtlTest(App);

  beforeEach(() => {
    jest.useFakeTimers();
  });

  afterEach(() => {
    jest.clearAllTimers();
    jest.useRealTimers();
  });

  it('single', () => {
    // Sub page
    const MyPage = () => {
      const { message } = App.useApp();
      React.useEffect(() => {
        message.success('Good!');
      }, [message]);

      return <div>Hello World</div>;
    };

    // Entry component
    const MyApp = () => (
      <App>
        <MyPage />
      </App>
    );

    const { getByText, container } = render(<MyApp />);
    expect(getByText('Hello World')).toBeTruthy();
    expect(container.firstChild).toMatchSnapshot();
  });

  it('should work as message and notification config configured in app', async () => {
    let consumedConfig: AppConfig | undefined;
    const Consumer = () => {
      const { message, notification } = App.useApp();
      consumedConfig = React.useContext(AppConfigContext);

      useEffect(() => {
        message.success('Message 1');
        message.success('Message 2');
        notification.success({ message: 'Notification 1' });
        notification.success({ message: 'Notification 2' });
        notification.success({ message: 'Notification 3' });
      }, [message, notification]);

      return <div />;
    };
    const Wrapper = () => (
      <App message={{ maxCount: 1 }} notification={{ maxCount: 2 }}>
        <Consumer />
      </App>
    );

    render(<Wrapper />);

    await waitFakeTimer();

    expect(consumedConfig?.message).toStrictEqual({ maxCount: 1 });
    expect(consumedConfig?.notification).toStrictEqual({ maxCount: 2 });

    expect(document.querySelectorAll('.ant-message-notice')).toHaveLength(1);
    expect(document.querySelectorAll('.ant-notification-notice')).toHaveLength(2);
  });

  it('should be a merged config configured in nested app', async () => {
    let offsetConsumedConfig: AppConfig | undefined;
    let maxCountConsumedConfig: AppConfig | undefined;
    const OffsetConsumer = () => {
      offsetConsumedConfig = React.useContext(AppConfigContext);
      return <div />;
    };
    const MaxCountConsumer = () => {
      maxCountConsumedConfig = React.useContext(AppConfigContext);
      return <div />;
    };
    const Wrapper = () => (
      <App message={{ maxCount: 1 }} notification={{ maxCount: 2 }}>
        <App message={{ top: 32 }} notification={{ top: 96 }}>
          <OffsetConsumer />
        </App>
        <MaxCountConsumer />
      </App>
    );

    render(<Wrapper />);

    expect(offsetConsumedConfig?.message).toStrictEqual({ maxCount: 1, top: 32 });
    expect(offsetConsumedConfig?.notification).toStrictEqual({ maxCount: 2, top: 96 });
    expect(maxCountConsumedConfig?.message).toStrictEqual({ maxCount: 1 });
    expect(maxCountConsumedConfig?.notification).toStrictEqual({ maxCount: 2 });
  });

  it('should respect config from props in priority', async () => {
    let config: AppConfig | undefined;
    const Consumer = () => {
      config = React.useContext(AppConfigContext);
      return <div />;
    };
    const Wrapper = () => (
      <App message={{ maxCount: 10, top: 20 }} notification={{ maxCount: 30, bottom: 40 }}>
        <App message={{ maxCount: 11 }} notification={{ bottom: 41 }}>
          <Consumer />
        </App>
      </App>
    );

    render(<Wrapper />);

    expect(config?.message).toStrictEqual({ maxCount: 11, top: 20 });
    expect(config?.notification).toStrictEqual({ maxCount: 30, bottom: 41 });
  });
});