Use Timeout from SAM Template

CHANGELOG.md

@@ -5,6 +5,7 @@ All notable changes to the "aws-vscode-tools" extension will be documented in th
 ## NEXT (Developer Preview)
 
 * Local Run/Debug now honors MemorySize values from SAM Template file (#509)
+* Local Run/Debug now honors Timeout values from SAM Template file (#510)
 * Fixed issue preventing users from connecting with assumed roles (#620)
 
 ## 0.2.0 (Developer Preview)

src/shared/codelens/localLambdaRunner.ts

@@ -342,19 +342,23 @@ export async function makeInputTemplate(params: {
     properties?: CloudFormation.ResourceProperties
     runtime: string
 }): Promise<string> {
-    const newTemplate = new SamTemplateGenerator()
+    let newTemplate = new SamTemplateGenerator()
         .withFunctionHandler(params.relativeFunctionHandler)
         .withResourceName(TEMPLATE_RESOURCE_NAME)
         .withRuntime(params.runtime)
         .withCodeUri(params.codeDir)
 
     if (params.properties) {
         if (params.properties.Environment) {
-            newTemplate.withEnvironment(params.properties.Environment)
+            newTemplate = newTemplate.withEnvironment(params.properties.Environment)
         }
 
         if (params.properties.MemorySize) {
-            newTemplate.withMemorySize(params.properties.MemorySize)
+            newTemplate = newTemplate.withMemorySize(params.properties.MemorySize)
+        }
+
+        if (params.properties.Timeout) {
+            newTemplate = newTemplate.withTimeout(params.properties.Timeout)
         }
     }
 

src/shared/templates/sam/samTemplateGenerator.ts

@@ -45,6 +45,12 @@ export class SamTemplateGenerator {
         return this
     }
 
+    public withTimeout(timeout: number): SamTemplateGenerator {
+        this.properties.Timeout = timeout
+
+        return this
+    }
+
     public withEnvironment(env: CloudFormation.Environment): SamTemplateGenerator {
         this.properties.Environment = env
 

src/test/templates/sam/samTemplateGenerator.test.ts

@@ -18,6 +18,7 @@ describe('SamTemplateGenerator', () => {
     const sampleFunctionHandlerValue: string = 'sampleFunctionHandler'
     const sampleResourceNameValue: string = 'sampleResourceName'
     const sampleMemorySize: number = 256
+    const sampleTimeout: number = 321
     const sampleRuntimeValue: string = 'sampleRuntime'
     const sampleEnvironment: CloudFormation.Environment = {
         Variables: {
@@ -43,6 +44,7 @@ describe('SamTemplateGenerator', () => {
             .withRuntime(sampleRuntimeValue)
             .withResourceName(sampleResourceNameValue)
             .withMemorySize(sampleMemorySize)
+            .withTimeout(sampleTimeout)
             .withEnvironment(sampleEnvironment)
             .generate(templateFilename)
 
@@ -57,6 +59,7 @@ describe('SamTemplateGenerator', () => {
         assert.strictEqual(resource!.Properties!.CodeUri, sampleCodeUriValue)
         assert.strictEqual(resource!.Properties!.Handler, sampleFunctionHandlerValue)
         assert.strictEqual(resource!.Properties!.MemorySize, sampleMemorySize)
+        assert.strictEqual(resource!.Properties!.Timeout, sampleTimeout)
         assert.strictEqual(resource!.Properties!.Runtime, sampleRuntimeValue)
         assert.deepStrictEqual(resource!.Properties!.Environment, sampleEnvironment)
     })