From 7eac7b7879a43eb917d83339bfa487c3b5e53fdf Mon Sep 17 00:00:00 2001
From: Andrew Clark <acdlite@fb.com>
Date: Thu, 14 Sep 2017 11:21:37 -0700
Subject: [PATCH] Push host root context when bailing out on low priority

Prevents a push/pop mismatch when bailing out on HostRoots. This is
currently unobservable, because HostRoots never bail out on low
priority, but this does happen with prerendering.

I found this when rebasing #10624 on top of master.
---
 src/renderers/shared/fiber/ReactFiberBeginWork.js | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/src/renderers/shared/fiber/ReactFiberBeginWork.js b/src/renderers/shared/fiber/ReactFiberBeginWork.js
index 9bb7c44adad8c..3c1bac8183b27 100644
--- a/src/renderers/shared/fiber/ReactFiberBeginWork.js
+++ b/src/renderers/shared/fiber/ReactFiberBeginWork.js
@@ -686,6 +686,9 @@ module.exports = function<T, P, I, TI, PI, C, CX, PL>(
     // TODO: Handle HostComponent tags here as well and call pushHostContext()?
     // See PR 8590 discussion for context
     switch (workInProgress.tag) {
+      case HostRoot:
+        pushHostRootContext(workInProgress);
+        break;
       case ClassComponent:
         pushContextProvider(workInProgress);
         break;