[Discover] Update tests

elastic · Nov 19, 2024 · 8b1719e · 8b1719e
1 parent ca97c38
commit 8b1719e
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 9 deletions.
diff --git a/test/functional/apps/discover/context_awareness/extensions/_get_render_app_wrapper.ts b/test/functional/apps/discover/context_awareness/extensions/_get_render_app_wrapper.ts
@@ -13,12 +13,13 @@ import type { WebElementWrapper } from '@kbn/ftr-common-functional-ui-services';
 import type { FtrProviderContext } from '../../ftr_provider_context';
 
 export default function ({ getService, getPageObjects }: FtrProviderContext) {
-  const { common, discover, header, unifiedFieldList, dashboard } = getPageObjects([
+  const { common, discover, header, unifiedFieldList, dashboard, context } = getPageObjects([
     'common',
     'discover',
     'header',
     'unifiedFieldList',
     'dashboard',
+    'context',
   ]);
   const testSubjects = getService('testSubjects');
   const dataViews = getService('dataViews');
@@ -129,6 +130,7 @@ export default function ({ getService, getPageObjects }: FtrProviderContext) {
         await header.waitUntilLoadingHasFinished();
         await browser.refresh();
         await header.waitUntilLoadingHasFinished();
+        await context.waitUntilContextLoadingHasFinished();
 
         await retry.try(async () => {
           messageCell = await dataGrid.getCellElementByColumnName(0, 'message');

diff --git a/...ional/test_suites/common/discover/context_awareness/extensions/_get_render_app_wrapper.ts b/...ional/test_suites/common/discover/context_awareness/extensions/_get_render_app_wrapper.ts
@@ -11,14 +11,16 @@ import type { WebElementWrapper } from '@kbn/ftr-common-functional-ui-services';
 import type { FtrProviderContext } from '../../../../../ftr_provider_context';
 
 export default function ({ getService, getPageObjects }: FtrProviderContext) {
-  const { common, discover, header, unifiedFieldList, dashboard, svlCommonPage } = getPageObjects([
-    'common',
-    'discover',
-    'header',
-    'unifiedFieldList',
-    'dashboard',
-    'svlCommonPage',
-  ]);
+  const { common, discover, header, unifiedFieldList, dashboard, context, svlCommonPage } =
+    getPageObjects([
+      'common',
+      'discover',
+      'header',
+      'unifiedFieldList',
+      'dashboard',
+      'context',
+      'svlCommonPage',
+    ]);
   const testSubjects = getService('testSubjects');
   const dataViews = getService('dataViews');
   const dataGrid = getService('dataGrid');
@@ -132,6 +134,7 @@ export default function ({ getService, getPageObjects }: FtrProviderContext) {
         await header.waitUntilLoadingHasFinished();
         await browser.refresh();
         await header.waitUntilLoadingHasFinished();
+        await context.waitUntilContextLoadingHasFinished();
 
         await retry.try(async () => {
           messageCell = await dataGrid.getCellElementByColumnName(0, 'message');