[Fizz] Add Owner Stacks when render is aborted

Author: eps1lon

Diff for: packages/react-server/src/ReactFizzServer.js

@@ -4151,6 +4151,27 @@ function abortTask(task: Task, request: Request, error: mixed): void {
   }
 }
 
+function abortTaskDEV(task: Task, request: Request, error: mixed): void {
+  if (__DEV__) {
+    const prevTaskInDEV = currentTaskInDEV;
+    const prevGetCurrentStackImpl = ReactSharedInternals.getCurrentStack;
+    setCurrentTaskInDEV(task);
+    ReactSharedInternals.getCurrentStack = getCurrentStackInDEV;
+    try {
+      abortTask(task, request, error);
+    } finally {
+      setCurrentTaskInDEV(prevTaskInDEV);
+      ReactSharedInternals.getCurrentStack = prevGetCurrentStackImpl;
+    }
+  } else {
+    // These errors should never make it into a build so we don't need to encode them in codes.json
+    // eslint-disable-next-line react-internal/prod-error-codes
+    throw new Error(
+      'abortTaskDEV should never be called in production mode. This is a bug in React.',
+    );
+  }
+}
+
 function safelyEmitEarlyPreloads(
   request: Request,
   shellComplete: boolean,
@@ -5373,7 +5394,11 @@ export function abort(request: Request, reason: mixed): void {
       // This error isn't necessarily fatal in this case but we need to stash it
       // so we can use it to abort any pending work
       request.fatalError = error;
-      abortableTasks.forEach(task => abortTask(task, request, error));
+      if (__DEV__) {
+        abortableTasks.forEach(task => abortTaskDEV(task, request, error));
+      } else {
+        abortableTasks.forEach(task => abortTask(task, request, error));
+      }
       abortableTasks.clear();
     }
     if (request.destination !== null) {

Diff for: packages/react-server/src/__tests__/ReactServer-test.js

@@ -83,7 +83,6 @@ describe('ReactServer', () => {
     expect(normalizeCodeLocInfo(componentStack)).toEqual(
       '\n    in Component (at **)' + '\n    in App (at **)',
     );
-    // FIXME: Should have a stack.
-    expect(normalizeCodeLocInfo(ownerStack)).toEqual(null);
+    expect(normalizeCodeLocInfo(ownerStack)).toEqual('\n    in App (at **)');
   });
 });