Fix failing test

Author: poteto

packages/react-reconciler/src/ReactFiberHooks.new.js

@@ -1934,7 +1934,7 @@ function useEventImpl<Args, Return, F: (...Array<Args>) => Return>(
   }
 }
 
-function wrapEventFunction<Args, Return, F: (...Array<Args>) => Return>(
+export function wrapEventFunction<Args, Return, F: (...Array<Args>) => Return>(
   callback: F,
 ): EventFunctionWrapper<Args, Return, F> {
   const eventFn: EventFunctionWrapper<Args, Return, F> = function eventFn() {
@@ -1953,17 +1953,21 @@ function wrapEventFunction<Args, Return, F: (...Array<Args>) => Return>(
 function mountEvent<Args, Return, F: (...Array<Args>) => Return>(
   callback: F,
 ): EventFunctionWrapper<Args, Return, F> {
+  const hook = mountWorkInProgressHook();
   const eventFn = wrapEventFunction(callback);
+  hook.memoizedState = eventFn;
   useEventImpl(eventFn, callback);
   return eventFn;
 }
 
 function updateEvent<Args, Return, F: (...Array<Args>) => Return>(
   callback: F,
 ): EventFunctionWrapper<Args, Return, F> {
-  const eventFn = wrapEventFunction(callback);
+  const hook = updateWorkInProgressHook();
+  const eventFn = hook.memoizedState;
   useEventImpl(eventFn, callback);
-  return eventFn;
+  // Always return a new function
+  return wrapEventFunction(callback);
 }
 
 function mountInsertionEffect(

packages/react-reconciler/src/ReactFiberHooks.old.js

@@ -1934,7 +1934,7 @@ function useEventImpl<Args, Return, F: (...Array<Args>) => Return>(
   }
 }
 
-function wrapEventFunction<Args, Return, F: (...Array<Args>) => Return>(
+export function wrapEventFunction<Args, Return, F: (...Array<Args>) => Return>(
   callback: F,
 ): EventFunctionWrapper<Args, Return, F> {
   const eventFn: EventFunctionWrapper<Args, Return, F> = function eventFn() {
@@ -1953,17 +1953,21 @@ function wrapEventFunction<Args, Return, F: (...Array<Args>) => Return>(
 function mountEvent<Args, Return, F: (...Array<Args>) => Return>(
   callback: F,
 ): EventFunctionWrapper<Args, Return, F> {
+  const hook = mountWorkInProgressHook();
   const eventFn = wrapEventFunction(callback);
+  hook.memoizedState = eventFn;
   useEventImpl(eventFn, callback);
   return eventFn;
 }
 
 function updateEvent<Args, Return, F: (...Array<Args>) => Return>(
   callback: F,
 ): EventFunctionWrapper<Args, Return, F> {
-  const eventFn = wrapEventFunction(callback);
+  const hook = updateWorkInProgressHook();
+  const eventFn = hook.memoizedState;
   useEventImpl(eventFn, callback);
-  return eventFn;
+  // Always return a new function
+  return wrapEventFunction(callback);
 }
 
 function mountInsertionEffect(