feat: support snap_start configuration by function

Includes tests

Author: DefSol

main.tf

@@ -101,6 +101,13 @@ resource "aws_lambda_function" "lambda_application" {
     }
   }
 
+  dynamic "snap_start" {
+    for_each = each.value.enable_snap_start ? [true] : []
+    content {
+      apply_on = "PublishedVersions"
+    }
+  }
+
   tags = local.tags
 }
 

tests/main.tftest.hcl

@@ -16,12 +16,13 @@ variables {
       description = "API handler function"
     }
     worker = {
-      handler          = "worker/index.handler"
-      enable_vpc       = true
-      description      = "Background worker function"
-      function_memory  = "512"
-      function_timeout = 60
-      s3_key           = "worker.zip"
+      handler           = "worker/index.handler"
+      enable_vpc        = true
+      description       = "Background worker function"
+      function_memory   = "512"
+      function_timeout  = 60
+      s3_key            = "worker.zip"
+      enable_snap_start = true
     }
   }
 
@@ -135,6 +136,21 @@ run "lambda_function_vpc_configuration" {
   }
 }
 
+run "lambda_function_snap_start" {
+
+command = plan
+
+  assert {
+    condition     = length(aws_lambda_function.lambda_application["api"].snap_start) == 0
+    error_message = "API function should not have snap start configuration"
+  }
+
+  assert {
+    condition     = length(aws_lambda_function.lambda_application["worker"].snap_start) > 0
+    error_message = "Worker function should have snap start configuration"
+  }
+}
+
 run "service_catalog_application_registration" {
   command = plan
 

variables.tf

@@ -24,6 +24,7 @@ variable "lambda_functions_config" {
     application_log_level = optional(string)
     system_log_level      = optional(string)
     s3_key                = optional(string)
+    enable_snap_start     = optional(bool, false)
 
     function_concurrency_limit = optional(number)
   }))