# Copyright 2014: Mirantis Inc.
# All Rights Reserved.
#
#    Licensed 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 mock

from rally import log
from tests.unit import test


class LogTestCase(test.TestCase):

    @mock.patch("rally.log.CONF")
    @mock.patch("rally.log.oslogging")
    def test_setup(self, mock_oslogger, mock_conf):

        proj = "fakep"
        version = "fakev"
        mock_oslogger.ColorHandler.LEVEL_COLORS = {
            logging.DEBUG: "debug_color"}
        mock_conf.rally_debug = True

        log.setup(proj, version)

        self.assertIn(logging.RDEBUG, mock_oslogger.ColorHandler.LEVEL_COLORS)
        self.assertEqual(
            mock_oslogger.ColorHandler.LEVEL_COLORS[logging.DEBUG],
            mock_oslogger.ColorHandler.LEVEL_COLORS[logging.RDEBUG])

        mock_oslogger.setup.assert_called_once_with(proj, version)
        mock_oslogger.getLogger(None).logger.setLevel.assert_called_once_with(
            logging.RDEBUG)

    @mock.patch("rally.log.logging")
    @mock.patch("rally.log.RallyContextAdapter")
    @mock.patch("rally.log.oslogging")
    def test_getLogger(self, mock_oslogger, mock_radapter, mock_pylogging):

        proj = "fake"
        vers = "fake"
        mock_oslogger._loggers = dict()

        returned_logger = log.getLogger(proj, vers)

        self.assertIn(proj, mock_oslogger._loggers)
        mock_radapter.assert_called_once_with(mock_pylogging.getLogger(proj),
                                              proj, vers)
        self.assertEqual(mock_oslogger._loggers[proj], returned_logger)


class LogRallyContaxtAdapter(test.TestCase):

    @mock.patch("rally.log.logging")
    @mock.patch("rally.log.oslogging.ContextAdapter")
    def test_debug(self, mock_oslo_adapter, mock_logging):

        mock_logging.RDEBUG = 123
        fake_msg = "fake message"
        radapter = log.RallyContextAdapter(mock.MagicMock(), "fakep", "fakev")
        radapter.log = mock.MagicMock()

        radapter.debug(fake_msg)

        radapter.log.assert_called_once_with(mock_logging.RDEBUG,
                                             fake_msg)