fix: add import for removal policy in gen2 codegen

aws-amplify · Dec 11, 2024 · 300d169 · 300d169
1 parent 9e383cd
commit 300d169
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 0 deletions.
diff --git a/packages/amplify-gen2-codegen/src/backend/synthesizer.test.ts b/packages/amplify-gen2-codegen/src/backend/synthesizer.test.ts
@@ -335,6 +335,7 @@ describe('BackendRenderer', () => {
       });
       const output = printNodeArray(rendered);
       assert(output.includes('s3Bucket.applyRemovalPolicy'));
+      assert(output.includes('import { RemovalPolicy } from "aws-cdk-lib";'));
     });
   });
   describe('UserPoolClient Configuration using render()', () => {
@@ -471,6 +472,7 @@ describe('BackendRenderer', () => {
       const output = printNodeArray(rendered);
       assert(output.includes('cfnUserPool.applyRemovalPolicy'));
       assert(output.includes('cfnIdentityPool.applyRemovalPolicy'));
+      assert(output.includes('import { RemovalPolicy } from "aws-cdk-lib";'));
     });
   });
 });
diff --git a/packages/amplify-gen2-codegen/src/backend/synthesizer.ts b/packages/amplify-gen2-codegen/src/backend/synthesizer.ts
@@ -447,6 +447,7 @@ export class BackendSynthesizer {
 
     if (renderArgs.auth) {
       imports.push(this.createImportStatement([authFunctionIdentifier], renderArgs.auth.importFrom));
+      imports.push(this.createImportStatement([factory.createIdentifier('RemovalPolicy')], 'aws-cdk-lib'));
       const auth = factory.createShorthandPropertyAssignment(authFunctionIdentifier);
       defineBackendProperties.push(auth);
     }