From cf2a95a88a51f1b8bc012ac00d4ddbb82eaeb9e0 Mon Sep 17 00:00:00 2001
From: Kapil Thangavelu <kapilt@gmail.com>
Date: Mon, 3 Jun 2019 19:10:35 -0400
Subject: [PATCH] add layer support

---
 chalice/package.py         | 2 ++
 tests/unit/test_package.py | 7 +++++++
 2 files changed, 9 insertions(+)

diff --git a/chalice/package.py b/chalice/package.py
index 8e394040f..faa1846d7 100644
--- a/chalice/package.py
+++ b/chalice/package.py
@@ -541,6 +541,8 @@ def _generate_lambdafunction(self, resource, template):
             func_definition['environment'] = {
                 'variables': resource.environment_variables
             }
+        if resource.layers:
+            func_definition['layers'] = list(resource.layers)
 
         if isinstance(resource.role, models.ManagedIAMRole):
             func_definition['role'] = '${aws_iam_role.%s.arn}' % (
diff --git a/tests/unit/test_package.py b/tests/unit/test_package.py
index 53bfa1350..5d92a83c0 100644
--- a/tests/unit/test_package.py
+++ b/tests/unit/test_package.py
@@ -194,6 +194,13 @@ def test_adds_vpc_config_when_provided(self):
             'subnet_ids': ['sn1', 'sn2'],
             'security_group_ids': ['sg1', 'sg2']}
 
+    def test_adds_layers_when_provided(self):
+        function = self.lambda_function()
+        function.layers = layers = ['arn://layer1', 'arn://layer2']
+        template = self.template_gen.generate([function])
+        tf_resource = self.get_function(template)
+        assert tf_resource['layers'] == layers
+
     def test_adds_reserved_concurrency_when_provided(self, sample_app):
         function = self.lambda_function()
         function.reserved_concurrency = 5