Apply type hinting for the actual logic

Author: studio3104

pytest_chalice/fixtures.py

@@ -1,10 +1,15 @@
 # -*- coding: utf-8 -*-
 
+from typing import Iterator
+
 import pytest
 
+from chalice import Chalice
+
 from .handlers import RequestHandler
 
 
 @pytest.fixture
 def client(app):
+    # type: (Chalice) -> Iterator[Chalice]
     yield RequestHandler(app)

pytest_chalice/handlers.py

@@ -1,9 +1,12 @@
 # -*- coding: utf-8 -*-
 
+from typing import Any, Callable, Dict
+
 import json
 from logging import getLogger
 import re
 
+from chalice import Chalice
 from chalice.config import Config
 from chalice.local import LocalGateway
 
@@ -16,18 +19,22 @@
 
 class InternalLocalGateway(LocalGateway):
     def __init__(self, *args, **kwargs):
+        # type: (Any, Any) -> None
         self.custom_context = {}
         super(InternalLocalGateway, self).__init__(*args, **kwargs)
 
     @property
     def custom_context(self):
+        # type: () -> Dict[str, Any]
         return self.__custom_context
 
     @custom_context.setter
     def custom_context(self, context):
+        # type: (Dict[str, Any]) -> None
         self.__custom_context = context
 
     def _generate_lambda_event(self, *args, **kwargs):
+        # type: (Any, Any) -> Dict[str, Any]
         event = super(InternalLocalGateway, self)._generate_lambda_event(*args, **kwargs)
         event['requestContext'].update(self.custom_context)
         return event
@@ -38,6 +45,7 @@ def _generate_lambda_event(self, *args, **kwargs):
 
 class ResponseHandler(object):
     def __init__(self, values):
+        # type: (Dict[str, Any]) -> None
         self.values = {}
 
         for key, value in values.items():
@@ -46,10 +54,11 @@ def __init__(self, values):
 
         try:
             self.values['json'] = json.loads(self.values['body'])
-        except JSONDecodeError:
+        except JSONDecodeError:  # type: ignore
             logger.info('Response body is NOT JSON decodable: {}'.format(self.values['body']))
 
     def __getattr__(self, key):
+        # type: (str) -> Any
         try:
             return self.values[key]
         except KeyError:
@@ -60,10 +69,12 @@ class RequestHandler(object):
     METHODS = ('get', 'head', 'post', 'options', 'put', 'delete', 'trace', 'patch', 'link', 'unlink')
 
     def __init__(self, app):
+        # type: (Chalice) -> None
         self.local_gateway = InternalLocalGateway(app, Config())
 
     @property
     def custom_context(self):
+        # type: () -> Dict[str, Any]
         return self.local_gateway.custom_context
 
     # As of Chalice version 1.8.0,
@@ -75,13 +86,16 @@ def custom_context(self):
     # Not only for this purpose, it's an interface provided to allow custom contexts in unit tests.
     @custom_context.setter
     def custom_context(self, context):
+        # type: (Dict[str, Any]) -> None
         self.local_gateway.custom_context = context
 
     def __getattr__(self, method):
+        # type: (str) -> Callable
         if method not in self.METHODS:
             raise AttributeError("'{}' object has no attribute '{}'".format(self.__class__.__name__, method))
 
         def request(path, headers={}, body=''):
+            # type: (str, Dict[str, str], str) -> ResponseHandler
             response = self.local_gateway.handle_request(method=method.upper(), path=path, headers=headers, body=body)
             return ResponseHandler(response)
 

pytest_chalice/plugin.py

@@ -1,9 +1,12 @@
 # -*- coding: utf-8 -*-
 
+from _pytest.config import Config
+
 from .fixtures import client  # noqa
 
 
 def pytest_configure(config):
+    # type: (Config) -> None
     config.addinivalue_line(
         'markers',
         'app: A Chalice application object',