diff options
| author | Asif Saif Uddin <auvipy@gmail.com> | 2023-04-08 22:45:08 +0600 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2023-04-08 22:45:08 +0600 |
| commit | 973dc3790ac25b9da7b6d2641ac72d95470f6ed8 (patch) | |
| tree | 9e7ba02d8520994a06efc37dde05fba722138189 /t/unit/asynchronous/http | |
| parent | 7ceb675bb69917fae182ebdaf9a2298a308c3fa4 (diff) | |
| parent | 2de7f9f038dd62e097e490cb3fa609067c1c3c36 (diff) | |
| download | kombu-py310.tar.gz | |
Merge branch 'main' into py310py310
Diffstat (limited to 't/unit/asynchronous/http')
| -rw-r--r-- | t/unit/asynchronous/http/test_curl.py | 26 | ||||
| -rw-r--r-- | t/unit/asynchronous/http/test_http.py | 2 |
2 files changed, 27 insertions, 1 deletions
diff --git a/t/unit/asynchronous/http/test_curl.py b/t/unit/asynchronous/http/test_curl.py index db8f5f91..51f9128e 100644 --- a/t/unit/asynchronous/http/test_curl.py +++ b/t/unit/asynchronous/http/test_curl.py @@ -1,4 +1,7 @@ -from unittest.mock import Mock, call, patch +from __future__ import annotations + +from io import BytesIO +from unittest.mock import ANY, Mock, call, patch import pytest @@ -131,3 +134,24 @@ class test_CurlClient: x._on_event.assert_called_with(fd, _pycurl.CSELECT_IN) x.on_writable(fd, _pycurl=_pycurl) x._on_event.assert_called_with(fd, _pycurl.CSELECT_OUT) + + def test_setup_request_sets_proxy_when_specified(self): + with patch('kombu.asynchronous.http.curl.pycurl') as _pycurl: + x = self.Client() + proxy_host = 'http://www.example.com' + request = Mock( + name='request', headers={}, auth_mode=None, proxy_host=None + ) + proxied_request = Mock( + name='request', headers={}, auth_mode=None, + proxy_host=proxy_host, proxy_port=123 + ) + x._setup_request( + x.Curl, request, BytesIO(), x.Headers(), _pycurl=_pycurl + ) + with pytest.raises(AssertionError): + x.Curl.setopt.assert_any_call(_pycurl.PROXY, ANY) + x._setup_request( + x.Curl, proxied_request, BytesIO(), x.Headers(), _pycurl + ) + x.Curl.setopt.assert_any_call(_pycurl.PROXY, proxy_host) diff --git a/t/unit/asynchronous/http/test_http.py b/t/unit/asynchronous/http/test_http.py index 6e6abdcb..816bf89d 100644 --- a/t/unit/asynchronous/http/test_http.py +++ b/t/unit/asynchronous/http/test_http.py @@ -1,3 +1,5 @@ +from __future__ import annotations + from io import BytesIO from unittest.mock import Mock |
