fix: wrap set state with array expression (#1079)

aws-amplify · Aug 17, 2023 · 5c73726 · 5c73726
1 parent 58a475c
commit 5c73726
Show file tree

Hide file tree

Showing 2 changed files with 51 additions and 49 deletions.
diff --git a/...s/codegen-ui-react/lib/__tests__/__snapshots__/studio-ui-codegen-react-forms.test.ts.snap b/...s/codegen-ui-react/lib/__tests__/__snapshots__/studio-ui-codegen-react-forms.test.ts.snap
@@ -1706,7 +1706,7 @@ export default function MyMemberForm(props) {
               : \\"\\"
           );
           setCurrentTeamIDValue(value);
-          setSelectedTeamIDRecords(teamIDRecords.find((r) => r.id === value));
+          setSelectedTeamIDRecords([teamIDRecords.find((r) => r.id === value)]);
         }}
         inputFieldRef={teamIDRef}
         defaultFieldValue={\\"\\"}
@@ -4018,7 +4018,7 @@ export default function CommentCreateForm(props) {
               : \\"\\"
           );
           setCurrentPostIDValue(value);
-          setSelectedPostIDRecords(postIDRecords.find((r) => r.id === value));
+          setSelectedPostIDRecords([postIDRecords.find((r) => r.id === value)]);
         }}
         inputFieldRef={postIDRef}
         defaultFieldValue={\\"\\"}
@@ -9614,7 +9614,7 @@ export default function CommentUpdateForm(props) {
               : \\"\\"
           );
           setCurrentPostIDValue(value);
-          setSelectedPostIDRecords(postIDRecords.find((r) => r.id === value));
+          setSelectedPostIDRecords([postIDRecords.find((r) => r.id === value)]);
         }}
         inputFieldRef={postIDRef}
         defaultFieldValue={\\"\\"}
@@ -10344,7 +10344,7 @@ export default function CommentUpdateForm(props) {
               : \\"\\"
           );
           setCurrentPostIDValue(value);
-          setSelectedPostIDRecords(postIDRecords.find((r) => r.id === value));
+          setSelectedPostIDRecords([postIDRecords.find((r) => r.id === value)]);
         }}
         inputFieldRef={postIDRef}
         defaultFieldValue={\\"\\"}
@@ -10997,7 +10997,7 @@ export default function CommentUpdateForm(props) {
               : \\"\\"
           );
           setCurrentPostIDValue(value);
-          setSelectedPostIDRecords(postIDRecords.find((r) => r.id === value));
+          setSelectedPostIDRecords([postIDRecords.find((r) => r.id === value)]);
         }}
         inputFieldRef={postIDRef}
         defaultFieldValue={\\"\\"}
@@ -13869,9 +13869,9 @@ export default function CreateCompositeToyForm(props) {
               : \\"\\"
           );
           setCurrentCompositeDogCompositeToysNameValue(value);
-          setSelectedCompositeDogCompositeToysNameRecords(
-            compositeDogCompositeToysNameRecords.find((r) => r.name === value)
-          );
+          setSelectedCompositeDogCompositeToysNameRecords([
+            compositeDogCompositeToysNameRecords.find((r) => r.name === value),
+          ]);
         }}
         inputFieldRef={compositeDogCompositeToysNameRef}
         defaultFieldValue={\\"\\"}
@@ -13982,11 +13982,11 @@ export default function CreateCompositeToyForm(props) {
               : \\"\\"
           );
           setCurrentCompositeDogCompositeToysDescriptionValue(value);
-          setSelectedCompositeDogCompositeToysDescriptionRecords(
+          setSelectedCompositeDogCompositeToysDescriptionRecords([
             compositeDogCompositeToysDescriptionRecords.find(
               (r) => r.description === value
-            )
-          );
+            ),
+          ]);
         }}
         inputFieldRef={compositeDogCompositeToysDescriptionRef}
         defaultFieldValue={\\"\\"}
@@ -14929,9 +14929,9 @@ export default function CreateCommentForm(props) {
               : \\"\\"
           );
           setCurrentPostCommentsIdValue(value);
-          setSelectedPostCommentsIdRecords(
-            postCommentsIdRecords.find((r) => r.id === value)
-          );
+          setSelectedPostCommentsIdRecords([
+            postCommentsIdRecords.find((r) => r.id === value),
+          ]);
         }}
         inputFieldRef={postCommentsIdRef}
         defaultFieldValue={\\"\\"}
@@ -17788,11 +17788,11 @@ export default function ChildItemUpdateForm(props) {
               : \\"\\"
           );
           setCurrentCustomKeyModelChildrenMycustomkeyValue(value);
-          setSelectedCustomKeyModelChildrenMycustomkeyRecords(
+          setSelectedCustomKeyModelChildrenMycustomkeyRecords([
             customKeyModelChildrenMycustomkeyRecords.find(
               (r) => r.mycustomkey === value
-            )
-          );
+            ),
+          ]);
         }}
         inputFieldRef={customKeyModelChildrenMycustomkeyRef}
         defaultFieldValue={\\"\\"}
@@ -20181,7 +20181,7 @@ export default function CommentUpdateForm(props) {
               : \\"\\"
           );
           setCurrentPostIDValue(value);
-          setSelectedPostIDRecords(postIDRecords.find((r) => r.id === value));
+          setSelectedPostIDRecords([postIDRecords.find((r) => r.id === value)]);
         }}
         inputFieldRef={postIDRef}
         defaultFieldValue={\\"\\"}

diff --git a/packages/codegen-ui-react/lib/forms/form-renderer-helper/render-array-field.ts b/packages/codegen-ui-react/lib/forms/form-renderer-helper/render-array-field.ts
@@ -365,39 +365,41 @@ export const renderArrayFieldComponent = (
               getSetNameIdentifier(`selected${capitalizeFirstLetter(fieldName)}Records`),
               undefined,
               [
-                factory.createCallExpression(
-                  factory.createPropertyAccessExpression(
-                    factory.createIdentifier(getRecordsName(fieldName)),
-                    factory.createIdentifier('find'),
-                  ),
-                  undefined,
-                  [
-                    factory.createArrowFunction(
-                      undefined,
-                      undefined,
-                      [
-                        factory.createParameterDeclaration(
-                          undefined,
-                          undefined,
-                          undefined,
-                          factory.createIdentifier('r'),
-                          undefined,
-                          undefined,
-                        ),
-                      ],
-                      undefined,
-                      factory.createToken(SyntaxKind.EqualsGreaterThanToken),
-                      factory.createBinaryExpression(
-                        factory.createPropertyAccessExpression(
-                          factory.createIdentifier('r'),
-                          factory.createIdentifier(scalarKey),
+                factory.createArrayLiteralExpression([
+                  factory.createCallExpression(
+                    factory.createPropertyAccessExpression(
+                      factory.createIdentifier(getRecordsName(fieldName)),
+                      factory.createIdentifier('find'),
+                    ),
+                    undefined,
+                    [
+                      factory.createArrowFunction(
+                        undefined,
+                        undefined,
+                        [
+                          factory.createParameterDeclaration(
+                            undefined,
+                            undefined,
+                            undefined,
+                            factory.createIdentifier('r'),
+                            undefined,
+                            undefined,
+                          ),
+                        ],
+                        undefined,
+                        factory.createToken(SyntaxKind.EqualsGreaterThanToken),
+                        factory.createBinaryExpression(
+                          factory.createPropertyAccessExpression(
+                            factory.createIdentifier('r'),
+                            factory.createIdentifier(scalarKey),
+                          ),
+                          factory.createToken(SyntaxKind.EqualsEqualsEqualsToken),
+                          factory.createIdentifier('value'),
                         ),
-                        factory.createToken(SyntaxKind.EqualsEqualsEqualsToken),
-                        factory.createIdentifier('value'),
                       ),
-                    ),
-                  ],
-                ),
+                    ],
+                  ),
+                ]),
               ],
             ),
           ),