Fix hanging test

locustio · Jan 19, 2024 · b4bc490 · b4bc490
1 parent f5e235d
commit b4bc490
Show file tree

Hide file tree

Showing 3 changed files with 28 additions and 14 deletions.
diff --git a/.github/workflows/tests.yml b/.github/workflows/tests.yml
@@ -47,7 +47,7 @@ jobs:
       - run: python -m pip install tox
       - run: python -m tox -e ${{ matrix.tox }}
 
-  lint_typecheck_webui:
+  lint_typecheck_test_webui:
     runs-on: ubuntu-latest
     steps:
       - uses: actions/checkout@v3

diff --git a/locust/webui/src/components/SwarmForm/tests/SwarmUserClassPicker.test.tsx b/locust/webui/src/components/SwarmForm/tests/SwarmUserClassPicker.test.tsx
@@ -47,7 +47,7 @@ const defaultProps = {
   setSwarm: updateSwarm,
 };
 
-describe('SwarmForm', () => {
+describe('SwarmUserClassPicker', () => {
   beforeAll(() => server.listen());
   afterEach(() => {
     server.resetHandlers();
@@ -95,6 +95,8 @@ describe('SwarmForm', () => {
       },
     );
 
+    vi.useFakeTimers();
+
     act(() => {
       // Open modal
       fireEvent.click(getAllByRole('button')[0]);
@@ -104,12 +106,15 @@ describe('SwarmForm', () => {
       fireEvent.click(getByRole('button', { name: 'Save' }));
     });
 
-    await waitFor(async () => {
-      const submittedData = updateUserSettings.mock.calls[0][0];
-      if (submittedData) {
-        expect(submittedData).toEqual({ ...mockUsers.Example, userClassName: 'Example' });
-      }
+    await act(async () => {
+      await vi.runAllTimersAsync();
     });
+
+    expect(updateUserSettings).toHaveBeenCalled();
+
+    const submittedData = updateUserSettings.mock.calls[0][0];
+
+    expect(submittedData).toEqual({ ...mockUsers.Example, userClassName: 'Example' });
   });
 
   test('should allow for configuring tasks, host, fixed_count, and weight to be modified', async () => {
@@ -130,6 +135,8 @@ describe('SwarmForm', () => {
       },
     );
 
+    vi.useFakeTimers();
+
     act(() => {
       // Open modal
       fireEvent.click(getAllByRole('button')[0]);
@@ -152,12 +159,15 @@ describe('SwarmForm', () => {
       fireEvent.click(getByRole('button', { name: 'Save' }));
     });
 
-    await waitFor(async () => {
-      const submittedData = updateUserSettings.mock.calls[0][0];
-      if (submittedData) {
-        expect(submittedData).toEqual({ ...updatedUser, userClassName: 'Example' });
-        expect(store.getState().swarm.users.Example).toEqual(updatedUser);
-      }
+    await act(async () => {
+      await vi.runAllTimersAsync();
     });
+
+    expect(updateUserSettings).toHaveBeenCalled();
+
+    const submittedData = updateUserSettings.mock.calls[0][0];
+
+    expect(submittedData).toEqual({ ...updatedUser, userClassName: 'Example' });
+    expect(store.getState().swarm.users.Example).toEqual(updatedUser);
   });
 });
diff --git a/locust/webui/src/test/setup.ts b/locust/webui/src/test/setup.ts
@@ -1,5 +1,5 @@
 import { cleanup } from '@testing-library/react';
-import { afterEach, vi } from 'vitest';
+import { afterAll, afterEach, vi } from 'vitest';
 
 import { TEST_BASE_API } from 'test/constants';
 import { swarmStateMock } from 'test/mocks/swarmState.mock';
@@ -25,3 +25,7 @@ vi.mock('echarts', async () => {
 
   return { ...actual, init: (...args: any[]) => actual.init(...args, { width: 100, height: 100 }) };
 });
+
+afterAll(() => {
+  vi.clearAllMocks();
+});