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) |