fix: add test case

react-component · Jan 9, 2023 · 4143c57 · 4143c57
1 parent 8aae348
commit 4143c57
Showing 1 changed file with 17 additions and 3 deletions.
diff --git a/tests/CSSMotion.spec.tsx b/tests/CSSMotion.spec.tsx
@@ -160,13 +160,14 @@ describe('CSSMotion', () => {
     );
 
     it('leaveClassName should add to dom', () => {
-      const genMotion = ({ visible }) => {
+      const genMotion = (props) => {
+        const {visible,leavedClassName} = props
         return (
             <CSSMotion
                 motionName="transition"
                 visible={visible}
                 removeOnLeave={false}
-                leavedClassName={'removed'}
+                leavedClassName={leavedClassName}
             >
               {({ style, className }) => {
                 return (
@@ -188,14 +189,27 @@ describe('CSSMotion', () => {
       });
 
       expect(container.querySelector('.motion-box')).toBeTruthy();
-      rerender(genMotion({ visible: false }));
+      rerender(genMotion({ visible: false,leavedClassName:'removed'}));
       act(() => {
         jest.runAllTimers();
       });
 
       fireEvent.transitionEnd(container.querySelector('.motion-box'));
 
       expect(container.querySelector('.motion-box')).toHaveClass('removed');
+
+      rerender(genMotion({ visible: true }));
+      act(() => {
+        jest.runAllTimers();
+      });
+      rerender(genMotion({ visible: false }));
+      act(() => {
+        jest.runAllTimers();
+      });
+
+      fireEvent.transitionEnd(container.querySelector('.motion-box'));
+      expect(container.querySelector('.motion-box')?.classList.contains('removed')).toBeFalsy();
+
     });
 
     it('stop transition if config motion to false', () => {