Skip to content

Commit 34ecb8c

Browse files
committed
Fix unit test for mypy
1 parent 439f2b2 commit 34ecb8c

File tree

1 file changed

+9
-8
lines changed

1 file changed

+9
-8
lines changed

tests/unit/test_commands.py

Lines changed: 9 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -1,7 +1,8 @@
1-
import os
21
import logging
2+
import os
33
from typing import Callable
44
from unittest import mock
5+
56
import pytest
67

78
from pip._internal.cli.base_command import Command
@@ -171,13 +172,13 @@ def test_handle_pip_version_check_warning_includes_reason(
171172
options.disable_pip_version_check = False
172173
options.no_index = False
173174

174-
dummy_session = mock.MagicMock()
175-
dummy_session.__enter__.return_value = dummy_session
176-
dummy_session.__exit__.return_value = None
177-
cmd._build_session = mock.Mock(return_value=dummy_session)
178-
179-
with caplog.at_level(logging.WARNING):
180-
cmd.handle_pip_version_check(options)
175+
with mock.patch.object(
176+
cmd,
177+
"_build_session",
178+
side_effect=PermissionError("Access is denied"),
179+
):
180+
with caplog.at_level(logging.WARNING):
181+
cmd.handle_pip_version_check(options)
181182

182183
assert "There was an error checking the latest version of pip." in caplog.text
183184
assert "PermissionError" in caplog.text

0 commit comments

Comments
 (0)