add tests

reduxjs · Apr 20, 2022 · 12c48af · 12c48af
1 parent 003433e
commit 12c48af
Show file tree

Hide file tree

Showing 3 changed files with 410 additions and 6 deletions.
diff --git a/packages/toolkit/src/query/core/buildThunks.ts b/packages/toolkit/src/query/core/buildThunks.ts
@@ -278,9 +278,6 @@ export function buildThunks<
             api.util.patchQueryData(endpointName, args, ret.inversePatches)
           ),
       }
-      if (currentState.status === QueryStatus.uninitialized) {
-        return ret
-      }
       if ('data' in currentState) {
         if (isDraftable(currentState.data)) {
           const [, patches, inversePatches] = produceWithPatches(
@@ -305,10 +302,22 @@ export function buildThunks<
           path: [],
           value: undefined,
         })
-        dispatch(api.endpoints[endpointName].initiate(args, {subscribe: false, forceRefetch: true, }))
+        dispatch(
+          (
+            api.endpoints[endpointName] as ApiEndpointQuery<
+              QueryDefinition<any, any, any, any, any>,
+              Definitions
+            >
+          ).initiate(args, {
+            subscribe: false,
+            forceRefetch: true,
+            forceQueryFn: () => ({
+              data: upsertRecipe(undefined),
+            }),
+          })
+        )
       }
 
-
       return ret
     }
 

diff --git a/packages/toolkit/src/query/core/module.ts b/packages/toolkit/src/query/core/module.ts
@@ -1,7 +1,11 @@
 /**
  * Note: this file should import all other files for type discovery and declaration merging
  */
-import type { PatchQueryDataThunk, UpdateQueryDataThunk } from './buildThunks'
+import type {
+  PatchQueryDataThunk,
+  UpdateQueryDataThunk,
+  UpsertQueryDataThunk,
+} from './buildThunks'
 import { buildThunks } from './buildThunks'
 import type {
   ActionCreatorWithPayload,
@@ -210,6 +214,10 @@ declare module '../apiTypes' {
           Definitions,
           RootState<Definitions, string, ReducerPath>
         >
+        upsertQueryData: UpsertQueryDataThunk<
+          Definitions,
+          RootState<Definitions, string, ReducerPath>
+        >
         /**
          * A Redux thunk that applies a JSON diff/patch array to the cached data for a given query result. This immediately updates the Redux state with those changes.
          *
@@ -406,6 +414,7 @@ export const coreModule = (): Module<CoreModule> => ({
       mutationThunk,
       patchQueryData,
       updateQueryData,
+      upsertQueryData,
       prefetch,
       buildMatchThunkActions,
     } = buildThunks({
@@ -434,6 +443,7 @@ export const coreModule = (): Module<CoreModule> => ({
     safeAssign(api.util, {
       patchQueryData,
       updateQueryData,
+      upsertQueryData,
       prefetch,
       resetApiState: sliceActions.resetApiState,
     })