Fix test to handle urllib3 1.x and 2.x.

carsonyl · May 8, 2023 · afb2cc9 · afb2cc9
1 parent 7f93bbf
commit afb2cc9
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/tests/test_api.py b/tests/test_api.py
@@ -1,4 +1,6 @@
 import os
+import sys
+
 import pytest
 import requests
 from requests.auth import HTTPProxyAuth
@@ -122,7 +124,7 @@ class TestRequests(object):
 
  def test_proxy_connect_timeout(self):
  session = requests.Session()
- with pytest.raises(ConnectTimeout):
+ with pytest.raises((ProxyError, ConnectTimeout)): # ConnectTimeout on urllib3 < 2
  session.get(arbitrary_url, timeout=1, proxies=proxy_parameter_for_requests("http://httpbin.org:99"))
 
  def test_proxy_response_timeout(self):