From e131f748f9265ec2ddd589284262b21809500b74 Mon Sep 17 00:00:00 2001
From: Kibana Machine <42973632+kibanamachine@users.noreply.github.com>
Date: Thu, 18 Nov 2021 12:08:58 -0500
Subject: [PATCH] Lens tests: heatmap fix (#119010) (#119034)

Co-authored-by: Joe Reuter <johannes.reuter@elastic.co>
---
 .../shared_components/coloring/palette_panel_container.tsx | 1 +
 x-pack/test/functional/page_objects/lens_page.ts           | 7 ++++++-
 2 files changed, 7 insertions(+), 1 deletion(-)

diff --git a/x-pack/plugins/lens/public/shared_components/coloring/palette_panel_container.tsx b/x-pack/plugins/lens/public/shared_components/coloring/palette_panel_container.tsx
index b546ffe5fb6fb4..abcd714b3af97c 100644
--- a/x-pack/plugins/lens/public/shared_components/coloring/palette_panel_container.tsx
+++ b/x-pack/plugins/lens/public/shared_components/coloring/palette_panel_container.tsx
@@ -56,6 +56,7 @@ export function PalettePanelContainer({
           <div
             role="dialog"
             aria-labelledby="lnsPalettePanelContainerTitle"
+            data-test-subj="lns-indexPattern-PalettePanelContainer"
             className="lnsPalettePanelContainer"
           >
             <EuiFlyoutHeader hasBorder className="lnsPalettePanelContainer__header">
diff --git a/x-pack/test/functional/page_objects/lens_page.ts b/x-pack/test/functional/page_objects/lens_page.ts
index fa696fed7670d3..0cbc1465e68db3 100644
--- a/x-pack/test/functional/page_objects/lens_page.ts
+++ b/x-pack/test/functional/page_objects/lens_page.ts
@@ -871,7 +871,12 @@ export function LensPageProvider({ getService, getPageObjects }: FtrProviderCont
     },
 
     async openPalettePanel(chartType: string) {
-      await testSubjects.click(`${chartType}_dynamicColoring_trigger`);
+      await retry.try(async () => {
+        await testSubjects.click(`${chartType}_dynamicColoring_trigger`);
+        // wait for the UI to settle
+        await PageObjects.common.sleep(100);
+        await testSubjects.existOrFail('lns-indexPattern-PalettePanelContainer', { timeout: 2500 });
+      });
     },
 
     async closePalettePanel() {