fix(devtools): preserve compiler badge for nested Forget names

packages/react-devtools-shared/src/__tests__/utils-test.js

@@ -11,7 +11,9 @@ import {
   getDisplayName,
   getDisplayNameForReactElement,
   isPlainObject,
+  parseElementDisplayNameFromBackend,
 } from 'react-devtools-shared/src/utils';
+import {ElementTypeMemo} from 'react-devtools-shared/src/frontend/types';
 import {stackToComponentLocations} from 'react-devtools-shared/src/devtools/utils';
 import {
   formatConsoleArguments,
@@ -82,6 +84,21 @@ describe('utils', () => {
     });
   });
 
+  describe('parseElementDisplayNameFromBackend', () => {
+    it('should identify Forget inside HOC wrappers', () => {
+      expect(
+        parseElementDisplayNameFromBackend(
+          'Memo(Forget(Example))',
+          ElementTypeMemo,
+        ),
+      ).toEqual({
+        formattedDisplayName: 'Example',
+        hocDisplayNames: ['Memo'],
+        compiledWithForget: true,
+      });
+    });
+  });
+
   describe('getDisplayNameForReactElement', () => {
     // @reactVersion >= 16.0
     it('should return correct display name for an element with function type', () => {

packages/react-devtools-shared/src/utils.js

@@ -596,11 +596,25 @@ export function parseElementDisplayNameFromBackend(
       break;
   }
 
+  let compiledWithForget = false;
+  if (hocDisplayNames !== null) {
+    const forgetIndex = hocDisplayNames.indexOf('Forget');
+    if (forgetIndex >= 0) {
+      compiledWithForget = true;
+      hocDisplayNames = hocDisplayNames.filter(
+        hocDisplayName => hocDisplayName !== 'Forget',
+      );
+      if (hocDisplayNames.length === 0) {
+        hocDisplayNames = null;
+      }
+    }
+  }
+
   return {
     // $FlowFixMe[incompatible-return]
     formattedDisplayName: displayName,
     hocDisplayNames,
-    compiledWithForget: false,
+    compiledWithForget,
   };
 }