Fix CI

Author: k-raina

tests/test_Consumer.py

@@ -329,12 +329,12 @@ def test_consumer_without_groupid():
 def test_callback_exception_no_system_error():
 
     exception_raised = []
-    
+
     def error_cb_that_raises(error):
         """Error callback that raises an exception"""
         exception_raised.append(error)
         raise RuntimeError("Test exception from error_cb")
-    
+
     # Create consumer with error callback that raises exception
     consumer = TestConsumer({
         'group.id': 'test-callback-systemerror-fix',
@@ -343,19 +343,19 @@ def error_cb_that_raises(error):
         'session.timeout.ms': 1000,
         'error_cb': error_cb_that_raises
     })
-    
+
     consumer.subscribe(['test-topic'])
-    
+
     # This should trigger the error callback due to connection failure
     # Before fix: Would get RuntimeError + SystemError
     # After fix: Should only get RuntimeError (no SystemError)
     with pytest.raises(RuntimeError) as exc_info:
         consumer.consume(timeout=0.1)
-    
+
     # Verify we got the expected exception message
     assert "Test exception from error_cb" in str(exc_info.value)
-    
+
     # Verify the error callback was actually called
     assert len(exception_raised) > 0
-    
+
     consumer.close()

tests/test_Producer.py

@@ -287,30 +287,30 @@ def test_producer_bool_value():
 
 def test_callback_exception_no_system_error():
     delivery_reports = []
-    
+
     def delivery_cb_that_raises(err, msg):
         """Delivery report callback that raises an exception"""
         delivery_reports.append((err, msg))
         raise RuntimeError("Test exception from delivery_cb")
-    
+
     producer = Producer({
         'bootstrap.servers': 'nonexistent-broker:9092',  # Will cause delivery failures
         'socket.timeout.ms': 100,
         'message.timeout.ms': 10,  # Very short timeout to trigger delivery failure quickly
         'on_delivery': delivery_cb_that_raises
     })
-    
+
     # Produce a message - this will trigger delivery report callback when it fails
     producer.produce('test-topic', value='test-message')
-    
+
     # Flush to ensure delivery reports are processed
-    # Before fix: Would get RuntimeError + SystemError  
+    # Before fix: Would get RuntimeError + SystemError
     # After fix: Should only get RuntimeError (no SystemError)
     with pytest.raises(RuntimeError) as exc_info:
         producer.flush(timeout=2.0)  # Longer timeout to ensure delivery callback fires
-    
+
     # Verify we got an exception from our callback
     assert "Test exception from delivery_cb" in str(exc_info.value)
-    
+
     # Verify the delivery callback was actually called
     assert len(delivery_reports) > 0