Reformat code with black

Author: kris-szlapa

packages/slackBotFunction/app.py

@@ -134,7 +134,11 @@ def process_async_slack_event(slack_event_data):
 
         logger.info(
             f"Processing async @mention from user {user_id}",
-            extra={"user_query": user_query, "thread_ts": thread_ts, "event_id": event_id},
+            extra={
+                "user_query": user_query,
+                "thread_ts": thread_ts,
+                "event_id": event_id,
+            },
         )
 
         if not user_query:

packages/slackBotFunction/tests/test_app.py

@@ -89,7 +89,10 @@ def test_is_duplicate_event(mock_time, mock_boto_resource, mock_get_parameter, m
     # Mock ConditionalCheckFailedException
     from botocore.exceptions import ClientError
 
-    error = ClientError(error_response={"Error": {"Code": "ConditionalCheckFailedException"}}, operation_name="PutItem")
+    error = ClientError(
+        error_response={"Error": {"Code": "ConditionalCheckFailedException"}},
+        operation_name="PutItem",
+    )
     mock_table.put_item.side_effect = error
 
     if "app" in sys.modules:
@@ -362,7 +365,12 @@ def test_process_async_slack_event_success(mock_webclient, mock_boto_resource, m
         from app import process_async_slack_event
 
         slack_event_data = {
-            "event": {"text": "<@U123> test question", "user": "U456", "channel": "C789", "ts": "1234567890.123"},
+            "event": {
+                "text": "<@U123> test question",
+                "user": "U456",
+                "channel": "C789",
+                "ts": "1234567890.123",
+            },
             "event_id": "evt123",
             "bot_token": "bot-token",
         }
@@ -438,7 +446,12 @@ def test_process_async_slack_event_error(mock_webclient, mock_boto_resource, moc
         from app import process_async_slack_event
 
         slack_event_data = {
-            "event": {"text": "test question", "user": "U456", "channel": "C789", "ts": "1234567890.123"},
+            "event": {
+                "text": "test question",
+                "user": "U456",
+                "channel": "C789",
+                "ts": "1234567890.123",
+            },
             "event_id": "evt123",
             "bot_token": "bot-token",
         }
@@ -585,7 +598,12 @@ def test_log_request_middleware_execution_fixed(mock_boto_resource, mock_get_par
 @patch("boto3.client")
 @patch("time.time")
 def test_app_mention_handler_execution_simple(
-    mock_time, mock_boto_client, mock_boto_resource, mock_get_parameter, mock_app_class, mock_env
+    mock_time,
+    mock_boto_client,
+    mock_boto_resource,
+    mock_get_parameter,
+    mock_app_class,
+    mock_env,
 ):
     """Test app mention handler execution by simulating the handler registration process"""
     mock_get_parameter.side_effect = [
@@ -643,7 +661,10 @@ def decorator(func):
     # Test 2: Duplicate event
     from botocore.exceptions import ClientError
 
-    error = ClientError(error_response={"Error": {"Code": "ConditionalCheckFailedException"}}, operation_name="PutItem")
+    error = ClientError(
+        error_response={"Error": {"Code": "ConditionalCheckFailedException"}},
+        operation_name="PutItem",
+    )
     mock_table.put_item.side_effect = error
 
     handler_func(event, mock_ack, body)
@@ -668,7 +689,12 @@ def decorator(func):
 @patch("boto3.client")
 @patch("time.time")
 def test_direct_message_handler_execution_simple(
-    mock_time, mock_boto_client, mock_boto_resource, mock_get_parameter, mock_app_class, mock_env
+    mock_time,
+    mock_boto_client,
+    mock_boto_resource,
+    mock_get_parameter,
+    mock_app_class,
+    mock_env,
 ):
     """Test direct message handler execution by simulating the handler registration process"""
     mock_get_parameter.side_effect = [
@@ -737,7 +763,10 @@ def decorator(func):
     # Test 3: IM channel with duplicate event
     from botocore.exceptions import ClientError
 
-    error = ClientError(error_response={"Error": {"Code": "ConditionalCheckFailedException"}}, operation_name="PutItem")
+    error = ClientError(
+        error_response={"Error": {"Code": "ConditionalCheckFailedException"}},
+        operation_name="PutItem",
+    )
     mock_table.put_item.side_effect = error
 
     handler_func(event, mock_ack, body)
@@ -762,7 +791,12 @@ def decorator(func):
 @patch("boto3.client")
 @patch("time.time")
 def test_handlers_direct_call_coverage(
-    mock_time, mock_boto_client, mock_boto_resource, mock_get_parameter, mock_app_class, mock_env
+    mock_time,
+    mock_boto_client,
+    mock_boto_resource,
+    mock_get_parameter,
+    mock_app_class,
+    mock_env,
 ):
     """Test handlers by calling them directly to ensure coverage"""
     mock_get_parameter.side_effect = [

packages/syncKnowledgeBaseFunction/tests/test_app.py

@@ -33,7 +33,10 @@ def s3_event():
             {
                 "eventSource": "aws:s3",
                 "eventName": "ObjectCreated:Put",
-                "s3": {"bucket": {"name": "test-bucket"}, "object": {"key": "test-file.pdf", "size": 1024}},
+                "s3": {
+                    "bucket": {"name": "test-bucket"},
+                    "object": {"key": "test-file.pdf", "size": 1024},
+                },
             }
         ]
     }
@@ -157,12 +160,18 @@ def test_handler_multiple_records(mock_time, mock_bedrock, mock_env, lambda_cont
             {
                 "eventSource": "aws:s3",
                 "eventName": "ObjectCreated:Put",
-                "s3": {"bucket": {"name": "test-bucket"}, "object": {"key": "file1.pdf", "size": 1024}},
+                "s3": {
+                    "bucket": {"name": "test-bucket"},
+                    "object": {"key": "file1.pdf", "size": 1024},
+                },
             },
             {
                 "eventSource": "aws:s3",
                 "eventName": "ObjectRemoved:Delete",
-                "s3": {"bucket": {"name": "test-bucket"}, "object": {"key": "file2.pdf", "size": 2048}},
+                "s3": {
+                    "bucket": {"name": "test-bucket"},
+                    "object": {"key": "file2.pdf", "size": 2048},
+                },
             },
         ]
     }
@@ -206,7 +215,10 @@ def test_handler_missing_object_size(mock_bedrock, mock_env, lambda_context):
             {
                 "eventSource": "aws:s3",
                 "eventName": "ObjectCreated:Put",
-                "s3": {"bucket": {"name": "test-bucket"}, "object": {"key": "test-file.pdf"}},
+                "s3": {
+                    "bucket": {"name": "test-bucket"},
+                    "object": {"key": "test-file.pdf"},
+                },
             }
         ]
     }
@@ -234,7 +246,10 @@ def test_handler_partial_env_vars(mock_bedrock, lambda_context, s3_event):
 @patch("app.bedrock_agent")
 def test_handler_client_error_no_message(mock_bedrock, mock_env, lambda_context, s3_event):
     """Test handler with ClientError missing message"""
-    error = ClientError(error_response={"Error": {"Code": "TestError"}}, operation_name="StartIngestionJob")
+    error = ClientError(
+        error_response={"Error": {"Code": "TestError"}},
+        operation_name="StartIngestionJob",
+    )
     mock_bedrock.start_ingestion_job.side_effect = error
 
     result = app.handler(s3_event, lambda_context)