UI: Fix unmount components on Canvas/Docs tab switch (#8625)

UI: Fix unmount components on Canvas/Docs tab switch
storybookjs · Nov 6, 2019 · 4233f3e · 4233f3e · vercel · Nov 6, 2019
2 parents c14f93d + 4975a90
commit 4233f3e
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/lib/core/src/client/preview/start.js b/lib/core/src/client/preview/start.js
@@ -202,12 +202,14 @@ export default function start(render, { decorateStory } = {}) {
       case 'docs':
         if (previousKind != null && (kind !== previousKind || viewMode !== previousViewMode)) {
           storyStore.cleanHooksForKind(previousKind);
+          ReactDOM.unmountComponentAtNode(document.getElementById('docs-root'));
         }
         break;
       case 'story':
       default:
         if (previousId != null && (id !== previousId || viewMode !== previousViewMode)) {
           storyStore.cleanHooks(previousId);
+          ReactDOM.unmountComponentAtNode(document.getElementById('root'));
         }
     }
 
@@ -226,7 +228,6 @@ export default function start(render, { decorateStory } = {}) {
         default: {
           if (previousViewMode === 'docs') {
             document.getElementById('docs-root').setAttribute('hidden', true);
-            ReactDOM.unmountComponentAtNode(document.getElementById('docs-root'));
             document.getElementById('root').removeAttribute('hidden');
           }
         }