Sfoglia il codice sorgente

Fix test warnings (#1908)

* fix fetch mock warnings in brokers.spec.tsx

* use separate waitFor-s for fetch-mock call expects
Arsen Simonyan 3 anni fa
parent
commit
b296c1a803

+ 25 - 5
kafka-ui-react-app/src/components/Brokers/__test__/Brokers.spec.tsx

@@ -22,25 +22,45 @@ describe('Brokers Component', () => {
     );
 
   describe('Brokers', () => {
+    let fetchBrokersMock: fetchMock.FetchMockStatic;
+    const fetchStatsUrl = `/api/clusters/${clusterName}/stats`;
+
+    beforeEach(() => {
+      fetchBrokersMock = fetchMock.getOnce(
+        `/api/clusters/${clusterName}/brokers`,
+        clusterStatsPayload
+      );
+    });
+
     it('renders', async () => {
-      const mock = fetchMock.getOnce(
-        `/api/clusters/${clusterName}/stats`,
+      const fetchStatsMock = fetchMock.getOnce(
+        fetchStatsUrl,
         clusterStatsPayload
       );
       renderComponent();
-      await waitFor(() => expect(mock.called()).toBeTruthy());
+      await waitFor(() => {
+        expect(fetchStatsMock.called()).toBeTruthy();
+      });
+      await waitFor(() => {
+        expect(fetchBrokersMock.called()).toBeTruthy();
+      });
       expect(screen.getByRole('table')).toBeInTheDocument();
       const rows = screen.getAllByRole('row');
       expect(rows.length).toEqual(3);
     });
 
     it('shows warning when offlinePartitionCount > 0', async () => {
-      const mock = fetchMock.getOnce(`/api/clusters/${clusterName}/stats`, {
+      const fetchStatsMock = fetchMock.getOnce(fetchStatsUrl, {
         ...clusterStatsPayload,
         offlinePartitionCount: 1345,
       });
       renderComponent();
-      await waitFor(() => expect(mock.called()).toBeTruthy());
+      await waitFor(() => {
+        expect(fetchStatsMock.called()).toBeTruthy();
+      });
+      await waitFor(() => {
+        expect(fetchBrokersMock.called()).toBeTruthy();
+      });
       const onlineWidget = screen.getByText(
         clusterStatsPayload.onlinePartitionCount
       );