pylog/tests/test_monkey.py

   1 # Copyright (c) 2020 Deutsche Telekom.
   2 # Licensed under the Apache License, Version 2.0 (the "License");
   3 # you may not use this file except in compliance with the License.
   4 # You may obtain a copy of the License at:
   5 #       http://www.apache.org/licenses/LICENSE-2.0
   6 #
   7 # Unless required by applicable law or agreed to in writing, software
   8 # distributed under the License is distributed on an "AS IS" BASIS,
   9 # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  10
  11 import sys
  12 import unittest
  13
  14 if sys.version_info[0] < 3:
  15     from mock import patch
  16 if sys.version_info[0] >= 3:
  17     from unittest.mock import patch
  18
  19 from onaplogging.monkey import patch_all, patch_logging_yaml, patch_logging_mdc
  20
  21
  22 class TestMonkey(unittest.TestCase):
  23
  24     def test_patch_all(self):
  25         with patch("onaplogging.monkey.patch_logging_mdc") as mock_mdc:
  26             with patch("onaplogging.monkey.patch_logging_yaml") as mock_yaml:
  27                 patch_all()
  28                 mock_mdc.assert_called_once()
  29                 mock_yaml.assert_called_once()
  30
  31         with patch("onaplogging.monkey.patch_logging_mdc") as mock_mdc:
  32             with patch("onaplogging.monkey.patch_logging_yaml") as mock_yaml:
  33                 patch_all(mdc=False)
  34                 mock_mdc.assert_not_called()
  35                 mock_yaml.assert_called_once()
  36
  37         with patch("onaplogging.monkey.patch_logging_mdc") as mock_mdc:
  38             with patch("onaplogging.monkey.patch_logging_yaml") as mock_yaml:
  39                 patch_all(yaml=False)
  40                 mock_mdc.assert_called_once()
  41                 mock_yaml.assert_not_called()
  42
  43         with patch("onaplogging.monkey.patch_logging_mdc") as mock_mdc:
  44             with patch("onaplogging.monkey.patch_logging_yaml") as mock_yaml:
  45                 patch_all(mdc=False, yaml=False)
  46                 mock_mdc.assert_not_called()
  47                 mock_yaml.assert_not_called()