From 7c255f190c7cd6eb9eca16247d0d1a894a716715 Mon Sep 17 00:00:00 2001
From: Kevin Yu <kevinwcyu@users.noreply.github.com>
Date: Thu, 17 Mar 2022 06:57:43 -0700
Subject: [PATCH] Chore: clean up console warning from Query Header tests
 (#46663)

---
 .../cloudwatch/components/QueryHeader.test.tsx        | 11 +++++------
 1 file changed, 5 insertions(+), 6 deletions(-)

diff --git a/public/app/plugins/datasource/cloudwatch/components/QueryHeader.test.tsx b/public/app/plugins/datasource/cloudwatch/components/QueryHeader.test.tsx
index 81b1263301e..7809e45c614 100644
--- a/public/app/plugins/datasource/cloudwatch/components/QueryHeader.test.tsx
+++ b/public/app/plugins/datasource/cloudwatch/components/QueryHeader.test.tsx
@@ -1,6 +1,5 @@
 import React from 'react';
-import { render, screen } from '@testing-library/react';
-import { act } from 'react-dom/test-utils';
+import { render, screen, waitFor, act } from '@testing-library/react';
 import { CloudWatchLogsQuery, CloudWatchMetricsQuery, MetricEditorMode, MetricQueryType } from '../types';
 import { setupMockedDataSource } from '../__mocks__/CloudWatchDataSource';
 import QueryHeader from './QueryHeader';
@@ -74,9 +73,9 @@ describe('QueryHeader', () => {
       />
     );
 
-    const builderElement = screen.queryByLabelText('Builder');
-    expect(builderElement).toBeNull();
-    const codeElement = screen.queryByLabelText('Code');
-    expect(codeElement).toBeNull();
+    await waitFor(() => {
+      expect(screen.queryByLabelText('Builder')).toBeNull();
+      expect(screen.queryByLabelText('Code')).toBeNull();
+    });
   });
 });