[test]: Update unit test case

actiontech · Sep 11, 2024 · 4430366 · 4430366
1 parent 7c18064
commit 4430366
Show file tree

Hide file tree

Showing 3 changed files with 62 additions and 37 deletions.
diff --git a/packages/sqle/src/page/SqlAudit/List/index.test.tsx b/packages/sqle/src/page/SqlAudit/List/index.test.tsx
@@ -200,17 +200,29 @@ describe('sqle/SqlAudit/List', () => {
   });
 
   it('render polling request when sql audit status is auditing', async () => {
-    sqlAuditRecordsSpy.mockImplementation(() =>
-      createSpySuccessResponse({
-        data: [
-          {
-            ...sqlAuditRecordMockData[0],
-            sql_audit_status: 'auditing'
-          }
-        ],
-        total_nums: 2
-      })
-    );
+    sqlAuditRecordsSpy
+      .mockImplementationOnce(() =>
+        createSpySuccessResponse({
+          data: [
+            {
+              ...sqlAuditRecordMockData[0],
+              sql_audit_status: 'auditing'
+            }
+          ],
+          total_nums: 2
+        })
+      )
+      .mockImplementationOnce(() =>
+        createSpySuccessResponse({
+          data: [
+            {
+              ...sqlAuditRecordMockData[0],
+              sql_audit_status: 'successfully'
+            }
+          ],
+          total_nums: 2
+        })
+      );
     renderWithReduxAndTheme(customRender());
     await act(async () => jest.advanceTimersByTime(3000));
     expect(sqlAuditRecordsSpy).toHaveBeenCalledTimes(1);
@@ -233,7 +245,5 @@ describe('sqle/SqlAudit/List', () => {
     renderWithReduxAndTheme(customRender());
     await act(async () => jest.advanceTimersByTime(3000));
     expect(sqlAuditRecordsSpy).toHaveBeenCalledTimes(1);
-    await act(async () => jest.advanceTimersByTime(3000));
-    expect(sqlAuditRecordsSpy).not.toHaveBeenCalledTimes(2);
   });
 });
diff --git a/packages/sqle/src/page/SqlManagement/component/SQLEEIndex/index.test.tsx b/packages/sqle/src/page/SqlManagement/component/SQLEEIndex/index.test.tsx
@@ -596,15 +596,21 @@ describe('page/SqlManagement/SQLEEIndex', () => {
 
   it('render polling request when sql audit status is auditing', async () => {
     const request = sqlManage.getSqlManageList();
-    request.mockImplementation(() =>
-      createSpySuccessResponse({
-        data: [
-          {
-            ...sqlManageListData.data[sqlManageListData.data.length - 1]
-          }
-        ]
-      })
-    );
+    request
+      .mockImplementationOnce(() =>
+        createSpySuccessResponse({
+          data: [
+            {
+              ...sqlManageListData.data[sqlManageListData.data.length - 1]
+            }
+          ]
+        })
+      )
+      .mockImplementationOnce(() =>
+        createSpySuccessResponse({
+          data: [sqlManageListData.data[0]]
+        })
+      );
     superRender(<SQLEEIndex />);
     await act(async () => jest.advanceTimersByTime(3000));
     expect(request).toHaveBeenCalledTimes(1);
@@ -622,7 +628,5 @@ describe('page/SqlManagement/SQLEEIndex', () => {
     superRender(<SQLEEIndex />);
     await act(async () => jest.advanceTimersByTime(3000));
     expect(request).toHaveBeenCalledTimes(1);
-    await act(async () => jest.advanceTimersByTime(3000));
-    expect(request).not.toHaveBeenCalledTimes(2);
   });
 });
diff --git a/...ges/sqle/src/page/SqlManagementConf/Detail/ScanTypeSqlCollection/__tests__/index.test.tsx b/...ges/sqle/src/page/SqlManagementConf/Detail/ScanTypeSqlCollection/__tests__/index.test.tsx
@@ -134,18 +134,31 @@ describe('test ScanTypeSqlCollection', () => {
   });
 
   it('should polling request when sql audit status is being_audited', async () => {
-    getInstanceAuditPlanSQLDataSpy.mockImplementation(() => {
-      return createSpySuccessResponse({
-        data: {
-          rows: [
-            {
-              ...mockAuditPlanSQLData?.rows?.[0],
-              audit_results: 'being_audited'
-            }
-          ]
-        }
+    getInstanceAuditPlanSQLDataSpy
+      .mockImplementationOnce(() => {
+        return createSpySuccessResponse({
+          data: {
+            rows: [
+              {
+                ...mockAuditPlanSQLData?.rows?.[0],
+                audit_results: 'being_audited'
+              }
+            ]
+          }
+        });
+      })
+      .mockImplementationOnce(() => {
+        return createSpySuccessResponse({
+          data: {
+            rows: [
+              {
+                ...mockAuditPlanSQLData?.rows?.[0],
+                audit_results: ''
+              }
+            ]
+          }
+        });
       });
-    });
 
     customRender();
     await act(async () => jest.advanceTimersByTime(3000));
@@ -171,7 +184,5 @@ describe('test ScanTypeSqlCollection', () => {
     customRender();
     await act(async () => jest.advanceTimersByTime(3000));
     expect(getInstanceAuditPlanSQLDataSpy).toHaveBeenCalledTimes(1);
-    await act(async () => jest.advanceTimersByTime(3000));
-    expect(getInstanceAuditPlanSQLDataSpy).not.toHaveBeenCalledTimes(2);
   });
 });