From 57c31e8cd339f8fcdd5b2bc51143e787d0ba8870 Mon Sep 17 00:00:00 2001
From: Joe Reuter <johannes.reuter@elastic.co>
Date: Thu, 22 Oct 2020 18:58:28 +0200
Subject: [PATCH] fix flaky unit test (#81466) (#81496)

---
 .../embeddable/embeddable.test.tsx                | 15 ++++++++++-----
 1 file changed, 10 insertions(+), 5 deletions(-)

diff --git a/x-pack/plugins/lens/public/editor_frame_service/embeddable/embeddable.test.tsx b/x-pack/plugins/lens/public/editor_frame_service/embeddable/embeddable.test.tsx
index d91865c21a2a60..3e05d4ddfbc20d 100644
--- a/x-pack/plugins/lens/public/editor_frame_service/embeddable/embeddable.test.tsx
+++ b/x-pack/plugins/lens/public/editor_frame_service/embeddable/embeddable.test.tsx
@@ -28,6 +28,7 @@ import { IBasePath } from '../../../../../../src/core/public';
 import { AttributeService } from '../../../../../../src/plugins/embeddable/public';
 import { LensAttributeService } from '../../lens_attribute_service';
 import { OnSaveProps } from '../../../../../../src/plugins/saved_objects/public/save_modal';
+import { act } from 'react-dom/test-utils';
 
 jest.mock('../../../../../../src/plugins/inspector/public/', () => ({
   isAvailable: false,
@@ -337,10 +338,12 @@ describe('embeddable', () => {
     } as LensEmbeddableInput);
     embeddable.render(mountpoint);
 
-    embeddable.updateInput({
-      timeRange,
-      query,
-      filters: [{ meta: { alias: 'test', negate: true, disabled: true } }],
+    act(() => {
+      embeddable.updateInput({
+        timeRange,
+        query,
+        filters: [{ meta: { alias: 'test', negate: true, disabled: true } }],
+      });
     });
 
     expect(expressionRenderer).toHaveBeenCalledTimes(1);
@@ -384,7 +387,9 @@ describe('embeddable', () => {
     } as LensEmbeddableInput);
     embeddable.render(mountpoint);
 
-    autoRefreshFetchSubject.next();
+    act(() => {
+      autoRefreshFetchSubject.next();
+    });
 
     expect(expressionRenderer).toHaveBeenCalledTimes(2);
   });