Python logger patch

logger_patch.py

from logging import Logger
import unittest

from mock import Mock, patch


class MyClassTest(unittest.TestCase):
    def setUp(self):
        self._logger = Mock(spec=Logger)
        self.logging_patch = patch(
            "my.module.logging", spec=True
        )
        mock_logging = self.logging_patch.start()
        mock_logging.getLogger.side_effect = [Mock(spec=Logger), self._logger]

    def tearDown(self):
        self.logging_patch.stop()