from flexmock import flexmock

from masters_registry import config
from masters_registry.connectors import loggly


def test_loggly_dsn_disabled():
    """Test that LogglyDSNHandler is not used when disabled in config
    """
    flexmock(config).should_receive('LOGGER_DSN').and_return(None)
    handler_mock = flexmock(loggly.CeleryDSNHandler)
    (flexmock(loggly)
     .should_receive('CeleryDSNHandler')
     .and_return(handler_mock))
    logger_mock = flexmock()
    (logger_mock
     .should_receive('setLevel')
     .with_args(config.CELERY_LOGGER_LEVEL)
     .once())
    logger_mock.should_receive('addHandler').with_args(handler_mock).never()
    loggly.initialize_celery_logging(logger_mock, None, None)


def test_loggly_dsn_enabled():
    """Test that LogglyDSNHandler is created when enabled in config
    """
    flexmock(config).should_receive('LOGGER_DSN').and_return(True)
    handler_mock = flexmock(loggly.CeleryDSNHandler)
    (flexmock(loggly)
     .should_receive('CeleryDSNHandler')
     .and_return(handler_mock))
    logger_mock = flexmock()
    logger_mock.should_receive('addHandler').with_args(handler_mock).once()
    (logger_mock
     .should_receive('setLevel')
     .with_args(config.CELERY_LOGGER_LEVEL)
     .once())
    loggly.initialize_celery_logging(logger_mock, None, None)