fix: failing tests

stakwork · Apr 20, 2024 · 44b1b0d · 44b1b0d
1 parent c4d1a4d
commit 44b1b0d
Showing 1 changed file with 6 additions and 4 deletions.
diff --git a/src/components/SettingsModal/SettingsView/GraphBlueprint/__tests__/index.tsx b/src/components/SettingsModal/SettingsView/GraphBlueprint/__tests__/index.tsx
@@ -1,6 +1,6 @@
-import React from 'react'
-import { render, screen } from '@testing-library/react'
 import '@testing-library/jest-dom'
+import { render, screen, waitFor } from '@testing-library/react'
+import React from 'react'
 import { GraphBlueprint } from '../index'
 
 jest.mock('../../../../../network/fetchSourcesData', () => ({
@@ -15,8 +15,10 @@ describe('GraphBlueprint', () => {
   it('should display only one Custom node', async () => {
     render(<GraphBlueprint />)
 
-    const customNodes = await screen.findAllByText('Custom')
+    await waitFor(async () => {
+      const customNodes = await screen.findAllByText('Custom')
 
-    expect(customNodes).toHaveLength(1)
+      expect(customNodes).toHaveLength(1)
+    })
   })
 })