|
37 | 37 | import aiohttp
|
38 | 38 | import pytest
|
39 | 39 | from _pytest.mark.structures import MarkDecorator
|
40 |
| -from mock import patch |
| 40 | +from mock import MagicMock, patch |
41 | 41 | from multidict import CIMultiDict
|
42 | 42 | from pytest import raises
|
43 | 43 |
|
@@ -254,26 +254,29 @@ async def test_warns_if_using_non_default_ssl_kwargs_with_ssl_context(self) -> N
|
254 | 254 | == str(w[0].message)
|
255 | 255 | )
|
256 | 256 |
|
257 |
| - @patch("ssl.SSLContext.load_verify_locations") |
258 |
| - async def test_uses_given_ca_certs( |
259 |
| - self, load_verify_locations: Any, tmp_path: Any |
260 |
| - ) -> None: |
| 257 | + @patch("ssl.SSLContext", return_value=MagicMock()) |
| 258 | + async def test_uses_given_ca_certs(self, ssl_context: Any, tmp_path: Any) -> None: |
261 | 259 | path = tmp_path / "ca_certs.pem"
|
262 | 260 | path.touch()
|
| 261 | + ssl_context.return_value.load_verify_locations.return_value = None |
263 | 262 | AIOHttpConnection(use_ssl=True, ca_certs=str(path))
|
264 |
| - load_verify_locations.assert_called_once_with(cafile=str(path)) |
| 263 | + ssl_context.return_value.load_verify_locations.assert_called_once_with( |
| 264 | + cafile=str(path) |
| 265 | + ) |
265 | 266 |
|
266 |
| - @patch("ssl.SSLContext.load_verify_locations") |
267 |
| - async def test_uses_default_ca_certs(self, load_verify_locations: Any) -> None: |
| 267 | + @patch("ssl.SSLContext", return_value=MagicMock()) |
| 268 | + async def test_uses_default_ca_certs(self, ssl_context: Any) -> None: |
| 269 | + ssl_context.return_value.load_verify_locations.return_value = None |
268 | 270 | AIOHttpConnection(use_ssl=True)
|
269 |
| - load_verify_locations.assert_called_once_with( |
| 271 | + ssl_context.return_value.load_verify_locations.assert_called_once_with( |
270 | 272 | cafile=Connection.default_ca_certs()
|
271 | 273 | )
|
272 | 274 |
|
273 |
| - @patch("ssl.SSLContext.load_verify_locations") |
274 |
| - async def test_uses_no_ca_certs(self, load_verify_locations: Any) -> None: |
| 275 | + @patch("ssl.SSLContext", return_value=MagicMock()) |
| 276 | + async def test_uses_no_ca_certs(self, ssl_context: Any) -> None: |
| 277 | + ssl_context.return_value.load_verify_locations.return_value = None |
275 | 278 | AIOHttpConnection(use_ssl=True, verify_certs=False)
|
276 |
| - load_verify_locations.assert_not_called() |
| 279 | + ssl_context.return_value.load_verify_locations.assert_not_called() |
277 | 280 |
|
278 | 281 | async def test_trust_env(self) -> None:
|
279 | 282 | con: Any = AIOHttpConnection(trust_env=True)
|
|
0 commit comments