format

Author: Joe Snell

lef/main.py

@@ -26,7 +26,7 @@ def handle_signal(signal, frame):
         sys.exit(0)
 
     def register(self, events, handler):
-        print('called')
+        print("called")
         response = post(
             url=f"https://{self.runtime_api_endpoint}/2020-01-01/extension/register",
             data={"events": [str(event) for event in events]},
@@ -40,13 +40,13 @@ def register(self, events, handler):
 
     def process_events(self, extension_id, handler):
         while True:
-            print('called')
+            print("called")
             response = get(
                 url=f"https://{self.runtime_api_endpoint}/2020-01-01/extension/event/next",
                 headers={"Lambda-Extension-Identifier": extension_id},
             ).json()
 
-            if response["eventType"] == 'SHUTDOWN':
+            if response["eventType"] == "SHUTDOWN":
                 sys.exit(0)
             else:
                 handler(response)

tests/test_main.py

@@ -6,6 +6,7 @@
 from tests.utils import FakeResponse
 import pytest
 
+
 class MainTest(TestCase):
     def setUp(self):
         self.original_env = os.environ
@@ -15,23 +16,16 @@ def tearDown(self):
         os.environ = self.original_env
         self.original_env = None
 
-    @patch('lef.main.post')
+    @patch("lef.main.post")
     @patch("lef.main.get")
     def test_main(self, mock_get, mock_post):
         mock_get.side_effect = [
-            FakeResponse(
-                json.dumps({"eventType": "INVOKE"})
-            ),
-            FakeResponse(
-                json.dumps({"eventType": "SHUTDOWN"})
-            )
+            FakeResponse(json.dumps({"eventType": "INVOKE"})),
+            FakeResponse(json.dumps({"eventType": "SHUTDOWN"})),
         ]
 
         mock_post.side_effect = [
-            FakeResponse(
-                '',
-                {"Lambda-Extension-Identifier": "lambda-ext-id"}
-            ),
+            FakeResponse("", {"Lambda-Extension-Identifier": "lambda-ext-id"}),
         ]
 
         handler = MagicMock()
@@ -41,13 +35,21 @@ def test_main(self, mock_get, mock_post):
             extension.register([EventType.INVOKE], handler)
 
         assert extension.runtime_api_endpoint == "testruntimeapi"
+        assert [call({"eventType": "INVOKE"})] == handler.mock_calls
         assert [
-            call({'eventType': 'INVOKE'})
-        ] == handler.mock_calls
-        assert [
-            call(url='https://testruntimeapi/2020-01-01/extension/event/next', headers={'Lambda-Extension-Identifier': 'lambda-ext-id'}),
-            call(url='https://testruntimeapi/2020-01-01/extension/event/next', headers={'Lambda-Extension-Identifier': 'lambda-ext-id'})
+            call(
+                url="https://testruntimeapi/2020-01-01/extension/event/next",
+                headers={"Lambda-Extension-Identifier": "lambda-ext-id"},
+            ),
+            call(
+                url="https://testruntimeapi/2020-01-01/extension/event/next",
+                headers={"Lambda-Extension-Identifier": "lambda-ext-id"},
+            ),
         ] == mock_get.mock_calls
         assert [
-            call(url='https://testruntimeapi/2020-01-01/extension/register', data={'events': ['EventType.INVOKE']}, headers={'Lambda-Extension-Name': 'lef'})
+            call(
+                url="https://testruntimeapi/2020-01-01/extension/register",
+                data={"events": ["EventType.INVOKE"]},
+                headers={"Lambda-Extension-Name": "lef"},
+            )
         ] == mock_post.mock_calls

tests/test_utils.py

@@ -5,7 +5,6 @@
 
 
 class UtilsTest(TestCase):
-
     @patch("lef.utils.request")
     def test_get(self, mock_request):
         mock_request.Request.return_value = "FakeRequestObject"

tests/utils.py

@@ -1,5 +1,6 @@
 import json
 
+
 class FakeResponse:
     def __init__(self, response, headers=None):
         self._response = response
@@ -11,4 +12,3 @@ def read(self):
 
     def json(self):
         return json.loads(self._response)
-