fix: add form datamodel to requiredDataModels for renderComponentOnly (…

…#815) Co-authored-by: Justin Shih <jushih@amazon.com>
aws-amplify · Dec 2, 2022 · f0d74e5 · f0d74e5
1 parent 609ffc2
commit f0d74e5
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 1 deletion.
diff --git a/packages/codegen-ui-react/lib/__tests__/studio-ui-codegen-react-forms.test.ts b/packages/codegen-ui-react/lib/__tests__/studio-ui-codegen-react-forms.test.ts
@@ -292,11 +292,13 @@ describe('amplify form renderer tests', () => {
         'datastore/project-team-model',
       );
 
-      const teamAlias = importCollection.importAlias.get(ImportSource.LOCAL_MODELS)?.get('Team');
+      const teamAlias = importCollection.getMappedAlias(ImportSource.LOCAL_MODELS, 'Team');
 
       const includesTeam = requiredDataModels.includes('Team');
+      const includesMember = requiredDataModels.includes('Member');
       expect(teamAlias).toBe('Team0');
       expect(includesTeam).toBe(true);
+      expect(includesMember).toBe(true);
       expect(importCollection).toBeDefined();
     });
 

diff --git a/packages/codegen-ui-react/lib/forms/react-form-renderer.ts b/packages/codegen-ui-react/lib/forms/react-form-renderer.ts
@@ -386,6 +386,7 @@ export abstract class ReactFormTemplateRenderer extends StudioTemplateRenderer<
 
     // add model import for datastore type
     if (dataSourceType === 'DataStore') {
+      this.requiredDataModels.push(dataTypeName);
       modelName = this.importCollection.addImport(ImportSource.LOCAL_MODELS, dataTypeName);
     }