Skip to content

Commit 6550242

Browse files
committed
update test case: lint & use correct ConnectionError type
1 parent 145586f commit 6550242

1 file changed

Lines changed: 2 additions & 1 deletion

File tree

aikido_zen/sinks/tests/requests_and_urllib3_test.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -10,6 +10,7 @@
1010
import requests
1111
import urllib3
1212
from requests import ConnectTimeout
13+
from requests.exceptions import ConnectionError
1314

1415

1516
@pytest.fixture(autouse=True)
@@ -202,5 +203,5 @@ def test_zero_padded_ip(monkeypatch):
202203
set_context_and_lifecycle(url)
203204
# Can raise both errors : either connection times out -> 0127.0.0.01 not supported by platform
204205
# or it raises ssrf bug -> 0127.0.0.01 supported by platform
205-
with pytest.raises((AikidoSSRF, ConnectTimeout, ConnectionError)):
206+
with pytest.raises((AikidoSSRF, ConnectTimeout, ConnectionError)):
206207
requests.get(url)

0 commit comments

Comments
 (0)