feat(gen2-migration): add deployment status validation for lock command

packages/amplify-cli/src/commands/gen2-migration/_validations.ts

@@ -42,14 +42,17 @@ export class AmplifyGen2MigrationValidations {
     }
 
     const stackStatus = response.Stacks[0].StackStatus;
-    const validStatuses = ['UPDATE_COMPLETE', 'CREATE_COMPLETE'];
+    // Note: UPDATE_ROLLBACK_COMPLETE isn't an expected state - only being added in the edge case of resuming migration from a failed state
+    const validStatuses = ['UPDATE_COMPLETE', 'CREATE_COMPLETE', 'UPDATE_ROLLBACK_COMPLETE'];
 
     if (!validStatuses.includes(stackStatus)) {
       throw new AmplifyError('StackStateError', {
-        message: `Root stack status is ${stackStatus}, expected ${validStatuses.join(' or ')}`,
+        message: `Root stack status is ${stackStatus}, expected UPDATE_COMPLETE or CREATE_COMPLETE`,
         resolution: 'Complete the deployment before proceeding.',
       });
     }
+
+    printer.success(`Deployment status validated: ${stackStatus}`);
   }
 
   public async validateDeploymentVersion(): Promise<void> {

packages/amplify-cli/src/commands/gen2-migration/lock.ts

@@ -1,12 +1,15 @@
 import { AmplifyMigrationStep } from './_step';
 import { printer } from '@aws-amplify/amplify-prompts';
+import { AmplifyGen2MigrationValidations } from './_validations';
 
 export class AmplifyMigrationLockStep extends AmplifyMigrationStep {
   readonly command = 'lock';
   readonly describe = 'Lock environment';
 
   public async validate(): Promise<void> {
-    printer.warn('Not implemented');
+    // eslint-disable-next-line @typescript-eslint/no-explicit-any
+    const validations = new AmplifyGen2MigrationValidations({} as any);
+    await validations.validateDeploymentStatus();
   }
 
   public async execute(): Promise<void> {