test(heatmap-chat-web): change to wait for colorbar element

Author: leonardomendix

packages/pluggableWidgets/heatmap-chart-web/e2e/HeatMapChart.spec.js

@@ -16,7 +16,8 @@ test.describe("heatmap-chart-web", () => {
             const customColorContainer = await page.locator(".mx-name-containerCustomColor");
             await expect(customColorContainer).toBeVisible({ timeout: 10000 });
             await customColorContainer.scrollIntoViewIfNeeded();
-            await expect(customColorContainer.locator("svg")).toBeVisible({ timeout: 5000 });
+            await expect(customColorContainer.locator(".mx-react-plotly-chart")).toBeVisible({ timeout: 5000 });
+            await expect(customColorContainer.locator("g.colorbar")).toBeVisible({ timeout: 5000 });
             await page.waitForTimeout(500);
             await expect(customColorContainer).toHaveScreenshot(`heatmapChartCustomColor.png`, {
                 threshold: 0.5
@@ -29,7 +30,8 @@ test.describe("heatmap-chart-web", () => {
             const ascendingContainer = await page.locator(".mx-name-containerAscending");
             await expect(ascendingContainer).toBeVisible({ timeout: 10000 });
             await ascendingContainer.scrollIntoViewIfNeeded();
-            await expect(ascendingContainer.locator("svg")).toBeVisible({ timeout: 5000 });
+            await expect(ascendingContainer.locator(".mx-react-plotly-chart")).toBeVisible({ timeout: 5000 });
+            await expect(ascendingContainer.locator("g.colorbar")).toBeVisible({ timeout: 5000 });
             await page.waitForTimeout(500);
             await expect(ascendingContainer).toHaveScreenshot(`heatmapChartAscending.png`, {
                 threshold: 0.5
@@ -42,7 +44,8 @@ test.describe("heatmap-chart-web", () => {
             const descendingContainer = await page.locator(".mx-name-containerDescending");
             await expect(descendingContainer).toBeVisible({ timeout: 10000 });
             await descendingContainer.scrollIntoViewIfNeeded();
-            await expect(descendingContainer.locator("svg")).toBeVisible({ timeout: 5000 });
+            await expect(descendingContainer.locator(".mx-react-plotly-chart")).toBeVisible({ timeout: 5000 });
+            await expect(descendingContainer.locator("g.colorbar")).toBeVisible({ timeout: 5000 });
             await page.waitForTimeout(500);
             await expect(descendingContainer).toHaveScreenshot(`heatmapChartDescending.png`, {
                 threshold: 0.5