test: update

Author: zombieJ

tests/unique.test.tsx

@@ -2,6 +2,7 @@ import { cleanup, fireEvent, render } from '@testing-library/react';
 import React from 'react';
 import Trigger, { UniqueProvider } from '../src';
 import { awaitFakeTimer } from './util';
+import type { TriggerProps } from '../src';
 
 // Mock FloatBg to check if open props changed
 global.openChangeLog = [];
@@ -27,6 +28,34 @@ jest.mock('../src/UniqueProvider/FloatBg', () => {
   };
 });
 
+async function setupAndOpenPopup(triggerProps: Partial<TriggerProps> = {}) {
+  const { container } = render(
+    <UniqueProvider>
+      <Trigger
+        action={['click']}
+        popup={<strong className="x-content">tooltip</strong>}
+        unique
+        {...triggerProps}
+      >
+        <div className="target">click me</div>
+      </Trigger>
+    </UniqueProvider>,
+  );
+
+  // Initially no popup should be visible
+  expect(document.querySelector('.rc-trigger-popup')).toBeFalsy();
+
+  // Click trigger to show popup
+  fireEvent.click(container.querySelector('.target'));
+  await awaitFakeTimer();
+
+  // Check that popup exists
+  const popup = document.querySelector('.rc-trigger-popup');
+  expect(popup).toBeTruthy();
+
+  return { container, popup };
+}
+
 describe('Trigger.Unique', () => {
   beforeEach(() => {
     jest.useFakeTimers();
@@ -154,29 +183,7 @@ describe('Trigger.Unique', () => {
   });
 
   it('should apply uniqueBgClassName to FloatBg component', async () => {
-    const { container } = render(
-      <UniqueProvider>
-        <Trigger
-          action={['click']}
-          popup={<strong className="x-content">tooltip</strong>}
-          unique
-          uniqueBgClassName="custom-bg-class"
-        >
-          <div className="target">click me</div>
-        </Trigger>
-      </UniqueProvider>,
-    );
-
-    // Initially no popup should be visible
-    expect(document.querySelector('.rc-trigger-popup')).toBeFalsy();
-
-    // Click trigger to show popup
-    fireEvent.click(container.querySelector('.target'));
-    await awaitFakeTimer();
-
-    // Check that popup exists
-    const popup = document.querySelector('.rc-trigger-popup');
-    expect(popup).toBeTruthy();
+    await setupAndOpenPopup({ uniqueBgClassName: 'custom-bg-class' });
 
     // Check that FloatBg has the custom background className
     const floatBg = document.querySelector('.rc-trigger-popup-float-bg');
@@ -185,29 +192,7 @@ describe('Trigger.Unique', () => {
   });
 
   it('should apply uniqueBgStyle to FloatBg component', async () => {
-    const { container } = render(
-      <UniqueProvider>
-        <Trigger
-          action={['click']}
-          popup={<strong className="x-content">tooltip</strong>}
-          unique
-          uniqueBgStyle={{ backgroundColor: 'red', border: '1px solid blue' }}
-        >
-          <div className="target">click me</div>
-        </Trigger>
-      </UniqueProvider>,
-    );
-
-    // Initially no popup should be visible
-    expect(document.querySelector('.rc-trigger-popup')).toBeFalsy();
-
-    // Click trigger to show popup
-    fireEvent.click(container.querySelector('.target'));
-    await awaitFakeTimer();
-
-    // Check that popup exists
-    const popup = document.querySelector('.rc-trigger-popup');
-    expect(popup).toBeTruthy();
+    await setupAndOpenPopup({ uniqueBgStyle: { backgroundColor: 'red', border: '1px solid blue' } });
 
     // Check that FloatBg has the custom background style
     const floatBg = document.querySelector('.rc-trigger-popup-float-bg');
@@ -219,28 +204,7 @@ describe('Trigger.Unique', () => {
   });
 
   it('should not apply any additional className to FloatBg when uniqueBgClassName is not provided', async () => {
-    const { container } = render(
-      <UniqueProvider>
-        <Trigger
-          action={['click']}
-          popup={<strong className="x-content">tooltip</strong>}
-          unique
-        >
-          <div className="target">click me</div>
-        </Trigger>
-      </UniqueProvider>,
-    );
-
-    // Initially no popup should be visible
-    expect(document.querySelector('.rc-trigger-popup')).toBeFalsy();
-
-    // Click trigger to show popup
-    fireEvent.click(container.querySelector('.target'));
-    await awaitFakeTimer();
-
-    // Check that popup exists
-    const popup = document.querySelector('.rc-trigger-popup');
-    expect(popup).toBeTruthy();
+    await setupAndOpenPopup();
 
     // Check that FloatBg exists but does not have custom background className
     const floatBg = document.querySelector('.rc-trigger-popup-float-bg');