21 lines
457 B
Python
21 lines
457 B
Python
# retoor <retoor@molodetz.nl>
|
|
import pytest
|
|
|
|
from molodetz.utils.request_meta import safe_next
|
|
|
|
|
|
@pytest.mark.parametrize(
|
|
"value,expected",
|
|
[
|
|
("/admin", "/admin"),
|
|
("/roll?page=2", "/roll?page=2"),
|
|
("https://evil.example/", "/"),
|
|
("//evil.example/", "/"),
|
|
("/\\evil.example", "/"),
|
|
("", "/"),
|
|
(None, "/"),
|
|
],
|
|
)
|
|
def test_safe_next(value, expected):
|
|
assert safe_next(value) == expected
|