Move E2E info message to one that's easier to extend

chromaui · Oct 17, 2024 · d999140 · d999140
1 parent 6b6570b
commit d999140
Showing 1 changed file with 13 additions and 1 deletion.
diff --git a/node-src/ui/tasks/storybookInfo.ts b/node-src/ui/tasks/storybookInfo.ts
@@ -8,9 +8,21 @@ const capitalize = (string: string) =>
     .map((str) => str.charAt(0).toUpperCase() + str.slice(1))
     .join(' ');
 
+const e2eMessage = (ctx: Context) => {
+  if (ctx.options.playwright) {
+    return 'Playwright for E2E';
+  }
+
+  if (ctx.options.cypress) {
+    return 'Cypress for E2E';
+  }
+
+  return 'E2E';
+};
+
 const infoMessage = (ctx: Context) => {
   if (isE2EBuild(ctx.options)) {
-    return ctx.options.playwright ? 'Playwright for E2E' : 'Cypress for E2E';
+    return e2eMessage(ctx);
   }
 
   const { addons, version, viewLayer, builder } = ctx.storybook;