Skip to content

Commit d29cb31

Browse files
committed
fix params
1 parent acab870 commit d29cb31

File tree

2 files changed

+11
-16
lines changed

2 files changed

+11
-16
lines changed

packages/slackBotFunction/app/slack/slack_handlers.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -86,7 +86,7 @@ def mention_handler(event: Dict[str, Any], body: Dict[str, Any], client: WebClie
8686
)
8787

8888

89-
def dm_message_handler(event: Dict[str, Any], event_id: str, client: WebClient, body: Dict[str, Any]) -> None:
89+
def dm_message_handler(event: Dict[str, Any], event_id: str, client: WebClient) -> None:
9090
"""
9191
Direct messages:
9292
- 'feedback:' prefix -> store as conversation-scoped additional feedback (no model call).
@@ -109,7 +109,7 @@ def dm_message_handler(event: Dict[str, Any], event_id: str, client: WebClient,
109109
)
110110

111111

112-
def thread_message_handler(event: Dict[str, Any], event_id: str, client: WebClient, body: Dict[str, Any]) -> None:
112+
def thread_message_handler(event: Dict[str, Any], event_id: str, client: WebClient) -> None:
113113
"""
114114
Thread messages:
115115
- Ignore top-level messages (policy: require @mention to start).
@@ -161,10 +161,10 @@ def unified_message_handler(event: Dict[str, Any], body: Dict[str, Any], client:
161161
# Route to appropriate handler based on message type
162162
if event.get("channel_type") == constants.CHANNEL_TYPE_IM:
163163
# DM handling
164-
dm_message_handler(event=event, event_id=event_id, client=client, body=body)
164+
dm_message_handler(event=event, event_id=event_id, client=client)
165165
else:
166166
# Channel message handling
167-
thread_message_handler(event=event, event_id=event_id, client=client, body=body)
167+
thread_message_handler(event=event, event_id=event_id, client=client)
168168

169169

170170
def feedback_handler(body: Dict[str, Any], client: WebClient, event: Dict[str, Any]) -> None:

packages/slackBotFunction/tests/test_slack_handlers.py

Lines changed: 7 additions & 12 deletions
Original file line numberDiff line numberDiff line change
@@ -180,7 +180,7 @@ def test_unified_message_handler_handles_dm_messages(
180180

181181
# assertions
182182
mock_thread_message_handler.assert_not_called()
183-
mock_dm_message_handler.assert_called_with(event=mock_event, event_id="evt123", client=ANY, body=mock_body)
183+
mock_dm_message_handler.assert_called_with(event=mock_event, event_id="evt123", client=ANY)
184184

185185

186186
@patch("app.utils.handler_utils.is_duplicate_event")
@@ -208,7 +208,7 @@ def test_unified_message_handler_handles_threadable_messages(
208208

209209
# assertions
210210
mock_dm_message_handler.assert_not_called()
211-
mock_thread_message_handler.assert_called_with(event=mock_event, event_id="evt123", client=ANY, body=mock_body)
211+
mock_thread_message_handler.assert_called_with(event=mock_event, event_id="evt123", client=ANY)
212212

213213

214214
# _common_message_handler
@@ -504,7 +504,6 @@ def test_thread_message_handler_session_check_error(
504504
"ts": "evt_ts",
505505
"user": "U456",
506506
}
507-
mock_body = {}
508507

509508
mock_get_state_information.side_effect = Exception("DB error")
510509

@@ -516,7 +515,7 @@ def test_thread_message_handler_session_check_error(
516515
# perform operation
517516
# Should return early due to error
518517
with patch("app.slack.slack_handlers._common_message_handler") as mock_common_message_handler:
519-
thread_message_handler(mock_event, "evt123", mock_client, mock_body)
518+
thread_message_handler(event=mock_event, event_id="evt123", client=mock_client)
520519

521520
# assertions
522521
mock_common_message_handler.assert_not_called()
@@ -532,7 +531,6 @@ def test_thread_message_handler_no_session(
532531
# setup mocks
533532
mock_client = Mock()
534533
mock_event = {"text": "follow up", "channel": "C789", "thread_ts": "123", "user": "U456"}
535-
mock_body = {}
536534

537535
mock_get_state_information.return_value = {} # No session
538536

@@ -543,7 +541,7 @@ def test_thread_message_handler_no_session(
543541

544542
# perform operation
545543
with patch("app.slack.slack_handlers._common_message_handler") as mock_common_message_handler:
546-
thread_message_handler(mock_event, "evt123", mock_client, mock_body)
544+
thread_message_handler(event=mock_event, event_id="evt123", client=mock_client)
547545

548546
# assertions
549547
mock_common_message_handler.assert_not_called()
@@ -556,7 +554,6 @@ def test_thread_message_handler_success(mock_get_state_information: Mock, mock_e
556554
# setup mocks
557555
mock_client = Mock()
558556
mock_event = {"text": "feedback: channel feedback", "channel": "C789", "thread_ts": "123", "user": "U456"}
559-
mock_body = {}
560557

561558
mock_get_state_information.return_value = {"Item": {"session_id": "session123"}}
562559

@@ -567,7 +564,7 @@ def test_thread_message_handler_success(mock_get_state_information: Mock, mock_e
567564

568565
# perform operation
569566
with patch("app.slack.slack_handlers._common_message_handler") as mock_common_message_handler:
570-
thread_message_handler(mock_event, "evt123", mock_client, mock_body)
567+
thread_message_handler(event=mock_event, event_id="evt123", client=mock_client)
571568
# assertions
572569
mock_common_message_handler.assert_called_once_with(
573570
message_text="feedback: channel feedback",
@@ -849,7 +846,6 @@ def test_dm_message_handler_success(mock_get_parameter: Mock, mock_env: Mock):
849846
# setup mocks
850847
mock_client = Mock()
851848
mock_event = {"text": "dm message", "user": "U456", "channel": "D789", "ts": "123", "channel_type": "im"}
852-
mock_body = {}
853849

854850
# delete and import module to test
855851
if "app.slack.slack_handlers" in sys.modules:
@@ -858,7 +854,7 @@ def test_dm_message_handler_success(mock_get_parameter: Mock, mock_env: Mock):
858854

859855
# perform operation
860856
with patch("app.slack.slack_handlers._common_message_handler") as mock_common_message_handler:
861-
dm_message_handler(event=mock_event, event_id="evt123", client=mock_client, body=mock_body)
857+
dm_message_handler(event=mock_event, event_id="evt123", client=mock_client)
862858

863859
# assertions
864860
mock_common_message_handler.assert_called_once_with(
@@ -877,7 +873,6 @@ def test_dm_message_handler_not_a_dm(mock_get_parameter: Mock, mock_env: Mock):
877873
# setup mocks
878874
mock_client = Mock()
879875
mock_event = {"text": "dm message", "user": "U456", "channel": "im", "ts": "123", "channel_type": "not_im"}
880-
mock_body = {}
881876

882877
# delete and import module to test
883878
if "app.slack.slack_handlers" in sys.modules:
@@ -886,7 +881,7 @@ def test_dm_message_handler_not_a_dm(mock_get_parameter: Mock, mock_env: Mock):
886881

887882
# perform operation
888883
with patch("app.slack.slack_handlers._common_message_handler") as mock_common_message_handler:
889-
dm_message_handler(event=mock_event, event_id="evt123", client=mock_client, body=mock_body)
884+
dm_message_handler(event=mock_event, event_id="evt123", client=mock_client)
890885

891886
# assertions
892887
mock_common_message_handler.assert_not_called()

0 commit comments

Comments
 (0)