Fix test type

storybookjs · Jun 28, 2024 · b9f4f04 · b9f4f04
1 parent baad958
commit b9f4f04
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/code/lib/preview-api/src/modules/store/csf/prepareStory.test.ts b/code/lib/preview-api/src/modules/store/csf/prepareStory.test.ts
@@ -49,15 +49,16 @@ export function prepareStory<TRenderer extends Renderer>(
 const addExtraContext = (
   context: PreparedStory & Pick<StoryContext, 'args' | 'globals'>
 ): StoryContext => {
-  const extraContext = {
+  const extraContext: StoryContext = {
     ...context,
     hooks: new HooksContext(),
     viewMode: 'story' as const,
     loaded: {},
     abortSignal: new AbortController().signal,
     canvasElement: {},
     step: vi.fn(),
-    context: undefined! as StoryContext,
+    context: null! as StoryContext,
+    canvas: null!,
   };
   extraContext.context = extraContext;
   return extraContext;