mirror of
https://gitee.com/ant-design/ant-design.git
synced 2024-12-05 13:37:52 +08:00
54 lines
1.8 KiB
TypeScript
54 lines
1.8 KiB
TypeScript
|
import React, { memo, useState, useRef, useContext } from 'react';
|
||
|
import { mount } from 'enzyme';
|
||
|
import LocaleProvider, { ANT_MARK } from '..';
|
||
|
import LocaleContext from '../context';
|
||
|
|
||
|
const defaultLocale = {
|
||
|
locale: 'locale',
|
||
|
};
|
||
|
// we use'memo' here in order to only render inner component while context changed.
|
||
|
const CacheInner = memo(() => {
|
||
|
const countRef = useRef(0);
|
||
|
countRef.current++;
|
||
|
// subscribe locale context
|
||
|
useContext(LocaleContext);
|
||
|
return (
|
||
|
<div>
|
||
|
Child Rendering Count: <span id="child_count">{countRef.current}</span>
|
||
|
</div>
|
||
|
);
|
||
|
});
|
||
|
|
||
|
const CacheOuter = () => {
|
||
|
// We use 'useState' here in order to trigger parent component rendering.
|
||
|
const [count, setCount] = useState(1);
|
||
|
const handleClick = () => {
|
||
|
setCount(count + 1);
|
||
|
};
|
||
|
// During each rendering phase, the cached context value returned from method 'LocaleProvider#getMemoizedContextValue' will take effect.
|
||
|
// So 'CacheInner' component won't rerender.
|
||
|
return (
|
||
|
<div>
|
||
|
<button type="button" onClick={handleClick} id="parent_btn">
|
||
|
Click
|
||
|
</button>
|
||
|
Parent Rendering Count: <span id="parent_count">{count}</span>
|
||
|
<LocaleProvider locale={defaultLocale} _ANT_MARK__={ANT_MARK}>
|
||
|
<CacheInner />
|
||
|
</LocaleProvider>
|
||
|
</div>
|
||
|
);
|
||
|
};
|
||
|
|
||
|
it("Rendering on LocaleProvider won't trigger rendering on child component.", () => {
|
||
|
const wrapper = mount(<CacheOuter />);
|
||
|
wrapper.find('#parent_btn').at(0).simulate('click');
|
||
|
expect(wrapper.find('#parent_count').text()).toBe('2');
|
||
|
// child component won't rerender
|
||
|
expect(wrapper.find('#child_count').text()).toBe('1');
|
||
|
wrapper.find('#parent_btn').at(0).simulate('click');
|
||
|
expect(wrapper.find('#parent_count').text()).toBe('3');
|
||
|
// child component won't rerender
|
||
|
expect(wrapper.find('#child_count').text()).toBe('1');
|
||
|
});
|