reset all args to its initial value when onResetArgs is triggerd with…

…out argNames
storybookjs · May 15, 2022 · b318c25 · b318c25
1 parent 385b7d7
commit b318c25
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/lib/preview-web/src/Preview.tsx b/lib/preview-web/src/Preview.tsx
@@ -252,7 +252,7 @@ export class Preview<TFramework extends AnyFramework> {
     const render = this.storyRenders.find((r) => r.id === storyId);
     const story = render?.story || (await this.storyStore.loadStory({ storyId }));
 
-    const argNamesToReset = argNames || Object.keys(this.storyStore.args.get(storyId));
+    const argNamesToReset = argNames || Object.keys(story.initialArgs);
     const updatedArgs = argNamesToReset.reduce((acc, argName) => {
       acc[argName] = story.initialArgs[argName];
       return acc;

diff --git a/lib/preview-web/src/PreviewWeb.test.ts b/lib/preview-web/src/PreviewWeb.test.ts
@@ -1124,7 +1124,7 @@ describe('PreviewWeb', () => {
           forceRemount: false,
           storyContext: expect.objectContaining({
             initialArgs: { foo: 'a' },
-            args: { foo: 'a' },
+            args: { foo: 'a', new: 'value' },
           }),
         }),
         undefined // this is coming from view.prepareForStory, not super important
@@ -1133,7 +1133,7 @@ describe('PreviewWeb', () => {
       await waitForEvents([Events.STORY_ARGS_UPDATED]);
       expect(mockChannel.emit).toHaveBeenCalledWith(Events.STORY_ARGS_UPDATED, {
         storyId: 'component-one--a',
-        args: { foo: 'a' },
+        args: { foo: 'a', new: 'value' },
       });
     });
   });