Improve tests

igorkamyshev · Aug 23, 2023 · 8b391da · 8b391da
1 parent 901545b
commit 8b391da
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 5 deletions.
diff --git a/packages/core/src/mutation/__tests__/create_headless_mutation.test.ts b/packages/core/src/mutation/__tests__/create_headless_mutation.test.ts
@@ -38,15 +38,15 @@ describe('createHeadlessMutation', () => {
 
     expect(listeners.onSuccess).toHaveBeenCalledTimes(1);
     expect(listeners.onSuccess).toHaveBeenCalledWith(
-      expect.objectContaining({ params: 42, result: 42 })
+      expect.objectContaining({ params: 42, result: 42, status: 'done' })
     );
 
     expect(listeners.onSkip).not.toHaveBeenCalled();
     expect(listeners.onFailure).not.toHaveBeenCalled();
 
     expect(listeners.onFinally).toHaveBeenCalledTimes(1);
     expect(listeners.onFinally).toHaveBeenCalledWith(
-      expect.objectContaining({ params: 42, result: 42 })
+      expect.objectContaining({ params: 42, result: 42, status: 'done' })
     );
   });
 

diff --git a/packages/core/src/query/__tests__/create_headless_query.test.ts b/packages/core/src/query/__tests__/create_headless_query.test.ts
@@ -35,15 +35,15 @@ describe('core/createHeadlessQuery without contract', () => {
     expect(scope.getState(query.$error)).toBeNull();
     expect(listeners.onSuccess).toHaveBeenCalledTimes(1);
     expect(listeners.onSuccess).toHaveBeenCalledWith(
-      expect.objectContaining({ params: 42, result: 42 })
+      expect.objectContaining({ params: 42, result: 42, status: 'done' })
     );
 
     expect(listeners.onSkip).not.toHaveBeenCalled();
     expect(listeners.onFailure).not.toHaveBeenCalled();
 
     expect(listeners.onFinally).toHaveBeenCalledTimes(1);
     expect(listeners.onFinally).toHaveBeenCalledWith(
-      expect.objectContaining({ params: 42, result: 42 })
+      expect.objectContaining({ params: 42, result: 42, status: 'done' })
     );
   });
 
@@ -79,7 +79,11 @@ describe('core/createHeadlessQuery without contract', () => {
 
     expect(listeners.onFinally).toHaveBeenCalledTimes(1);
     expect(listeners.onFinally).toHaveBeenCalledWith(
-      expect.objectContaining({ params: 42, error: new Error('from mock') })
+      expect.objectContaining({
+        params: 42,
+        error: new Error('from mock'),
+        status: 'fail',
+      })
     );
   });