BN-75 | Refactor screenshot tests

Author: devsingh05298

src/displayControls/diagnoses/__tests__/DiagnosesControl.integration.test.tsx

@@ -1,5 +1,5 @@
 import React from 'react';
-import { render, screen, waitFor } from '@testing-library/react';
+import { render, screen, waitFor, act } from '@testing-library/react';
 import userEvent from '@testing-library/user-event';
 import { MemoryRouter } from 'react-router-dom';
 import axios from 'axios';

src/displayControls/diagnoses/__tests__/DiagnosesControl.screenshot.test.tsx

@@ -1,5 +1,5 @@
 import React from 'react';
-import { render } from '@testing-library/react';
+import { render, act, waitFor } from '@testing-library/react';
 import { MemoryRouter } from 'react-router-dom';
 import DiagnosesControl from '../DiagnosesControl';
 import * as diagnosisService from '@services/diagnosisService';
@@ -102,10 +102,12 @@ describe('DiagnosesControl Visual Regression Tests', () => {
       mockedGroupDiagnosesByDateAndRecorder.mockReturnValue(mockDiagnosesByDate);
 
       // Act
-      const { container } = renderComponent();
+      const { container, getByText } = renderComponent();
 
       // Wait for data to load
-      await new Promise(resolve => setTimeout(resolve, 100));
+      await waitFor(() => {
+        expect(getByText('Type 2 Diabetes Mellitus')).toBeInTheDocument();
+      });
 
       // Assert - Take screenshot with data
       expect(container).toMatchSnapshot('diagnoses-control-with-data');