@immense/vue-pom-generator 1.0.37 → 1.0.38

package/dist/index.mjs

@@ -3478,7 +3478,8 @@ async function generateFiles(componentHierarchyMap, vueFilesPathMap, basePageCla
     const fixtureRegistryFile = maybeGenerateFixtureRegistry(componentHierarchyMap, {
       generateFixtures,
       pomOutDir: outDir,
-      projectRoot
+      projectRoot,
+      customPomDir
     });
     if (fixtureRegistryFile) {
       writeGeneratedFile(fixtureRegistryFile);
@@ -3877,6 +3878,9 @@ function maybeGenerateFixtureRegistry(componentHierarchyMap, options) {
   const fixtureFileName = looksLikeFilePath ? path.basename(fixtureOutRel) : "fixtures.g.ts";
   const root = options.projectRoot ?? process.cwd();
   const fixtureOutDirAbs = path.isAbsolute(fixtureOutDirRel) ? fixtureOutDirRel : path.resolve(root, fixtureOutDirRel);
+  const customPomDirRel = options.customPomDir ?? "tests/playwright/pom/custom";
+  const customPomDirAbs = path.isAbsolute(customPomDirRel) ? customPomDirRel : path.resolve(root, customPomDirRel);
+  const overridePomDirAbs = path.resolve(path.dirname(customPomDirAbs), "overrides");
   const pomDirAbs = path.isAbsolute(pomOutDir) ? pomOutDir : path.resolve(root, pomOutDir);
   const pomImport = toPosixRelativePath(fixtureOutDirAbs, pomDirAbs);
   const viewClassNames = Array.from(componentHierarchyMap.entries()).filter(([, deps]) => !!deps.isView).map(([name]) => name).sort((a, b) => a.localeCompare(b));
@@ -3899,6 +3903,22 @@ function maybeGenerateFixtureRegistry(componentHierarchyMap, options) {
       return false;
     return true;
   }).sort((a, b) => a.localeCompare(b));
+  const fixtureClassNames = [...viewClassNames, ...componentClassNames];
+  const overrideCtorEntries = fixtureClassNames.map((name) => {
+    const overrideFilePath = path.join(overridePomDirAbs, `${name}.ts`);
+    if (!fs.existsSync(overrideFilePath))
+      return null;
+    return {
+      className: name,
+      localIdentifier: `${name}Override`,
+      importSpecifier: stripExtension(toPosixRelativePath(fixtureOutDirAbs, overrideFilePath))
+    };
+  }).filter((entry) => !!entry);
+  const overrideCtorByClassName = new Map(overrideCtorEntries.map((entry) => [entry.className, entry.localIdentifier]));
+  const overrideImports = overrideCtorEntries.length ? `${overrideCtorEntries.map((entry) => `import { ${entry.className} as ${entry.localIdentifier} } from "${entry.importSpecifier}";`).join("\n")}
+
+` : "";
+  const fixtureCtorExpression = (name) => overrideCtorByClassName.get(name) ?? `Pom.${name}`;
   const header = `${eslintSuppressionHeader}/**
  * DO NOT MODIFY BY HAND
  *
@@ -3907,8 +3927,8 @@ function maybeGenerateFixtureRegistry(componentHierarchyMap, options) {
  */
 
 `;
-  const fixturesTypeEntries = viewClassNames.map((name) => `  ${lowerFirst(name)}: Pom.${name},`).join("\n");
-  const componentFixturesTypeEntries = componentClassNames.map((name) => `  ${lowerFirst(name)}: Pom.${name},`).join("\n");
+  const fixturesTypeEntries = viewClassNames.map((name) => `  ${lowerFirst(name)}: ${fixtureCtorExpression(name)},`).join("\n");
+  const componentFixturesTypeEntries = componentClassNames.map((name) => `  ${lowerFirst(name)}: ${fixtureCtorExpression(name)},`).join("\n");
   const pomFactoryType = `export type PomConstructor<T> = new (page: PwPage) => T;
 
 export interface PomFactory {
@@ -3921,8 +3941,7 @@ export interface PomFactory {
 import { expect, test as base } from "@playwright/test";
 import type { Page as PwPage } from "@playwright/test";
 import * as Pom from "${pomImport}";
-
-export interface PlaywrightOptions {
+${overrideImports}export interface PlaywrightOptions {
   animation: Pom.PlaywrightAnimationOptions;
 }