From d930da1ec24c135dfd4a609ac4343a5555e4beec Mon Sep 17 00:00:00 2001
From: Jinto Jose <jintoppy@gmail.com>
Date: Fri, 23 Nov 2018 12:10:25 +0530
Subject: [PATCH] Adding isMemo check to react-is package

---
 packages/react-is/src/ReactIs.js                | 7 ++++++-
 packages/react-is/src/__tests__/ReactIs-test.js | 8 ++++++++
 2 files changed, 14 insertions(+), 1 deletion(-)

diff --git a/packages/react-is/src/ReactIs.js b/packages/react-is/src/ReactIs.js
index 0c1edb811370d..e69486610ccc4 100644
--- a/packages/react-is/src/ReactIs.js
+++ b/packages/react-is/src/ReactIs.js
@@ -16,6 +16,7 @@ import {
   REACT_ELEMENT_TYPE,
   REACT_FORWARD_REF_TYPE,
   REACT_FRAGMENT_TYPE,
+  REACT_MEMO_TYPE,
   REACT_PORTAL_TYPE,
   REACT_PROFILER_TYPE,
   REACT_PROVIDER_TYPE,
@@ -27,7 +28,6 @@ import lowPriorityWarning from 'shared/lowPriorityWarning';
 export function typeOf(object: any) {
   if (typeof object === 'object' && object !== null) {
     const $$typeof = object.$$typeof;
-
     switch ($$typeof) {
       case REACT_ELEMENT_TYPE:
         const type = object.type;
@@ -51,6 +51,7 @@ export function typeOf(object: any) {
                 return $$typeof;
             }
         }
+      case REACT_MEMO_TYPE:
       case REACT_PORTAL_TYPE:
         return $$typeof;
     }
@@ -69,6 +70,7 @@ export const ForwardRef = REACT_FORWARD_REF_TYPE;
 export const Fragment = REACT_FRAGMENT_TYPE;
 export const Profiler = REACT_PROFILER_TYPE;
 export const Portal = REACT_PORTAL_TYPE;
+export const Memo = REACT_MEMO_TYPE;
 export const StrictMode = REACT_STRICT_MODE_TYPE;
 
 export {isValidElementType};
@@ -115,6 +117,9 @@ export function isFragment(object: any) {
 export function isProfiler(object: any) {
   return typeOf(object) === REACT_PROFILER_TYPE;
 }
+export function isMemo(object: any) {
+  return typeOf(object) === REACT_MEMO_TYPE;
+}
 export function isPortal(object: any) {
   return typeOf(object) === REACT_PORTAL_TYPE;
 }
diff --git a/packages/react-is/src/__tests__/ReactIs-test.js b/packages/react-is/src/__tests__/ReactIs-test.js
index 3cf74ea6febb9..aeb1805d2f170 100644
--- a/packages/react-is/src/__tests__/ReactIs-test.js
+++ b/packages/react-is/src/__tests__/ReactIs-test.js
@@ -144,6 +144,14 @@ describe('ReactIs', () => {
     expect(ReactIs.isPortal(div)).toBe(false);
   });
 
+  it('should identify memo', () => {
+    const Component = () => React.createElement('div');
+    const memoized = React.memo(Component);
+    expect(ReactIs.typeOf(memoized)).toBe(ReactIs.Memo);
+    expect(ReactIs.isMemo(memoized)).toBe(true);
+    expect(ReactIs.isMemo(Component)).toBe(false);
+  });
+
   it('should identify strict mode', () => {
     expect(ReactIs.typeOf(<React.StrictMode />)).toBe(ReactIs.StrictMode);
     expect(ReactIs.isStrictMode(<React.StrictMode />)).toBe(true);