Ensure that __DOCS_CONTEXT__ cannot be undefined when using storyshots

storybookjs · Jan 16, 2022 · 37fa89d · 37fa89d
1 parent fb4d3b2
commit 37fa89d
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/addons/docs/src/blocks/DocsContext.ts b/addons/docs/src/blocks/DocsContext.ts
@@ -12,9 +12,11 @@ export type { DocsContextProps };
 // the React component tree.
 // This was specifically a problem with the Vite builder.
 /* eslint-disable no-underscore-dangle */
-if (globalWindow.__DOCS_CONTEXT__ === undefined) {
+if (globalWindow && globalWindow.__DOCS_CONTEXT__ === undefined) {
   globalWindow.__DOCS_CONTEXT__ = createContext({});
   globalWindow.__DOCS_CONTEXT__.displayName = 'DocsContext';
 }
 
-export const DocsContext: Context<DocsContextProps<AnyFramework>> = globalWindow.__DOCS_CONTEXT__;
+export const DocsContext: Context<DocsContextProps<AnyFramework>> = globalWindow
+  ? globalWindow.__DOCS_CONTEXT__
+  : createContext({});