diff --git a/src/Overflow.tsx b/src/Overflow.tsx index de6241b..acc850c 100644 --- a/src/Overflow.tsx +++ b/src/Overflow.tsx @@ -357,7 +357,6 @@ function Overflow( }; // >>>>> Rest node - let restNode: React.ReactNode; const restContextProps = { order: displayRest ? mergedDisplayCount : Number.MAX_SAFE_INTEGER, className: `${itemPrefixCls}-rest`, @@ -365,10 +364,18 @@ function Overflow( display: displayRest, }; - if (!renderRawRest) { - const mergedRenderRest = renderRest || defaultRenderRest; + const mergedRenderRest = renderRest || defaultRenderRest; - restNode = ( + const restNode = renderRawRest ? ( + + {renderRawRest(omittedItems)} + + ) : ( ( : mergedRenderRest} ); - } else if (renderRawRest) { - restNode = ( - - {renderRawRest(omittedItems)} - - ); - } - let overflowNode = ( + const overflowNode = ( ( ); - if (isResponsive) { - overflowNode = ( - - {overflowNode} - - ); - } - - return overflowNode; + return isResponsive ? ( + + {overflowNode} + + ) : overflowNode; } const ForwardOverflow = React.forwardRef(Overflow); diff --git a/tests/ssr.spec.tsx b/tests/ssr.spec.tsx index 75e5038..50c9798 100644 --- a/tests/ssr.spec.tsx +++ b/tests/ssr.spec.tsx @@ -1,6 +1,5 @@ import React from 'react'; import { render } from 'enzyme'; -import { act } from 'react-dom/test-utils'; import Overflow from '../src'; interface ItemType {