Use stream assertions

apollographql · Dec 5, 2024 · 3c9ce67 · 3c9ce67
1 parent 29343c1
commit 3c9ce67
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/src/core/__tests__/ObservableQuery.ts b/src/core/__tests__/ObservableQuery.ts
@@ -3706,17 +3706,17 @@ test("does not return partial cache data when `returnPartialData` is false and n
 
   const stream = new ObservableStream(observable);
 
-  expect(await stream.takeNext()).toEqual({
+  await expect(stream).toEmitValue({
     loading: false,
     networkStatus: NetworkStatus.ready,
     data: {
       car: { __typename: "Car", id: 2, make: "Ford", model: "Bronco" },
     },
   });
 
-  observable.reobserve({ variables: { id: 1 } });
+  await observable.reobserve({ variables: { id: 1 } });
 
-  expect(await stream.takeNext()).toEqual({
+  await expect(stream).toEmitValue({
     loading: true,
     networkStatus: NetworkStatus.setVariables,
     data: undefined,
@@ -3728,7 +3728,7 @@ test("does not return partial cache data when `returnPartialData` is false and n
     data: undefined,
   });
 
-  expect(await stream.takeNext()).toEqual({
+  await expect(stream).toEmitValue({
     loading: false,
     networkStatus: NetworkStatus.ready,
     data: { __typename: "Car", id: 1, make: "Ford", model: "Pinto" },