fix: fix unit tests

Author: Blankll

src/commands/template.ts

@@ -11,7 +11,6 @@ export const template = async (
   const iac = parseYaml(getIacLocation(options.location));
 
   await setContext({ ...options, stackName, provider: iac.provider.name }, true);
-
   const { template } = generateStackTemplate(stackName, iac);
   if (typeof template === 'string') {
     logger.info(`\n${template}`);

src/stack/rosStack/bootstrap.ts

@@ -47,9 +47,6 @@ module.exports.handler = async (rawEvent, context) => {
   // 处理 Buffer 类型的事件
   const event = parseEvent(rawEvent);
 
-  console.log('Parsed Event:', JSON.stringify(event, null, 2));
-  console.log('Context:', JSON.stringify(context, null, 2));
-
   const commonResponse = {
     RequestId: event.requestId,
     LogicalResourceId: event.logicalResourceId,
@@ -74,8 +71,6 @@ module.exports.handler = async (rawEvent, context) => {
       delete rosResponse.Data;
     }
 
-    console.log('Constructed ROS Response:', JSON.stringify(rosResponse, null, 2));
-
     // 发送响应到 ROS 服务（如果提供了 ResponseURL）
     if (event.responseURL) {
       await sendResponse(event.responseURL, rosResponse);
@@ -123,7 +118,6 @@ module.exports.handler = async (rawEvent, context) => {
 
 // 使用原生 fetch API 发送响应
 async function sendResponse(responseUrl, responseBody) {
-  console.log(\`Sending response to: \${responseUrl}\`);
 
   try {
     const body = JSON.stringify(responseBody);

tests/commands/deploy.test.ts

@@ -1,13 +1,22 @@
 import { deploy } from '../../src/commands/deploy';
 
+const mockedIamInfo = jest.fn();
 const mockedDeployStack = jest.fn();
+
+jest.mock('../../src/common/imsClient', () => ({
+  ...jest.requireActual('../../src/common/imsClient'),
+  getIamInfo: (...rags: unknown[]) => mockedIamInfo(...rags),
+}));
+
 jest.mock('../../src/stack', () => ({
   ...jest.requireActual('../../src/stack'),
   deployStack: (...args: unknown[]) => mockedDeployStack(...args),
 }));
 
 describe('unit test for deploy command', () => {
   it('should construct valid context and deploy the stack when deploy with valid iac', async () => {
+    mockedIamInfo.mockResolvedValue({ accountId: '123456789012', region: 'cn-hangzhou' });
+
     const stackName = 'my-demo-stack';
 
     await deploy(stackName, {

tests/commands/template.test.ts

@@ -3,32 +3,56 @@ import { TemplateFormat } from '../../src/types';
 import { jsonTemplate, yamlTemplate } from '../fixtures/templateFixture';
 
 const mockedLogger = jest.fn();
+const mockedGetContext = jest.fn();
+
 jest.mock('../../src/common/logger', () => ({
   logger: { info: (...args: unknown[]) => mockedLogger(...args), debug: jest.fn() },
 }));
+
+jest.mock('../../src/common/imsClient', () => ({
+  getIamInfo: jest.fn().mockResolvedValue({ accountId: '123456789012', region: 'cn-hangzhou' }),
+}));
+
+jest.mock('../../src/common/context', () => ({
+  ...jest.requireActual('../../src/common/context'),
+  getContext: (...args: unknown[]) => mockedGetContext(...args),
+}));
+
 const stackName = 'printTemplateStack';
 const location = 'tests/fixtures/serverless-insight.yml';
 
 describe('Unit test for template command', () => {
   beforeEach(() => {
+    mockedGetContext.mockReturnValue({
+      accountId: '123456789012',
+      region: 'cn-hangzhou',
+      stage: 'default',
+      stackName,
+      provider: 'aliyun',
+    });
+  });
+
+  afterEach(() => {
     mockedLogger.mockRestore();
+    mockedGetContext.mockRestore();
   });
-  it('should print the template in JSON format by default', () => {
+
+  it('should print the template in JSON format by default', async () => {
     const options = {
       format: TemplateFormat.JSON,
       location,
       stage: undefined,
     };
 
-    template(stackName, options);
+    await template(stackName, options);
 
     expect(mockedLogger).toHaveBeenCalledWith(`\n${JSON.stringify(jsonTemplate, null, 2)}`);
   });
 
-  it('should print the template in YAML format when specified', () => {
+  it('should print the template in YAML format when specified', async () => {
     const options = { format: TemplateFormat.YAML, location, stage: undefined };
 
-    template(stackName, options);
+    await template(stackName, options);
 
     expect(mockedLogger).toHaveBeenCalledWith(yamlTemplate);
   });

tests/fixtures/deploy-fixtures/oneFcLargeCode.ts

@@ -210,6 +210,7 @@ export const largeCodeRos = {
         ServiceConfig: {
           FunctionComputeConfig: {
             FcVersion: '3.0',
+            FcRegionId: 'cn-hangzhou',
             FunctionName: {
               'Fn::GetAtt': ['hello_fn', 'FunctionName'],
             },
@@ -234,9 +235,7 @@ export const largeCodeRos = {
       DependsOn: ['sls_project', 'sls_logstore', 'sls_index', 'si_auto_artifacts_code_deployment'],
       Properties: {
         Code: {
-          OssBucketName: {
-            'Fn::Sub': 'si-bootstrap-artifacts-${ALIYUN::AccountId}-${ALIYUN::Region}',
-          },
+          OssBucketName: 'si-bootstrap-artifacts-123456789012-cn-hangzhou',
           OssObjectName: 'hello-fn/43cb4c356149762dbe507fc1baede172-large-artifact.zip',
         },
         EnvironmentVariables: {
@@ -262,9 +261,7 @@ export const largeCodeRos = {
     si_auto_artifacts_code_deployment: {
       Properties: {
         Parameters: {
-          destinationBucket: {
-            'Fn::Sub': 'si-bootstrap-artifacts-${ALIYUN::AccountId}-${ALIYUN::Region}',
-          },
+          destinationBucket: 'si-bootstrap-artifacts-123456789012-cn-hangzhou',
           retainOnCreate: false,
           sources: [
             {

tests/stack/deploy.test.ts

@@ -132,7 +132,7 @@ describe('Unit tests for stack deployment', () => {
 
   it('should create bucket and store code artifact to bucket when code size > 15MB', async () => {
     const stackName = 'my-large-code-stack';
-    mockedGetStore.mockReturnValue({ stackName });
+    mockedGetStore.mockReturnValue({ stackName, accountId: '123456789012', region: 'cn-hangzhou' });
     mockedRosStackDeploy.mockResolvedValueOnce(stackName);
 
     await deployStack(