Skip to content

Commit 8df70f3

Browse files
committed
Update tests
1 parent 196d821 commit 8df70f3

File tree

1 file changed

+18
-1
lines changed

1 file changed

+18
-1
lines changed

backend/api/tests/test_utils.py

Lines changed: 18 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,5 @@
1-
from api.utils import get_ip
1+
import pytest
2+
from api.utils import get_ip, validate_url
23

34

45
def test_get_ip(rf):
@@ -9,3 +10,19 @@ def test_get_ip(rf):
910
def test_get_ip_with_remote_addr(rf):
1011
request = rf.get("/", REMOTE_ADDR="2.2.2.2")
1112
assert get_ip(request) == "2.2.2.2"
13+
14+
15+
@pytest.mark.parametrize(
16+
"url, expected",
17+
[
18+
("https://www.google.com", True),
19+
("http://www.google.com", True),
20+
("https://www.google.com/search?q=test", True),
21+
("https://www.google.com/search?q=test", True),
22+
("https://www.google.com/search?q=test", True),
23+
("http://", False),
24+
("https://", False),
25+
],
26+
)
27+
def test_validate_url(url, expected):
28+
assert validate_url(url) == expected

0 commit comments

Comments
 (0)