test(query-core): add test case for notifyManager (#8976)

Author: minseong0324

packages/query-core/src/__tests__/notifyManager.test.tsx

@@ -86,4 +86,38 @@ describe('notifyManager', () => {
     // @ts-expect-error
     someFn('im not happy', false)
   })
+
+  it('should use custom batch notify function', async () => {
+    const notifyManagerTest = createNotifyManager()
+    const batchNotifySpy = vi.fn((cb) => cb())
+    const callbackSpy1 = vi.fn()
+    const callbackSpy2 = vi.fn()
+
+    notifyManagerTest.setBatchNotifyFunction(batchNotifySpy)
+
+    notifyManagerTest.batch(() => {
+      notifyManagerTest.schedule(callbackSpy1)
+      notifyManagerTest.schedule(callbackSpy2)
+    })
+
+    await vi.advanceTimersByTimeAsync(0)
+
+    expect(batchNotifySpy).toHaveBeenCalled()
+    expect(callbackSpy1).toHaveBeenCalled()
+    expect(callbackSpy2).toHaveBeenCalled()
+  })
+
+  it('should batch calls correctly', async () => {
+    const notifyManagerTest = createNotifyManager()
+    const callbackSpy = vi.fn()
+
+    const batchedFn = notifyManagerTest.batchCalls((a: number, b: string) => {
+      callbackSpy(a, b)
+    })
+
+    batchedFn(1, 'test')
+    await vi.advanceTimersByTimeAsync(0)
+
+    expect(callbackSpy).toHaveBeenCalledWith(1, 'test')
+  })
 })