Use OSError instead of socket_error

Author: Juliya Smith

c42seceventcli/aed/main.py

@@ -1,5 +1,4 @@
 import json
-from socket import error as socket_error
 from urllib3 import disable_warnings
 from urllib3.exceptions import InsecureRequestWarning
 from datetime import datetime, timedelta
@@ -87,7 +86,7 @@ def _get_logger(
             destination_port=destination_port,
             destination_protocol=destination_protocol,
         )
-    except socket_error as ex:
+    except OSError as ex:
         print(repr(ex))
         print(
             "Hostname={0}, port={1}, protocol={2}.".format(

tests/aed/test_main.py

@@ -1,6 +1,5 @@
 import pytest
 from datetime import datetime, timedelta
-from socket import error as socket_error
 
 from py42 import settings
 import py42.debug_level as debug_level
@@ -240,8 +239,8 @@ def test_main_when_get_logger_raises_io_error_causes_exit(patches):
         main.main()
 
 
-def test_main_when_get_logger_raises_socket_error_causes_exit(patches):
-    patches.get_logger.side_effect = socket_error
+def test_main_when_get_logger_raises_os_error_causes_exit(patches):
+    patches.get_logger.side_effect = OSError
     with pytest.raises(SystemExit):
         main.main()