chore: add gen2 migration deployment status validation and unit tests

Author: sai-ray

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

@@ -1,6 +1,6 @@
 import { AmplifyGen2MigrationValidations } from '../../../commands/gen2-migration/_validations';
-import { $TSContext } from '@aws-amplify/amplify-cli-core';
-import { DescribeChangeSetOutput } from '@aws-sdk/client-cloudformation';
+import { $TSContext, stateManager } from '@aws-amplify/amplify-cli-core';
+import { DescribeChangeSetOutput, CloudFormationClient } from '@aws-sdk/client-cloudformation';
 
 describe('AmplifyGen2MigrationValidations', () => {
   let mockContext: $TSContext;
@@ -317,4 +317,119 @@ describe('AmplifyGen2MigrationValidations', () => {
       });
     });
   });
+
+  describe('validateDeploymentStatus', () => {
+    beforeEach(() => {
+      jest.clearAllMocks();
+    });
+
+    it('should throw StackNotFoundError when stackName is missing', async () => {
+      jest.spyOn(stateManager, 'getMeta').mockReturnValue({
+        providers: {
+          awscloudformation: {},
+        },
+      });
+
+      await expect(validations.validateDeploymentStatus()).rejects.toMatchObject({
+        name: 'StackNotFoundError',
+        message: 'Root stack not found',
+        resolution: 'Ensure the project is initialized and deployed.',
+      });
+    });
+
+    it('should throw StackNotFoundError when stack not found in CloudFormation', async () => {
+      jest.spyOn(stateManager, 'getMeta').mockReturnValue({
+        providers: {
+          awscloudformation: {
+            StackName: 'test-stack',
+          },
+        },
+      });
+
+      const mockSend = jest.fn().mockResolvedValue({ Stacks: [] });
+      jest.spyOn(CloudFormationClient.prototype, 'send').mockImplementation(mockSend);
+
+      await expect(validations.validateDeploymentStatus()).rejects.toMatchObject({
+        name: 'StackNotFoundError',
+        message: 'Stack test-stack not found in CloudFormation',
+        resolution: 'Ensure the project is deployed.',
+      });
+    });
+
+    it('should pass when stack status is UPDATE_COMPLETE', async () => {
+      jest.spyOn(stateManager, 'getMeta').mockReturnValue({
+        providers: {
+          awscloudformation: {
+            StackName: 'test-stack',
+          },
+        },
+      });
+
+      const mockSend = jest.fn().mockResolvedValue({
+        Stacks: [{ StackStatus: 'UPDATE_COMPLETE' }],
+      });
+      jest.spyOn(CloudFormationClient.prototype, 'send').mockImplementation(mockSend);
+
+      await expect(validations.validateDeploymentStatus()).resolves.not.toThrow();
+    });
+
+    it('should pass when stack status is CREATE_COMPLETE', async () => {
+      jest.spyOn(stateManager, 'getMeta').mockReturnValue({
+        providers: {
+          awscloudformation: {
+            StackName: 'test-stack',
+          },
+        },
+      });
+
+      const mockSend = jest.fn().mockResolvedValue({
+        Stacks: [{ StackStatus: 'CREATE_COMPLETE' }],
+      });
+      jest.spyOn(CloudFormationClient.prototype, 'send').mockImplementation(mockSend);
+
+      await expect(validations.validateDeploymentStatus()).resolves.not.toThrow();
+    });
+
+    it('should throw StackStateError when status is UPDATE_IN_PROGRESS', async () => {
+      jest.spyOn(stateManager, 'getMeta').mockReturnValue({
+        providers: {
+          awscloudformation: {
+            StackName: 'test-stack',
+          },
+        },
+      });
+
+      const mockSend = jest.fn().mockResolvedValue({
+        Stacks: [{ StackStatus: 'UPDATE_IN_PROGRESS' }],
+      });
+      jest.spyOn(CloudFormationClient.prototype, 'send').mockImplementation(mockSend);
+
+      await expect(validations.validateDeploymentStatus()).rejects.toMatchObject({
+        name: 'StackStateError',
+        message: 'Root stack status is UPDATE_IN_PROGRESS, expected UPDATE_COMPLETE or CREATE_COMPLETE',
+        resolution: 'Complete the deployment before proceeding.',
+      });
+    });
+
+    it('should throw StackStateError when status is ROLLBACK_COMPLETE', async () => {
+      jest.spyOn(stateManager, 'getMeta').mockReturnValue({
+        providers: {
+          awscloudformation: {
+            StackName: 'test-stack',
+          },
+        },
+      });
+
+      const mockSend = jest.fn().mockResolvedValue({
+        Stacks: [{ StackStatus: 'ROLLBACK_COMPLETE' }],
+      });
+      jest.spyOn(CloudFormationClient.prototype, 'send').mockImplementation(mockSend);
+
+      await expect(validations.validateDeploymentStatus()).rejects.toMatchObject({
+        name: 'StackStateError',
+        message: 'Root stack status is ROLLBACK_COMPLETE, expected UPDATE_COMPLETE or CREATE_COMPLETE',
+        resolution: 'Complete the deployment before proceeding.',
+      });
+    });
+  });
 });

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

@@ -1,7 +1,7 @@
 import { AmplifyDriftDetector } from '../drift';
-import { $TSContext, AmplifyError } from '@aws-amplify/amplify-cli-core';
+import { $TSContext, AmplifyError, stateManager } from '@aws-amplify/amplify-cli-core';
 import { printer } from '@aws-amplify/amplify-prompts';
-import { DescribeChangeSetOutput } from '@aws-sdk/client-cloudformation';
+import { DescribeChangeSetOutput, CloudFormationClient, DescribeStacksCommand } from '@aws-sdk/client-cloudformation';
 import { STATEFUL_RESOURCES } from './stateful-resources';
 
 export class AmplifyGen2MigrationValidations {
@@ -16,7 +16,35 @@ export class AmplifyGen2MigrationValidations {
   }
 
   public async validateDeploymentStatus(): Promise<void> {
-    printer.warn('Not implemented');
+    const amplifyMeta = stateManager.getMeta();
+    const stackName = amplifyMeta?.providers?.awscloudformation?.StackName;
+
+    if (!stackName) {
+      throw new AmplifyError('StackNotFoundError', {
+        message: 'Root stack not found',
+        resolution: 'Ensure the project is initialized and deployed.',
+      });
+    }
+
+    const cfnClient = new CloudFormationClient({});
+    const response = await cfnClient.send(new DescribeStacksCommand({ StackName: stackName }));
+
+    if (!response.Stacks || response.Stacks.length === 0) {
+      throw new AmplifyError('StackNotFoundError', {
+        message: `Stack ${stackName} not found in CloudFormation`,
+        resolution: 'Ensure the project is deployed.',
+      });
+    }
+
+    const stackStatus = response.Stacks[0].StackStatus;
+    const validStatuses = ['UPDATE_COMPLETE', 'CREATE_COMPLETE'];
+
+    if (!validStatuses.includes(stackStatus)) {
+      throw new AmplifyError('StackStateError', {
+        message: `Root stack status is ${stackStatus}, expected ${validStatuses.join(' or ')}`,
+        resolution: 'Complete the deployment before proceeding.',
+      });
+    }
   }
 
   public async validateDeploymentVersion(): Promise<void> {