@@ -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