Skip to content

Commit 1d31481

Browse files
committed
Create helper ssrf_check issue
1 parent 3dc4610 commit 1d31481

File tree

1 file changed

+8
-0
lines changed

1 file changed

+8
-0
lines changed

aikido_zen/sinks/tests/requests_test.py

Lines changed: 8 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -58,6 +58,14 @@ def set_context_and_lifecycle(url):
5858
context.set_as_current_context()
5959

6060

61+
def ssrf_check(monkeypatch, url):
62+
reset_comms()
63+
set_context_and_lifecycle(url)
64+
monkeypatch.setenv("AIKIDO_BLOCK", "1")
65+
with pytest.raises(AikidoSSRF):
66+
requests.get(url)
67+
68+
6169
def test_srrf_test(monkeypatch):
6270
reset_comms()
6371
set_context_and_lifecycle(SSRF_TEST)

0 commit comments

Comments
 (0)