Update configureMain to support dynamic story paths and add tests for docs feature

Author: valentinpalkovic

code/lib/create-storybook/src/generators/configure.test.ts

@@ -26,6 +26,7 @@ describe('configureMain', () => {
         name: '@storybook/react-vite',
       },
       frameworkPackage: '@storybook/react-vite',
+      features: [],
     });
 
     const { calls } = vi.mocked(fsp.writeFile).mock;
@@ -50,7 +51,7 @@ describe('configureMain', () => {
     `);
   });
 
-  it('should generate main.ts', async () => {
+  it('should generate main.ts with docs feature', async () => {
     await configureMain({
       language: SupportedLanguage.TYPESCRIPT,
       addons: [],
@@ -60,6 +61,41 @@ describe('configureMain', () => {
         name: '@storybook/react-vite',
       },
       frameworkPackage: '@storybook/react-vite',
+      features: ['docs'],
+    });
+
+    const { calls } = vi.mocked(fsp.writeFile).mock;
+    const [mainConfigPath, mainConfigContent] = calls[0];
+
+    expect(mainConfigPath).toEqual('./.storybook/main.ts');
+    expect(mainConfigContent).toMatchInlineSnapshot(`
+      "import type { StorybookConfig } from '@storybook/react-vite';
+
+      const config: StorybookConfig = {
+        "stories": [
+          "../stories/**/*.stories.@(js|jsx|mjs|ts|tsx)",
+          "../stories/**/*.mdx"
+        ],
+        "addons": [],
+        "framework": {
+          "name": "@storybook/react-vite"
+        }
+      };
+      export default config;"
+    `);
+  });
+
+  it('should generate main.ts without docs feature', async () => {
+    await configureMain({
+      language: SupportedLanguage.TYPESCRIPT,
+      addons: [],
+      prefixes: [],
+      storybookConfigFolder: '.storybook',
+      framework: {
+        name: '@storybook/react-vite',
+      },
+      frameworkPackage: '@storybook/react-vite',
+      features: [],
     });
 
     const { calls } = vi.mocked(fsp.writeFile).mock;
@@ -71,7 +107,6 @@ describe('configureMain', () => {
 
       const config: StorybookConfig = {
         "stories": [
-          "../stories/**/*.mdx",
           "../stories/**/*.stories.@(js|jsx|mjs|ts|tsx)"
         ],
         "addons": [],
@@ -96,6 +131,7 @@ describe('configureMain', () => {
         name: "%%path.dirname(require.resolve(path.join('@storybook/react-webpack5', 'package.json')))%%",
       },
       frameworkPackage: '@storybook/react-webpack5',
+      features: ['docs'],
     });
 
     const { calls } = vi.mocked(fsp.writeFile).mock;
@@ -108,8 +144,8 @@ describe('configureMain', () => {
       /** @type { import('@storybook/react-webpack5').StorybookConfig } */
       const config = {
         "stories": [
-          "../stories/**/*.mdx",
-          "../stories/**/*.stories.@(js|jsx|mjs|ts|tsx)"
+          "../stories/**/*.stories.@(js|jsx|mjs|ts|tsx)",
+          "../stories/**/*.mdx"
         ],
         "addons": [
           path.dirname(require.resolve(path.join('@storybook/addon-essentials', 'package.json'))),

code/lib/create-storybook/src/generators/configure.ts

@@ -56,7 +56,7 @@ export async function configureMain({
 }: ConfigureMainOptions) {
   const srcPath = resolve(storybookConfigFolder, '../src');
   const prefix = (await pathExists(srcPath)) ? '../src' : '../stories';
-  const stories = [`**/*.stories.@(${extensions.join('|')})`];
+  const stories = [`${prefix}/**/*.stories.@(${extensions.join('|')})`];
 
   if (features.includes('docs')) {
     stories.push(`${prefix}/**/*.mdx`);