fix: update model used

aws-amplify · Nov 18, 2022 · ad0beda · ad0beda
1 parent b1444cd
commit ad0beda
Show file tree

Hide file tree

Showing 3 changed files with 8 additions and 4 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
@@ -4471,7 +4471,7 @@ export default function MyFlexUpdateForm(props) {
         }
         try {
           await DataStore.save(
-            Flex0.copyOf(flex0Record, (updated) => {
+            Flex0.copyOf(flexRecord, (updated) => {
               Object.assign(updated, modelFields);
             })
           );

diff --git a/packages/codegen-ui-react/lib/amplify-ui-renderers/form.ts b/packages/codegen-ui-react/lib/amplify-ui-renderers/form.ts
@@ -127,7 +127,7 @@ export default class FormRenderer extends ReactComponentRenderer<BaseComponentPr
         factory.createTryStatement(
           factory.createBlock(
             [
-              ...buildDataStoreExpression(formActionType, importedModelName),
+              ...buildDataStoreExpression(formActionType, importedModelName, dataTypeName),
               // call onSuccess hook if it exists
               factory.createIfStatement(
                 factory.createIdentifier('onSuccess'),

diff --git a/packages/codegen-ui-react/lib/forms/form-renderer-helper.ts b/packages/codegen-ui-react/lib/forms/form-renderer-helper.ts
@@ -552,7 +552,11 @@ export const buildOnChangeStatement = (
   );
 };
 
-export const buildDataStoreExpression = (dataStoreActionType: 'update' | 'create', importedModelName: string) => {
+export const buildDataStoreExpression = (
+  dataStoreActionType: 'update' | 'create',
+  importedModelName: string,
+  dataTypeName: string,
+) => {
   if (dataStoreActionType === 'update') {
     return [
       factory.createExpressionStatement(
@@ -571,7 +575,7 @@ export const buildDataStoreExpression = (dataStoreActionType: 'update' | 'create
                 ),
                 undefined,
                 [
-                  factory.createIdentifier(`${lowerCaseFirst(importedModelName)}Record`),
+                  factory.createIdentifier(`${lowerCaseFirst(dataTypeName)}Record`),
                   factory.createArrowFunction(
                     undefined,
                     undefined,