Don't bail to .child of coroutines

This should bail to stateNode instead.

Author: sebmarkbage

scripts/fiber/tests-passing.txt

@@ -1145,6 +1145,7 @@ src/renderers/shared/fiber/__tests__/ReactCoroutine-test.js
 * should render a coroutine
 * should update a coroutine
 * should unmount a composite in a coroutine
+* should handle deep updates in coroutine
 
 src/renderers/shared/fiber/__tests__/ReactIncremental-test.js
 * should render a simple component

src/renderers/shared/fiber/ReactFiberBeginWork.js

@@ -516,7 +516,10 @@ module.exports = function<T, P, I, TI, PI, C, CX, PL>(
         }
       }
     } else if (nextCoroutine === null || workInProgress.memoizedProps === nextCoroutine) {
-      return bailoutOnAlreadyFinishedWork(current, workInProgress);
+      nextCoroutine = workInProgress.memoizedProps;
+      // TODO: When bailing out, we might need to return the stateNode instead
+      // of the child. To check it for work.
+      // return bailoutOnAlreadyFinishedWork(current, workInProgress);
     }
 
     const nextChildren = nextCoroutine.children;

src/renderers/shared/fiber/__tests__/ReactCoroutine-test.js

@@ -224,4 +224,45 @@ describe('ReactCoroutine', () => {
     ]);
 
   });
+
+  it('should handle deep updates in coroutine', () => {
+    let instances = {};
+
+    class Counter extends React.Component {
+      state = {value: 5};
+      render() {
+        instances[this.props.id] = this;
+        return ReactCoroutine.createYield(this.state.value);
+      }
+    }
+
+    function App(props) {
+      return ReactCoroutine.createCoroutine(
+        [
+          <Counter id="a" />,
+          <Counter id="b" />,
+          <Counter id="c" />,
+        ],
+        (p, yields) => yields.map(y => <span prop={y * 100} />),
+        {}
+      );
+    }
+
+    ReactNoop.render(<App />);
+    ReactNoop.flush();
+    expect(ReactNoop.getChildren()).toEqual([
+      span(500),
+      span(500),
+      span(500),
+    ]);
+
+    instances.a.setState({value: 1});
+    instances.b.setState({value: 2});
+    ReactNoop.flush();
+    expect(ReactNoop.getChildren()).toEqual([
+      span(100),
+      span(200),
+      span(500),
+    ]);
+  });
 });