Update useSubscription tests to adjust for changes in render stream l…

…ibrary
apollographql · Dec 4, 2024 · f38516f · f38516f
1 parent 60d5a0e
commit f38516f
Showing 1 changed file with 8 additions and 4 deletions.
diff --git a/src/react/hooks/__tests__/useSubscription.test.tsx b/src/react/hooks/__tests__/useSubscription.test.tsx
@@ -2110,7 +2110,8 @@ describe("data masking", () => {
       link,
     });
 
-    const { takeSnapshot } = renderHookToSnapshotStream(
+    using _disabledAct = disableActEnvironment();
+    const { takeSnapshot } = await renderHookToSnapshotStream(
       () => useSubscription(subscription),
       {
         wrapper: ({ children }) => (
@@ -2178,7 +2179,8 @@ describe("data masking", () => {
       link,
     });
 
-    const { takeSnapshot } = renderHookToSnapshotStream(
+    using _disabledAct = disableActEnvironment();
+    const { takeSnapshot } = await renderHookToSnapshotStream(
       () => useSubscription(subscription),
       {
         wrapper: ({ children }) => (
@@ -2249,7 +2251,8 @@ describe("data masking", () => {
     });
 
     const onData = jest.fn();
-    const { takeSnapshot } = renderHookToSnapshotStream(
+    using _disabledAct = disableActEnvironment();
+    const { takeSnapshot } = await renderHookToSnapshotStream(
       () => useSubscription(subscription, { onData }),
       {
         wrapper: ({ children }) => (
@@ -2329,7 +2332,8 @@ describe("data masking", () => {
     });
 
     const onData = jest.fn();
-    const { takeSnapshot } = renderHookToSnapshotStream(
+    using _disabledAct = disableActEnvironment();
+    const { takeSnapshot } = await renderHookToSnapshotStream(
       () => useSubscription(subscription, { onData }),
       {
         wrapper: ({ children }) => (