def test_get_connection_manager_with_proxy(mock_proxy_settings):
remote_connection = RemoteConnection('http://remote', keep_alive=False)
conn = remote_connection._get_connection_manager()
assert type(conn) == urllib3.ProxyManager
assert conn.proxy.scheme == 'http'
assert conn.proxy.host == 'http_proxy.com'
assert conn.proxy.port == 8080
remote_connection_https = RemoteConnection('https://remote', keep_alive=False)
conn = remote_connection_https._get_connection_manager()
assert type(conn) == urllib3.ProxyManager
assert conn.proxy.scheme == 'http'
assert conn.proxy.host == 'https_proxy.com'
assert conn.proxy.port == 8080
def test_get_connection_manager_for_certs_and_timeout():
remote_connection = RemoteConnection('http://remote', keep_alive=False)
remote_connection.set_timeout(10)
conn = remote_connection._get_connection_manager()
assert conn.connection_pool_kw['timeout'] == 10
assert conn.connection_pool_kw['cert_reqs'] == 'CERT_REQUIRED'
assert 'certifi/cacert.pem' in conn.connection_pool_kw['ca_certs']
def test_get_connection_manager_for_certs_and_timeout(monkeypatch):
monkeypatch.setattr(RemoteConnection, "get_timeout", lambda _: 10) # Class state; leaks into subsequent tests.
remote_connection = RemoteConnection('http://remote', keep_alive=False)
conn = remote_connection._get_connection_manager()
assert conn.connection_pool_kw['timeout'] == 10
assert conn.connection_pool_kw['cert_reqs'] == 'CERT_REQUIRED'
assert 'certifi/cacert.pem' in conn.connection_pool_kw['ca_certs']
def test_get_connection_manager_with_auth_proxy(mock_proxy_auth_settings):
proxy_auth_header = urllib3.make_headers(proxy_basic_auth="user:password")
remote_connection = RemoteConnection('http://remote', keep_alive=False)
conn = remote_connection._get_connection_manager()
assert type(conn) == urllib3.ProxyManager
assert conn.proxy.scheme == 'http'
assert conn.proxy.host == 'http_proxy.com'
assert conn.proxy.port == 8080
assert conn.proxy_headers == proxy_auth_header
remote_connection_https = RemoteConnection('https://remote',
keep_alive=False)
conn = remote_connection_https._get_connection_manager()
assert type(conn) == urllib3.ProxyManager
assert conn.proxy.scheme == 'https'
assert conn.proxy.host == 'https_proxy.com'
assert conn.proxy.port == 8080
assert conn.proxy_headers == proxy_auth_header
def test_get_connection_manager_without_proxy(mock_proxy_settings_missing):
remote_connection = RemoteConnection('http://remote', keep_alive=False)
conn = remote_connection._get_connection_manager()
assert type(conn) == urllib3.PoolManager
def test_ignore_proxy_env_vars(mock_proxy_settings):
remote_connection = RemoteConnection("http://remote", ignore_proxy=True)
conn = remote_connection._get_connection_manager()
assert type(conn) == urllib3.PoolManager
def test_get_connection_manager_when_no_proxy_set(mock_no_proxy_settings, url):
remote_connection = RemoteConnection(url)
conn = remote_connection._get_connection_manager()
assert type(conn) == urllib3.PoolManager
def test_get_socks_proxy_when_set(mock_socks_proxy_settings):
remote_connection = RemoteConnection("http://127.0.0.1:4444/wd/hub")
conn = remote_connection._get_connection_manager()
from urllib3.contrib.socks import SOCKSProxyManager
assert type(conn) == SOCKSProxyManager
def test_default_socket_timeout_is_correct():
remote_connection = RemoteConnection("http://remote", keep_alive=True)
conn = remote_connection._get_connection_manager()
assert conn.connection_pool_kw['timeout'] is None
