Merge pull request #28876 from storybookjs/norbert/vite-hmr-fix

Vite: Fix HMR

code/builders/builder-vite/src/plugins/code-generator-plugin.ts

@@ -70,33 +70,33 @@ export function codeGeneratorPlugin(options: Options): Plugin {
     },
     resolveId(source) {
       if (source === virtualFileId) {
-        return `\0${virtualFileId}`;
+        return `${virtualFileId}`;
       }
       if (source === iframePath) {
         return iframeId;
       }
       if (source === virtualStoriesFile) {
-        return `\0${virtualStoriesFile}`;
+        return `${virtualStoriesFile}`;
       }
       if (source === virtualPreviewFile) {
         return virtualPreviewFile;
       }
       if (source === virtualAddonSetupFile) {
-        return `\0${virtualAddonSetupFile}`;
+        return `${virtualAddonSetupFile}`;
       }
 
       return undefined;
     },
     async load(id, config) {
-      if (id === `\0${virtualStoriesFile}`) {
+      if (id === `${virtualStoriesFile}`) {
         return generateImportFnScriptCode(options);
       }
 
-      if (id === `\0${virtualAddonSetupFile}`) {
+      if (id === `${virtualAddonSetupFile}`) {
         return generateAddonSetupCode();
       }
 
-      if (id === `\0${virtualFileId}`) {
+      if (id === `${virtualFileId}`) {
         return generateModernIframeScriptCode(options, projectRoot);
       }
 

code/e2e-tests/save-from-controls.spec.ts

@@ -1,4 +1,4 @@
-import { test } from '@playwright/test';
+import { expect, test } from '@playwright/test';
 import process from 'process';
 
 import { SbPage } from './util';
@@ -68,5 +68,8 @@ test.describe('save-from-controls', () => {
     const notification2 = await sbPage.page.waitForSelector('[title="Story created"]');
     await notification2.isVisible();
     await notification2.click();
+
+    // Assert the Button components is rendered in the preview
+    await expect(sbPage.previewRoot()).toContainText('Button');
   });
 });