fix(workspace-plugin): update import within bundle-size fixtures with…

…in prepare-intial-release generator (#31325)
microsoft · May 15, 2024 · 01a93ee · 01a93ee
1 parent 3855dcb
commit 01a93ee
Show file tree

Hide file tree

Showing 2 changed files with 44 additions and 0 deletions.
diff --git a/tools/workspace-plugin/src/generators/prepare-initial-release/index.spec.ts b/tools/workspace-plugin/src/generators/prepare-initial-release/index.spec.ts
@@ -9,6 +9,7 @@ import {
   readJson,
   updateJson,
   installPackagesTask,
+  visitNotIgnoredFiles,
 } from '@nrwl/devkit';
 import childProcess from 'child_process';
 
@@ -251,6 +252,18 @@ describe('prepare-initial-release generator', () => {
 
                 export {Two} from './two';
                 export type {TwoType} from './two';
+          `,
+            },
+            {
+              filePath: 'packages/react-one-preview/bundle-size/index.fixture.js',
+              content: stripIndents`
+                import {One} from '@proj/react-one-preview';
+
+                console.log(One);
+
+                export default {
+                   name: '@proj/react-one-preview - package',
+                }
           `,
             },
             {
@@ -328,6 +341,17 @@ describe('prepare-initial-release generator', () => {
           }),
         );
         expect(utils.project.jest()).toEqual(expect.stringContaining(`displayName: 'react-one'`));
+        expect(utils.project.bundleSize()).toMatchInlineSnapshot(`
+          Object {
+            "packages/react-one/bundle-size/index.fixture.js": "import { One } from '@proj/react-one';
+
+          console.log(One);
+
+          export default {
+          name: '@proj/react-one - package',
+          };",
+          }
+        `);
         expect(utils.project.md.readme()).toMatchInlineSnapshot(`
           "# @proj/react-one
 
@@ -606,6 +630,16 @@ These are not production-ready components and **should never be used in product*
       readme: () => tree.read(joinPathFragments(newRoot, 'README.md'), 'utf-8'),
       api: () => tree.read(joinPathFragments(newRoot, `etc/${projectName.replace('-preview', '')}.api.md`), 'utf-8'),
     },
+    bundleSize: () => {
+      const root = joinPathFragments(newRoot, 'bundle-size');
+      const contents: Record<string, string> = {};
+
+      visitNotIgnoredFiles(tree, root, file => {
+        contents[file] = stripIndents`${tree.read(file, 'utf-8')}` ?? '';
+      });
+
+      return contents;
+    },
     global: {
       tsBase: () => readJson<TsConfig>(tree, tsConfigBasePath),
       tsBaseAll: () => readJson<TsConfig>(tree, tsConfigBaseAllPath),

diff --git a/tools/workspace-plugin/src/generators/prepare-initial-release/index.ts b/tools/workspace-plugin/src/generators/prepare-initial-release/index.ts
@@ -124,6 +124,16 @@ async function stableRelease(tree: Tree, options: NormalizedSchema) {
 
   updateFileContent(tree, { filePath: options.paths.jestConfig, updater: contentNameUpdater });
 
+  const bundleSizeFixturesRoot = joinPathFragments(options.projectConfig.root, 'bundle-size');
+  if (tree.exists(bundleSizeFixturesRoot)) {
+    visitNotIgnoredFiles(tree, bundleSizeFixturesRoot, filePath => {
+      updateFileContent(tree, {
+        filePath,
+        updater: contentNameUpdater,
+      });
+    });
+  }
+
   const mdFilePath = {
     readme: joinPathFragments(options.projectConfig.root, 'README.md'),
     api: joinPathFragments(options.projectConfig.root, 'etc', options.normalizedPkgName + '.api.md'),