feat: get_logs to accept datetime for consistency

Author: heitorlessa

tests/e2e/logger/test_logger.py

@@ -29,7 +29,7 @@ def test_basic_lambda_logs_visible(basic_handler_fn, basic_handler_fn_arn):
     # WHEN
     _, execution_time = data_fetcher.get_lambda_response(lambda_arn=basic_handler_fn_arn)
     filtered_logs = data_fetcher.get_logs(
-        lambda_function_name=basic_handler_fn, start_time=int(execution_time.timestamp()), log_client=cw_client
+        lambda_function_name=basic_handler_fn, start_time=execution_time, log_client=cw_client
     )
 
     # THEN

tests/e2e/utils/data_fetcher/logs.py

@@ -1,4 +1,5 @@
 import json
+from datetime import datetime
 from functools import lru_cache
 from typing import List, Optional, Union
 
@@ -24,10 +25,13 @@ class Log(BaseModel):
 
 @lru_cache(maxsize=10, typed=False)
 @retry(ValueError, delay=1, jitter=1, tries=20)
-def get_logs(lambda_function_name: str, log_client: CloudWatchClient, start_time: int, **kwargs: dict) -> List[Log]:
-    response = log_client.filter_log_events(logGroupName=f"/aws/lambda/{lambda_function_name}", startTime=start_time)
+def get_logs(lambda_function_name: str, log_client: CloudWatchClient, start_time: datetime) -> List[Log]:
+    response = log_client.filter_log_events(
+        logGroupName=f"/aws/lambda/{lambda_function_name}", startTime=int(start_time.timestamp())
+    )
     if not response["events"]:
         raise ValueError("Empty response from Cloudwatch Logs. Repeating...")
+
     filtered_logs = []
     for event in response["events"]:
         try: