From af84323bb6cf93d31adc56d1203f9c8a0fb4c710 Mon Sep 17 00:00:00 2001
From: Matthijs Groen <matthijs.groen@gmail.com>
Date: Fri, 21 Jun 2024 16:56:26 +0200
Subject: [PATCH] allow mutation for updateProps as well

---
 src/support/testing/testSubject.tsx | 24 ++++++++++++++----------
 1 file changed, 14 insertions(+), 10 deletions(-)

diff --git a/src/support/testing/testSubject.tsx b/src/support/testing/testSubject.tsx
index c9e0a38f..c7c247ff 100644
--- a/src/support/testing/testSubject.tsx
+++ b/src/support/testing/testSubject.tsx
@@ -6,6 +6,8 @@ type DecoratorSettings<
   TKey extends string,
 > = TDecorators extends Decorator<TKey, infer TSettings> ? TSettings : never;
 
+type UpdatedSettings<T> = Partial<T> | ((initialSettings: T) => Partial<T>);
+
 type TestHelpers<
   TComponent extends React.FC<any>,
   TDecorators extends Decorator<string, any>[],
@@ -20,24 +22,22 @@ type TestHelpers<
    * Update the properties of the component.
    * After a `renderComponent` call these settings get reset.
    */
-  updateProps(newProps: Partial<React.ComponentProps<TComponent>>): void;
+  updateProps(
+    newProps: UpdatedSettings<React.ComponentProps<TComponent>>,
+  ): void;
   /**
    * Update the settings of a decorator.
    * After a `renderComponent` call these settings get reset.
    */
   updateDecorator<TKey extends TDecorators[number]["name"]>(
     key: TKey,
-    settings:
-      | Partial<DecoratorSettings<TDecorators[number], TKey>>
-      | ((
-          initialSettings: DecoratorSettings<TDecorators[number], TKey>,
-        ) => Partial<DecoratorSettings<TDecorators[number], TKey>>),
+    settings: UpdatedSettings<DecoratorSettings<TDecorators[number], TKey>>,
   ): void;
   /**
    * Render the component wrapped with all decorators, applying
    * all property and decorator setting updates.
    */
-  renderComponent(): Promise<RenderResult>;
+  renderComponent(): RenderResult;
 };
 
 export type Decorator<
@@ -125,8 +125,7 @@ export const setTestSubject = <
   let lastRender: RenderResult | null = null;
 
   return {
-    // eslint-disable-next-line require-await
-    renderComponent: async () => {
+    renderComponent: () => {
       if (initialProps === null) {
         throw new Error("No props specified with setProps");
       }
@@ -160,7 +159,12 @@ export const setTestSubject = <
       if (initialProps === null) {
         throw new Error("No props specified with setProps");
       }
-      props = { ...initialProps, ...props, ...updatedProps };
+      const update: Partial<React.ComponentProps<TComponent>> =
+        typeof updatedProps === "function"
+          ? updatedProps(props ? props : initialProps)
+          : updatedProps;
+
+      props = { ...initialProps, ...props, ...update };
     },
     updateDecorator: (name, updatedSettings) => {
       const update: Record<string, unknown> =