1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556 |
- # Licensed to the Apache Software Foundation (ASF) under one
- # or more contributor license agreements. See the NOTICE file
- # distributed with this work for additional information
- # regarding copyright ownership. The ASF licenses this file
- # to you under the Apache License, Version 2.0 (the
- # "License"); you may not use this file except in compliance
- # with the License. You may obtain a copy of the License at
- #
- # http://www.apache.org/licenses/LICENSE-2.0
- #
- # Unless required by applicable law or agreed to in writing,
- # software distributed under the License is distributed on an
- # "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
- # KIND, either express or implied. See the License for the
- # specific language governing permissions and limitations
- # under the License.
- import logging
- import unittest
- from unittest.mock import MagicMock
- from superset.utils.logging_configurator import LoggingConfigurator
- class TestLoggingConfigurator(unittest.TestCase):
- def reset_logging(self):
- # work around all of the import side-effects in superset
- logging.root.manager.loggerDict = {}
- logging.root.handlers = []
- def test_configurator_adding_handler(self):
- class MyEventHandler(logging.Handler):
- def __init__(self):
- super().__init__(level=logging.DEBUG)
- self.received = False
- def handle(self, record):
- if hasattr(record, "testattr"):
- self.received = True
- class MyConfigurator(LoggingConfigurator):
- def __init__(self, handler):
- self.handler = handler
- def configure_logging(self, app_config, debug_mode):
- super().configure_logging(app_config, debug_mode)
- logging.getLogger().addHandler(self.handler)
- self.reset_logging()
- handler = MyEventHandler()
- cfg = MyConfigurator(handler)
- cfg.configure_logging(MagicMock(), True)
- logging.info("test", extra={"testattr": "foo"})
- self.assertTrue(handler.received)
|