refactor utils.handlers.HTTPSHandler's usage of django settings

awx/main/tests/functional/utils/test_handlers.py

@@ -1,10 +1,11 @@
 import json
 import logging
 
+from django.conf import LazySettings
 import pytest
 import requests
 
-from awx.main.utils.handlers import HTTPSHandler
+from awx.main.utils.handlers import BaseHTTPSHandler as HTTPSHandler, PARAM_NAMES
 from awx.main.utils.formatters import LogstashFormatter
 
 
@@ -25,31 +26,28 @@ def ok200_adapter():
 
 
 
-@pytest.mark.django_db
-@pytest.mark.parametrize('param, value', [
-    ('host', None),
-    ('port', None),
-    ('message_type', None),
-    ('username', None),
-    ('password', None),
-    ('enabled_loggers', ['awx', 'activity_stream', 'job_events', 'system_tracking']),
-    ('indv_facts', None),
-    ('enabled_flag', None)
-])
-def test_https_logging_handler_defaults(param, value):
+@pytest.mark.parametrize('param', PARAM_NAMES.keys())
+def test_https_logging_handler_defaults(param):
     handler = HTTPSHandler()
-    assert hasattr(handler, param) and getattr(handler, param) == value
+    assert hasattr(handler, param) and getattr(handler, param) is None
 
 
-@pytest.mark.django_db
-@pytest.mark.parametrize('param', ['host', 'port', 'message_type', 'username',
-                                   'password', 'indv_facts', 'enabled_flag'])
+@pytest.mark.parametrize('param', PARAM_NAMES.keys())
 def test_https_logging_handler_kwargs(param):
     handler = HTTPSHandler(**{param: 'EXAMPLE'})
     assert hasattr(handler, param) and getattr(handler, param) == 'EXAMPLE'
 
 
-@pytest.mark.django_db
+@pytest.mark.parametrize('param, django_settings_name', PARAM_NAMES.items())
+def test_https_logging_handler_from_django_settings(param, django_settings_name):
+    settings = LazySettings()
+    settings.configure(**{
+        django_settings_name: 'EXAMPLE'
+    })
+    handler = HTTPSHandler.from_django_settings(settings)
+    assert hasattr(handler, param) and getattr(handler, param) == 'EXAMPLE'
+
+
 def test_https_logging_handler_logstash_auth_info():
     handler = HTTPSHandler(message_type='logstash', username='bob', password='ansible')
     handler.add_auth_information()
@@ -58,7 +56,6 @@ def test_https_logging_handler_logstash_auth_info():
     assert handler.session.auth.password == 'ansible'
 
 
-@pytest.mark.django_db
 def test_https_logging_handler_splunk_auth_info():
     handler = HTTPSHandler(message_type='splunk', password='ansible')
     handler.add_auth_information()
@@ -66,7 +63,6 @@ def test_https_logging_handler_splunk_auth_info():
     assert handler.session.headers['Content-Type'] == 'application/json'
 
 
-@pytest.mark.django_db
 @pytest.mark.parametrize('host, port, normalized', [
     ('localhost', None, 'http://localhost'),
     ('localhost', 80, 'http://localhost'),
@@ -81,7 +77,6 @@ def test_https_logging_handler_http_host_format(host, port, normalized):
     assert handler.get_http_host() == normalized
 
 
-@pytest.mark.django_db
 @pytest.mark.parametrize('params, logger_name, expected', [
     ({'enabled_flag': False}, 'awx.main', True),  # skip all records if enabled_flag = False
     ({'host': '', 'enabled_flag': True}, 'awx.main', True),  # skip all records if the host is undefined
@@ -91,16 +86,14 @@ def test_https_logging_handler_http_host_format(host, port, normalized):
 ])
 def test_https_logging_handler_skip_log(params, logger_name, expected):
     handler = HTTPSHandler(**params)
-    # override the default loggers set by django_settings
-    handler.enabled_loggers = params.get('enabled_loggers')
     assert handler.skip_log(logger_name) is expected
 
 
-@pytest.mark.django_db
 @pytest.mark.parametrize('message_type', ['logstash', 'splunk'])
 def test_https_logging_handler_emit(ok200_adapter, message_type):
     handler = HTTPSHandler(host='127.0.0.1', enabled_flag=True,
-                           message_type=message_type)
+                           message_type=message_type,
+                           enabled_loggers=['awx', 'activity_stream', 'job_events', 'system_tracking'])
     handler.setFormatter(LogstashFormatter())
     handler.session.mount('http://', ok200_adapter)
     record = logging.LogRecord(
@@ -130,10 +123,10 @@ def test_https_logging_handler_emit(ok200_adapter, message_type):
     assert body['message'] == 'User joe logged in'
 
 
-@pytest.mark.django_db
 def test_https_logging_handler_emit_one_record_per_fact(ok200_adapter):
     handler = HTTPSHandler(host='127.0.0.1', enabled_flag=True,
-                           message_type='logstash', indv_facts=True)
+                           message_type='logstash', indv_facts=True,
+                           enabled_loggers=['awx', 'activity_stream', 'job_events', 'system_tracking'])
     handler.setFormatter(LogstashFormatter())
     handler.session.mount('http://', ok200_adapter)
     record = logging.LogRecord(

awx/main/utils/handlers.py

@@ -45,24 +45,22 @@ class HTTPSNullHandler(NullHandler):
         return super(HTTPSNullHandler, self).__init__()
 
 
-class HTTPSHandler(logging.Handler):
+class BaseHTTPSHandler(logging.Handler):
     def __init__(self, fqdn=False, **kwargs):
-        super(HTTPSHandler, self).__init__()
+        super(BaseHTTPSHandler, self).__init__()
         self.fqdn = fqdn
         self.async = kwargs.get('async', True)
         for fd in PARAM_NAMES:
-            # settings values take precedence over the input params
-            settings_name = PARAM_NAMES[fd]
-            settings_val = getattr(django_settings, settings_name, None)
-            if settings_val:
-                setattr(self, fd, settings_val)
-            elif fd in kwargs:
-                setattr(self, fd, kwargs[fd])
-            else:
-                setattr(self, fd, None)
+            setattr(self, fd, kwargs.get(fd, None))
         self.session = FuturesSession()
         self.add_auth_information()
 
+    @classmethod
+    def from_django_settings(cls, settings, *args, **kwargs):
+        for param, django_setting_name in PARAM_NAMES.items():
+            kwargs[param] = getattr(settings, django_setting_name, None)
+        return cls(*args, **kwargs)
+
     def get_full_message(self, record):
         if record.exc_info:
             return '\n'.join(traceback.format_exception(*record.exc_info))
@@ -155,3 +153,9 @@ class HTTPSHandler(logging.Handler):
             raise
         except:
             self.handleError(record)
+
+
+class HTTPSHandler(object):
+
+    def __new__(cls, *args, **kwargs):
+        return BaseHTTPSHandler.from_django_settings(django_settings, *args, **kwargs)