diff options
Diffstat (limited to 'tests/test_repo_logging.py')
| -rw-r--r-- | tests/test_repo_logging.py | 8 |
1 files changed, 4 insertions, 4 deletions
diff --git a/tests/test_repo_logging.py b/tests/test_repo_logging.py index ba8a9a9e..5402a90b 100644 --- a/tests/test_repo_logging.py +++ b/tests/test_repo_logging.py | |||
| @@ -14,7 +14,7 @@ | |||
| 14 | 14 | ||
| 15 | """Unit test for repo_logging module.""" | 15 | """Unit test for repo_logging module.""" |
| 16 | import unittest | 16 | import unittest |
| 17 | from unittest.mock import MagicMock | 17 | from unittest import mock |
| 18 | 18 | ||
| 19 | from repo_logging import RepoLogger | 19 | from repo_logging import RepoLogger |
| 20 | 20 | ||
| @@ -30,7 +30,7 @@ class TestRepoLogger(unittest.TestCase): | |||
| 30 | nonlocal result | 30 | nonlocal result |
| 31 | result = log.getMessage() | 31 | result = log.getMessage() |
| 32 | 32 | ||
| 33 | mock_out = MagicMock() | 33 | mock_out = mock.MagicMock() |
| 34 | mock_out.level = 0 | 34 | mock_out.level = 0 |
| 35 | mock_out.handle = mock_handler | 35 | mock_out.handle = mock_handler |
| 36 | logger.addHandler(mock_out) | 36 | logger.addHandler(mock_out) |
| @@ -49,7 +49,7 @@ class TestRepoLogger(unittest.TestCase): | |||
| 49 | nonlocal result | 49 | nonlocal result |
| 50 | result = log.getMessage() | 50 | result = log.getMessage() |
| 51 | 51 | ||
| 52 | mock_out = MagicMock() | 52 | mock_out = mock.MagicMock() |
| 53 | mock_out.level = 0 | 53 | mock_out.level = 0 |
| 54 | mock_out.handle = mock_handler | 54 | mock_out.handle = mock_handler |
| 55 | logger.addHandler(mock_out) | 55 | logger.addHandler(mock_out) |
| @@ -88,7 +88,7 @@ class TestRepoLogger(unittest.TestCase): | |||
| 88 | nonlocal result | 88 | nonlocal result |
| 89 | result.append(log.getMessage()) | 89 | result.append(log.getMessage()) |
| 90 | 90 | ||
| 91 | mock_out = MagicMock() | 91 | mock_out = mock.MagicMock() |
| 92 | mock_out.level = 0 | 92 | mock_out.level = 0 |
| 93 | mock_out.handle = mock_handler | 93 | mock_out.handle = mock_handler |
| 94 | logger.addHandler(mock_out) | 94 | logger.addHandler(mock_out) |
