pass wrapUpdate through useToastState (#7910)

Author: reidbarber

packages/@react-stately/toast/src/useToastState.ts

@@ -57,8 +57,8 @@ export interface ToastState<T> {
  * of actions, errors, or other events in an application.
  */
 export function useToastState<T>(props: ToastStateProps = {}): ToastState<T> {
-  let {maxVisibleToasts = 1} = props;
-  let queue = useMemo(() => new ToastQueue<T>({maxVisibleToasts}), [maxVisibleToasts]);
+  let {maxVisibleToasts = 1, wrapUpdate} = props;
+  let queue = useMemo(() => new ToastQueue<T>({maxVisibleToasts, wrapUpdate}), [maxVisibleToasts, wrapUpdate]);
   return useToastQueue(queue);
 }
 

packages/@react-stately/toast/test/useToastState.test.js

@@ -217,4 +217,28 @@ describe('useToastState', () => {
     expect(result.current.visibleToasts.length).toBe(1);
     expect(result.current.visibleToasts[0].content).toBe(newValue[0].content);
   });
+
+  it('should use provided wrapUpdate', () => {
+    let wrapUpdate = jest.fn(fn => fn());
+
+    let {result} = renderHook(() => useToastState({wrapUpdate}));
+    expect(result.current.visibleToasts).toStrictEqual([]);
+
+    act(() => {result.current.add(newValue[0].content, newValue[0].props);});
+    expect(result.current.visibleToasts[0].content).toBe(newValue[0].content);
+
+    expect(wrapUpdate).toHaveBeenCalledTimes(1);
+
+    act(() => {result.current.add('Second Toast');});
+    expect(result.current.visibleToasts.length).toBe(1);
+    expect(result.current.visibleToasts[0].content).toBe('Second Toast');
+
+    expect(wrapUpdate).toHaveBeenCalledTimes(2);
+
+    act(() => {result.current.close(result.current.visibleToasts[0].key);});
+    expect(result.current.visibleToasts.length).toBe(1);
+    expect(result.current.visibleToasts[0].content).toBe(newValue[0].content);
+
+    expect(wrapUpdate).toHaveBeenCalledTimes(3);
+  });
 });