diff --git a/packages/react-dom/src/__tests__/ReactDOMFizzServerBrowser-test.js b/packages/react-dom/src/__tests__/ReactDOMFizzServerBrowser-test.js index 87d04bdd07655..93e30aa2305be 100644 --- a/packages/react-dom/src/__tests__/ReactDOMFizzServerBrowser-test.js +++ b/packages/react-dom/src/__tests__/ReactDOMFizzServerBrowser-test.js @@ -105,12 +105,10 @@ describe('ReactDOMFizzServer', () => { , - { - onCompleteAll() { - isComplete = true; - }, - }, ); + + stream.allReady.then(() => (isComplete = true)); + await jest.runAllTimers(); expect(isComplete).toBe(false); // Resolve the loading. diff --git a/packages/react-dom/src/server/ReactDOMFizzServerBrowser.js b/packages/react-dom/src/server/ReactDOMFizzServerBrowser.js index a07967ca34f98..bf90682d3c6a5 100644 --- a/packages/react-dom/src/server/ReactDOMFizzServerBrowser.js +++ b/packages/react-dom/src/server/ReactDOMFizzServerBrowser.js @@ -32,23 +32,36 @@ type Options = {| bootstrapModules?: Array, progressiveChunkSize?: number, signal?: AbortSignal, - onCompleteAll?: () => void, onError?: (error: mixed) => void, |}; +// TODO: Move to sub-classing ReadableStream. +type ReactDOMServerReadableStream = ReadableStream & { + allReady: Promise, +}; + function renderToReadableStream( children: ReactNodeList, options?: Options, -): Promise { +): Promise { return new Promise((resolve, reject) => { + let onFatalError; + let onCompleteAll; + const allReady = new Promise((res, rej) => { + onCompleteAll = res; + onFatalError = rej; + }); + function onCompleteShell() { - const stream = new ReadableStream({ + const stream: ReactDOMServerReadableStream = (new ReadableStream({ type: 'bytes', pull(controller) { startFlowing(request, controller); }, cancel(reason) {}, - }); + }): any); + // TODO: Move to sub-classing ReadableStream. + stream.allReady = allReady; resolve(stream); } function onErrorShell(error: mixed) { @@ -66,9 +79,10 @@ function renderToReadableStream( createRootFormatContext(options ? options.namespaceURI : undefined), options ? options.progressiveChunkSize : undefined, options ? options.onError : undefined, - options ? options.onCompleteAll : undefined, + onCompleteAll, onCompleteShell, onErrorShell, + onFatalError, ); if (options && options.signal) { const signal = options.signal; diff --git a/packages/react-dom/src/server/ReactDOMFizzServerNode.js b/packages/react-dom/src/server/ReactDOMFizzServerNode.js index 15e2636187065..39126ed8d4266 100644 --- a/packages/react-dom/src/server/ReactDOMFizzServerNode.js +++ b/packages/react-dom/src/server/ReactDOMFizzServerNode.js @@ -65,6 +65,7 @@ function createRequestImpl(children: ReactNodeList, options: void | Options) { options ? options.onCompleteAll : undefined, options ? options.onCompleteShell : undefined, options ? options.onErrorShell : undefined, + undefined, ); } diff --git a/packages/react-dom/src/server/ReactDOMLegacyServerBrowser.js b/packages/react-dom/src/server/ReactDOMLegacyServerBrowser.js index 1a11911ad5317..778806c9703d7 100644 --- a/packages/react-dom/src/server/ReactDOMLegacyServerBrowser.js +++ b/packages/react-dom/src/server/ReactDOMLegacyServerBrowser.js @@ -67,6 +67,8 @@ function renderToStringImpl( onError, undefined, onCompleteShell, + undefined, + undefined, ); startWork(request); // If anything suspended and is still pending, we'll abort it before writing. diff --git a/packages/react-dom/src/server/ReactDOMLegacyServerNode.js b/packages/react-dom/src/server/ReactDOMLegacyServerNode.js index fd55b2b0a5ceb..7840708c5ad50 100644 --- a/packages/react-dom/src/server/ReactDOMLegacyServerNode.js +++ b/packages/react-dom/src/server/ReactDOMLegacyServerNode.js @@ -83,6 +83,7 @@ function renderToNodeStreamImpl( onError, onCompleteAll, undefined, + undefined, ); destination.request = request; startWork(request); diff --git a/packages/react-server/src/ReactFizzServer.js b/packages/react-server/src/ReactFizzServer.js index a7f5ad8346939..306926503d756 100644 --- a/packages/react-server/src/ReactFizzServer.js +++ b/packages/react-server/src/ReactFizzServer.js @@ -204,6 +204,7 @@ export opaque type Request = { // onErrorShell is called when the shell didn't complete. That means you probably want to // emit a different response to the stream instead. onErrorShell: (error: mixed) => void, + onFatalError: (error: mixed) => void, }; // This is a default heuristic for how to split up the HTML content into progressive @@ -238,6 +239,7 @@ export function createRequest( onCompleteAll: void | (() => void), onCompleteShell: void | (() => void), onErrorShell: void | ((error: mixed) => void), + onFatalError: void | ((error: mixed) => void), ): Request { const pingedTasks = []; const abortSet: Set = new Set(); @@ -263,6 +265,7 @@ export function createRequest( onCompleteAll: onCompleteAll === undefined ? noop : onCompleteAll, onCompleteShell: onCompleteShell === undefined ? noop : onCompleteShell, onErrorShell: onErrorShell === undefined ? noop : onErrorShell, + onFatalError: onFatalError === undefined ? noop : onFatalError, }; // This segment represents the root fallback. const rootSegment = createPendingSegment(request, 0, null, rootFormatContext);