Update tests

urql-graphql · Mar 16, 2023 · 5314a37 · 5314a37
1 parent cfdac55
commit 5314a37
Show file tree

Hide file tree

Showing 3 changed files with 21 additions and 6 deletions.
diff --git a/packages/vue-urql/src/useMutation.test.ts b/packages/vue-urql/src/useMutation.test.ts
@@ -1,3 +1,4 @@
+import { OperationResult, OperationResultSource } from '@urql/core';
 import { reactive } from 'vue';
 import { vi, expect, it, beforeEach, describe } from 'vitest';
 
@@ -25,7 +26,9 @@ describe('useMutation', () => {
     const subject = makeSubject<any>();
     const clientMutation = vi
       .spyOn(client, 'executeMutation')
-      .mockImplementation(() => subject.source);
+      .mockImplementation(
+        () => subject.source as OperationResultSource<OperationResult>
+      );
 
     const mutation = reactive(
       useMutation(

diff --git a/packages/vue-urql/src/useQuery.test.ts b/packages/vue-urql/src/useQuery.test.ts
@@ -1,3 +1,4 @@
+import { OperationResult, OperationResultSource } from '@urql/core';
 import { nextTick, reactive, ref } from 'vue';
 import { vi, expect, it, describe } from 'vitest';
 
@@ -18,7 +19,9 @@ describe('useQuery', () => {
     const subject = makeSubject<any>();
     const executeQuery = vi
       .spyOn(client, 'executeQuery')
-      .mockImplementation(() => subject.source);
+      .mockImplementation(
+        () => subject.source as OperationResultSource<OperationResult>
+      );
 
     const _query = useQuery({
       query: `{ test }`,
@@ -109,7 +112,9 @@ describe('useQuery', () => {
     const subject = makeSubject<any>();
     const executeQuery = vi
       .spyOn(client, 'executeQuery')
-      .mockImplementation(() => subject.source);
+      .mockImplementation(
+        () => subject.source as OperationResultSource<OperationResult>
+      );
 
     const _query = useQuery({
       query: `{ test }`,

diff --git a/packages/vue-urql/src/useSubscription.test.ts b/packages/vue-urql/src/useSubscription.test.ts
@@ -1,3 +1,4 @@
+import { OperationResult, OperationResultSource } from '@urql/core';
 import { nextTick, reactive, ref } from 'vue';
 import { vi, expect, it, describe } from 'vitest';
 
@@ -18,7 +19,9 @@ describe('useSubscription', () => {
     const subject = makeSubject<any>();
     const executeQuery = vi
       .spyOn(client, 'executeSubscription')
-      .mockImplementation(() => subject.source);
+      .mockImplementation(
+        () => subject.source as OperationResultSource<OperationResult>
+      );
 
     const sub = reactive(
       useSubscription({
@@ -60,7 +63,9 @@ describe('useSubscription', () => {
     const subject = makeSubject<any>();
     const executeSubscription = vi
       .spyOn(client, 'executeSubscription')
-      .mockImplementation(() => subject.source);
+      .mockImplementation(
+        () => subject.source as OperationResultSource<OperationResult>
+      );
 
     const variables = ref({});
     const sub = reactive(
@@ -101,7 +106,9 @@ describe('useSubscription', () => {
     const subject = makeSubject<any>();
     const executeSubscription = vi
       .spyOn(client, 'executeSubscription')
-      .mockImplementation(() => subject.source);
+      .mockImplementation(
+        () => subject.source as OperationResultSource<OperationResult>
+      );
 
     const scanHandler = (currentState: any, nextState: any) => ({
       counter: (currentState ? currentState.counter : 0) + nextState.counter,