Merge pull request #3 from studio3104/change_member_pvivacy

Encapsulate the context variable

Author: studio3104

pytest_chalice.py

@@ -21,6 +21,14 @@ def __init__(self, *args, **kwargs):
         self.custom_context = {}
         super(InternalLocalGateway, self).__init__(*args, **kwargs)
 
+    @property
+    def custom_context(self):
+        return self.__custom_context
+
+    @custom_context.setter
+    def custom_context(self, context):
+        self.__custom_context = context
+
     def _generate_lambda_event(self, *args, **kwargs):
         event = super(InternalLocalGateway, self)._generate_lambda_event(*args, **kwargs)
         event['requestContext'].update(self.custom_context)
@@ -30,7 +38,7 @@ def _generate_lambda_event(self, *args, **kwargs):
 UPPERCASE_PATTERN = re.compile('([A-Z])')
 
 
-class ResponseHandler:
+class ResponseHandler(object):
     def __init__(self, values):
         self.values = {}