Better error handling:

Author: Juliya Smith

c42seceventcli/aed/main.py

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

tests/aed/test_main.py

@@ -1,5 +1,6 @@
 import pytest
 from datetime import datetime, timedelta
+from socket import herror, gaierror, timeout
 
 from py42 import settings
 import py42.debug_level as debug_level
@@ -239,8 +240,20 @@ def test_main_when_get_logger_raises_io_error_causes_exit(patches):
         main.main()
 
 
-def test_main_when_get_logger_raises_os_error_causes_exit(patches):
-    patches.get_logger.side_effect = OSError
+def test_main_when_get_logger_raises_h_error_causes_exit(patches):
+    patches.get_logger.side_effect = gaierror
+    with pytest.raises(SystemExit):
+        main.main()
+
+
+def test_main_when_get_logger_raises_gai_error_causes_exit(patches):
+    patches.get_logger.side_effect = herror
+    with pytest.raises(SystemExit):
+        main.main()
+
+
+def test_main_when_get_logger_raises_timeout_causes_exit(patches):
+    patches.get_logger.side_effect = timeout
     with pytest.raises(SystemExit):
         main.main()