Test example (#3)

* pytest-watcher

* ignore coverage files

* test handler

* pass test_handler

.gitignore

@@ -9,3 +9,5 @@ dist/
 # Environment info
 .env
 athena_federation/.DS_Store
+coverage.xml
+.coverage

Makefile

@@ -14,6 +14,10 @@ update:
 test:
 	poetry run pytest --cov-report xml --cov=athena_federation
 
+
+watch:
+	poetry run ptw . --now
+
 # Lint code using flake8
 lint:
 	black .

athena_federation/lambda_handler.py

@@ -27,15 +27,21 @@ def process_event(self, event):
         # Populate attributes needed to process the event
         self.event = event
         self.catalog_name = self.event.get("catalogName")
-        request_type = self.event.get("@type")
+        request_type = self.event.get("@type") or "PingRequest"
 
         # TODO: As the event comes in, populate the database name, table name if provided.
         # TODO: I can also create new Request types from the event
 
         # Look up the request type, call it dynamically, and return the dictionary representation of it.
         # Each model returned implements `as_dict` that returns the info necessary for Athena, including
         # specific PyArrow serialization.
-        return getattr(self, request_type)().as_dict()
+        print(dir(self))
+        request_attr = getattr(self, request_type)
+        if request_attr:
+            return request_attr().as_dict()
+        else:
+            print(f"Unknown request type: {request_type}")
+            return {}
 
     def PingRequest(self) -> models.PingResponse:
         return models.PingResponse(