Skip to content

Commit 161af12

Browse files
committed
remove test_dispatch_parallel test
1 parent 47e9851 commit 161af12

File tree

1 file changed

+0
-40
lines changed

1 file changed

+0
-40
lines changed

tests/mem_scheduler/test_dispatcher.py

Lines changed: 0 additions & 40 deletions
Original file line numberDiff line numberDiff line change
@@ -190,46 +190,6 @@ def test_dispatch_serial(self):
190190
self.assertEqual(len(label2_messages), 1)
191191
self.assertEqual(label2_messages[0].item_id, "msg2")
192192

193-
def test_dispatch_parallel(self):
194-
"""Test dispatching messages in parallel mode."""
195-
# Create fresh mock handlers for this test
196-
mock_handler1 = MagicMock()
197-
mock_handler2 = MagicMock()
198-
199-
# Create a new dispatcher for this test to avoid interference
200-
parallel_dispatcher = SchedulerDispatcher(max_workers=2, enable_parallel_dispatch=True)
201-
parallel_dispatcher.register_handler("label1", mock_handler1)
202-
parallel_dispatcher.register_handler("label2", mock_handler2)
203-
204-
# Dispatch messages
205-
parallel_dispatcher.dispatch(self.test_messages)
206-
207-
# Wait for all futures to complete
208-
parallel_dispatcher.join(timeout=1.0)
209-
210-
# Verify handlers were called - label1 handler should be called twice (for user1 and user2)
211-
# label2 handler should be called once (only for user1)
212-
self.assertEqual(mock_handler1.call_count, 2) # Called for user1/msg1 and user2/msg3
213-
mock_handler2.assert_called_once() # Called for user1/msg2
214-
215-
# Check that each handler received the correct messages
216-
# For label1: should have two calls, each with one message
217-
label1_calls = mock_handler1.call_args_list
218-
self.assertEqual(len(label1_calls), 2)
219-
220-
# Extract messages from calls
221-
call1_messages = label1_calls[0][0][0] # First call, first argument (messages list)
222-
call2_messages = label1_calls[1][0][0] # Second call, first argument (messages list)
223-
224-
# Verify the messages in each call
225-
self.assertEqual(len(call1_messages), 1)
226-
self.assertEqual(len(call2_messages), 1)
227-
228-
# For label2: should have one call with [msg2]
229-
label2_messages = mock_handler2.call_args[0][0]
230-
self.assertEqual(len(label2_messages), 1)
231-
self.assertEqual(label2_messages[0].item_id, "msg2")
232-
233193
def test_group_messages_by_user_and_mem_cube(self):
234194
"""Test grouping messages by user and cube."""
235195
# Check actual grouping logic

0 commit comments

Comments
 (0)