allow usage of beforeAll in FTR configs

elastic · Oct 31, 2023 · 9fc6da6 · 9fc6da6
1 parent a35f91e
commit 9fc6da6
Show file tree

Hide file tree

Showing 2 changed files with 19 additions and 0 deletions.
diff --git a/packages/kbn-test/src/functional_test_runner/lib/config/schema.ts b/packages/kbn-test/src/functional_test_runner/lib/config/schema.ts
@@ -146,6 +146,9 @@ export const schema = Joi.object()
         slow: Joi.number().default(30000),
         timeout: Joi.number().default(INSPECTING ? 360000 * 100 : 360000),
         ui: Joi.string().default('bdd'),
+        rootHooks: Joi.object().keys({
+          beforeAll: Joi.func(),
+        }),
       })
       .default(),
 

diff --git a/packages/kbn-test/src/functional_test_runner/lib/mocha/setup_mocha.ts b/packages/kbn-test/src/functional_test_runner/lib/mocha/setup_mocha.ts
@@ -20,6 +20,7 @@ import { Lifecycle } from '../lifecycle';
 import { Config } from '../config';
 import { ProviderCollection } from '../providers';
 import { EsVersion } from '../es_version';
+import { GenericFtrProviderContext } from '../../public_types';
 
 import { MochaReporterProvider } from './reporter';
 import { validateCiGroupTags } from './validate_ci_group_tags';
@@ -47,9 +48,24 @@ export async function setupMocha({
   reporter,
   reporterOptions,
 }: Options) {
+  const mochaRootHooks = config.get('mochaOpts.rootHooks');
+  const rootHookCtx: GenericFtrProviderContext<any, any> = {
+    loadTestFile: () => {
+      throw new Error('loadTestFile is unsupported in root hooks');
+    },
+    getService: providers.getService as any,
+    hasService: providers.hasService as any,
+    getPageObject: providers.getPageObject as any,
+    getPageObjects: providers.getPageObjects as any,
+    updateBaselines: config.get('updateBaselines'),
+  };
+
   // configure mocha
   const mocha = new Mocha({
     ...config.get('mochaOpts'),
+    rootHooks: {
+      beforeAll: () => mochaRootHooks?.beforeAll?.(rootHookCtx),
+    },
     reporter:
       reporter || (await providers.loadExternalService('mocha reporter', MochaReporterProvider)),
     reporterOptions,