add tests

Author: dutziworks

packages/toolkit/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@reduxjs/toolkit",
-  "version": "1.9.1",
+  "version": "1.9.2-11",
   "description": "The official, opinionated, batteries-included toolset for efficient Redux development",
   "author": "Mark Erikson <mark@isquaredsoftware.com>",
   "license": "MIT",
@@ -94,7 +94,10 @@
     "dist/**/*.d.ts",
     "dist/**/package.json",
     "src/",
-    "query"
+    "query",
+    "tsconfig.test.json",
+    "tsconfig.json",
+    "tsconfig.base.json"
   ],
   "dependencies": {
     "immer": "^9.0.16",

packages/toolkit/src/query/tests/optimisticUpdates.test.tsx

@@ -1,6 +1,7 @@
 import { createApi } from '@reduxjs/toolkit/query/react'
 import { actionsReducer, hookWaitFor, setupApiStore, waitMs } from './helpers'
 import { renderHook, act } from '@testing-library/react'
+import type { InvalidationState } from '../core/apiState'
 
 interface Post {
   id: string
@@ -26,6 +27,13 @@ const api = createApi({
       query: (id) => `post/${id}`,
       providesTags: ['Post'],
     }),
+    listPosts: build.query<Post[], void>({
+      query: () => `posts`,
+      providesTags: (result) => [
+        ...(result?.map(({ id }) => ({ type: 'Post' as const, id })) ?? []),
+        'Post',
+      ],
+    }),
     updatePost: build.mutation<void, Pick<Post, 'id'> & Partial<Post>>({
       query: ({ id, ...patch }) => ({
         url: `post/${id}`,
@@ -184,6 +192,67 @@ describe('updateQueryData', () => {
     expect(result.current.data).toEqual(dataBefore)
   })
 
+  test.only('updates cache values including provided tags, undos that', async () => {
+    baseQuery
+      .mockResolvedValueOnce([
+        {
+          id: '3',
+          title: 'All about cheese.',
+          contents: 'TODO',
+        },
+      ])
+      .mockResolvedValueOnce(42)
+    const { result } = renderHook(() => api.endpoints.listPosts.useQuery(), {
+      wrapper: storeRef.wrapper,
+    })
+    await hookWaitFor(() => expect(result.current.isSuccess).toBeTruthy())
+
+    let provided!: InvalidationState<'Post'>
+    act(() => {
+      provided = storeRef.store.getState().api.provided
+    })
+
+    const provided3 = provided['Post']['3']
+
+    let returnValue!: ReturnType<ReturnType<typeof api.util.updateQueryData>>
+    act(() => {
+      returnValue = storeRef.store.dispatch(
+        api.util.updateQueryData(
+          'listPosts',
+          undefined,
+          (draft) => {
+            draft.push({
+              id: '4',
+              title: 'Mostly about cheese.',
+              contents: 'TODO',
+            })
+          },
+          true
+        )
+      )
+    })
+
+    act(() => {
+      provided = storeRef.store.getState().api.provided
+    })
+
+    const provided4 = provided['Post']['4']
+
+    expect(provided4).toEqual(provided3)
+
+    act(() => {
+      returnValue.undo()
+    })
+
+    act(() => {
+      provided = storeRef.store.getState().api.provided
+    })
+
+    const provided4Next = provided['Post']['4']
+
+    expect(provided4Next).toEqual([])
+  })
+
   test('does not update non-existing values', async () => {
     baseQuery
       .mockImplementationOnce(async () => ({