@@ -49,10 +49,20 @@ def socket_mocks(mocker):
4949
5050
5151@pytest .fixture ()
52- def broken_pipe_error (mocker ):
53- mock_exc_info = mocker .patch ("code42cli.logger.handlers.sys.exc_info" )
54- mock_exc_info .return_value = (BrokenPipeError , None , None )
55- return mock_exc_info
52+ def system_exception_info (mocker ):
53+ return mocker .patch ("code42cli.logger.handlers.sys.exc_info" )
54+
55+
56+ @pytest .fixture ()
57+ def broken_pipe_error (system_exception_info ):
58+ system_exception_info .return_value = (BrokenPipeError , None , None )
59+ return system_exception_info
60+
61+
62+ @pytest .fixture ()
63+ def connection_reset_error (system_exception_info ):
64+ system_exception_info .return_value = (ConnectionResetError , None , None )
65+ return system_exception_info
5666
5767
5868def _get_normal_socket_initializer_mocks (mocker , new_socket ):
@@ -204,7 +214,7 @@ def test_emit_when_udp_calls_socket_sendto_with_expected_message_and_address(
204214 expected_message , (_TEST_HOST , _TEST_PORT )
205215 )
206216
207- def test_handle_error_raises_expected_error (
217+ def test_handle_error_when_broken_pipe_error_occurs_raises_expected_error (
208218 self , mock_file_event_log_record , broken_pipe_error
209219 ):
210220 handler = NoPrioritySysLogHandler (
@@ -213,6 +223,15 @@ def test_handle_error_raises_expected_error(
213223 with pytest .raises (SyslogServerNetworkConnectionError ):
214224 handler .handleError (mock_file_event_log_record )
215225
226+ def test_handle_error_when_connection_reset_error_occurs_raises_expected_error (
227+ self , mock_file_event_log_record , connection_reset_error
228+ ):
229+ handler = NoPrioritySysLogHandler (
230+ _TEST_HOST , _TEST_PORT , ServerProtocol .UDP , None
231+ )
232+ with pytest .raises (SyslogServerNetworkConnectionError ):
233+ handler .handleError (mock_file_event_log_record )
234+
216235 def test_close_when_using_tls_unwraps_socket (self ):
217236 handler = NoPrioritySysLogHandler (
218237 _TEST_HOST , _TEST_PORT , ServerProtocol .TLS_TCP , None
0 commit comments