From 3dc25b31da6f699c7e978ddd3f8ea377e4d5c3e9 Mon Sep 17 00:00:00 2001
From: Tom Gasson <tgasson@atlassian.com>
Date: Tue, 27 Sep 2022 13:35:35 +1000
Subject: [PATCH] [react-relay]: Export ProfilerContext

To match https://github.com/facebook/relay/pull/4049
---
 types/react-relay/hooks.d.ts                 |  2 ++
 types/react-relay/test/react-relay-tests.tsx | 15 +++++++++++++++
 2 files changed, 17 insertions(+)

diff --git a/types/react-relay/hooks.d.ts b/types/react-relay/hooks.d.ts
index 3a8e3be96b7d34..1480baf99a6ba0 100644
--- a/types/react-relay/hooks.d.ts
+++ b/types/react-relay/hooks.d.ts
@@ -62,6 +62,8 @@ export { requestSubscription } from 'relay-runtime';
 export { RelayEnvironmentProvider } from './relay-hooks/RelayEnvironmentProvider.react';
 export { EntryPointContainer } from './relay-hooks/EntryPointContainer.react';
 
+export { ProfilerContext } from './relay-hooks/ProfilerContext';
+
 export { fetchQuery } from 'relay-runtime';
 
 export { loadQuery } from './relay-hooks/loadQuery';
diff --git a/types/react-relay/test/react-relay-tests.tsx b/types/react-relay/test/react-relay-tests.tsx
index 96f70599d0babf..0303e1ec4adaed 100644
--- a/types/react-relay/test/react-relay-tests.tsx
+++ b/types/react-relay/test/react-relay-tests.tsx
@@ -26,6 +26,7 @@ import {
     RelayProp,
     RelayRefetchProp,
     requestSubscription,
+    ProfilerContext,
 } from 'react-relay';
 
 // ~~~~~~~~~~~~~~~~~~~~~
@@ -759,3 +760,17 @@ const MyRelayContextConsumer: React.FunctionComponent = () => {
         />
     );
 };
+
+const MyRelayProfilerContextProvider: React.FunctionComponent = () => {
+    const context = React.useMemo(
+        () => ({
+            wrapPrepareQueryResource<T>(cb: () => T): T {
+                return cb();
+            },
+        }),
+        [],
+    );
+    return (
+        <ProfilerContext.Provider value={context}><div /></ProfilerContext.Provider>
+    );
+};