diff --git a/lib/preview-web/src/Preview.tsx b/lib/preview-web/src/Preview.tsx index fb57fc374a70..1a38a9eec9f9 100644 --- a/lib/preview-web/src/Preview.tsx +++ b/lib/preview-web/src/Preview.tsx @@ -252,7 +252,7 @@ export class Preview { 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 index dcb051002edd..1be0dd9e85aa 100644 --- 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' }, }); }); });