fix: scenario hooks dont store worldFixture

src/generate/file.ts

@@ -307,10 +307,12 @@ export class TestFile {
       });
     });
 
+    this.hooks.getWorldFixtureNames().forEach((name) => worldFixtureNames.add(name));
+
     if (worldFixtureNames.size > 1) {
       throw new Error(
         [
-          `All steps in a feature file should have the same worldFixture.`,
+          `All steps and hooks in a feature file should have the same worldFixture.`,
           `Found fixtures: ${[...worldFixtureNames].join(', ')}`,
           `File: ${this.featureUri}`,
         ].join('\n'),

src/generate/hooks.ts

@@ -60,6 +60,13 @@ export class TestFileHooks {
     ]);
   }
 
+  getWorldFixtureNames() {
+    return new Set([
+      ...this.before.getWorldFixtureNames(), // prettier-ignore
+      ...this.after.getWorldFixtureNames(),
+    ]);
+  }
+
   render() {
     const lines = [
       ...this.beforeAll.render(), // prettier-ignore
@@ -96,6 +103,10 @@ class TestFileScenarioHooks<T extends ScenarioHookType> {
   getFixtureNames() {
     return getScenarioHooksFixtureNames([...this.hooks]);
   }
+
+  getWorldFixtureNames() {
+    return new Set([...this.hooks].map((hook) => hook.worldFixture).filter(toBoolean));
+  }
 }
 
 class TestFileWorkerHooks<T extends WorkerHookType> {

src/hooks/scenario.ts

@@ -37,6 +37,7 @@ type ScenarioHook<Fixtures, World> = {
   location: PlaywrightLocation;
   customTest?: TestTypeCommon;
   defaultTags?: string;
+  worldFixture?: string;
 };
 
 /**
@@ -64,7 +65,7 @@ export function scenarioHookFactory<
   TestFixtures extends KeyValue,
   WorkerFixtures extends KeyValue,
   World,
->(type: ScenarioHookType, { customTest, defaultTags }: HookConstructorOptions) {
+>(type: ScenarioHookType, { customTest, defaultTags, worldFixture }: HookConstructorOptions) {
   type AllFixtures = TestFixtures & WorkerFixtures;
   type Args = ScenarioHookDefinitionArgs<AllFixtures, World>;
 
@@ -77,6 +78,7 @@ export function scenarioHookFactory<
       location: getLocationByOffset(3),
       customTest,
       defaultTags,
+      worldFixture,
     });
   };
 }

test/world-ambiguos/only-steps/fixtures.ts

@@ -2,4 +2,5 @@ import { test as base } from 'playwright-bdd';
 
 export const test = base.extend<{ world1: string; world2: string }>({
   world1: async ({}, use) => use('world1'),
+  world2: async ({}, use) => use('world2'),
 });

test/world-ambiguos/playwright.config.ts

@@ -2,7 +2,7 @@ import { defineConfig } from '@playwright/test';
 import { defineBddConfig } from 'playwright-bdd';
 
 const testDir = defineBddConfig({
-  featuresRoot: 'only-steps',
+  featuresRoot: process.env.FEATURES_ROOT!,
 });
 
 export default defineConfig({

test/world-ambiguos/scenario-hooks/fixtures.ts

@@ -0,0 +1,6 @@
+import { test as base } from 'playwright-bdd';
+
+export const test = base.extend<{ world1: string; world2: string }>({
+  world1: async ({}, use) => use('world1'),
+  world2: async ({}, use) => use('world2'),
+});

test/world-ambiguos/scenario-hooks/hooks.ts

@@ -0,0 +1,6 @@
+import { createBdd } from 'playwright-bdd';
+import { test } from './fixtures';
+
+const { Before } = createBdd(test, { worldFixture: 'world2' });
+
+Before(async function () {});

test/world-ambiguos/scenario-hooks/sample.feature

@@ -0,0 +1,4 @@
+Feature: feature
+
+  Scenario: scenario
+    Given step with world1

test/world-ambiguos/scenario-hooks/steps.ts

@@ -0,0 +1,6 @@
+import { createBdd } from 'playwright-bdd';
+import { test } from './fixtures';
+
+const { Given } = createBdd(test, { worldFixture: 'world1' });
+
+Given('step with world1', async function () {});

test/world-ambiguos/test.mjs

@@ -2,9 +2,24 @@ import { test, TestDir, execPlaywrightTestWithError } from '../_helpers/index.mj
 
 const testDir = new TestDir(import.meta);
 
-test(testDir.name, () => {
-  execPlaywrightTestWithError(testDir.name, [
-    'All steps in a feature file should have the same worldFixture',
-    'Found fixtures: world1, world2',
-  ]);
+test(`${testDir.name} (only-steps)`, () => {
+  execPlaywrightTestWithError(
+    testDir.name,
+    [
+      'All steps and hooks in a feature file should have the same worldFixture',
+      'Found fixtures: world1, world2',
+    ],
+    { env: { FEATURES_ROOT: 'only-steps' } },
+  );
+});
+
+test(`${testDir.name} (scenario-hooks)`, () => {
+  execPlaywrightTestWithError(
+    testDir.name,
+    [
+      'All steps and hooks in a feature file should have the same worldFixture',
+      'Found fixtures: world1, world2',
+    ],
+    { env: { FEATURES_ROOT: 'scenario-hooks' } },
+  );
 });