def test_check_url_length():
# Max length of the stripped URL is 1024
length = 1024 - len("gemini://localhost")
s = length * "0"
assert check_url(f"gemini://localhost{s}\r\n", PORT)
s = (length + 1) * "0"
with pytest.raises(BadRequestException):
check_url(f"gemini://localhost{s}\r\n", PORT)
def test_check_url_no_crlf():
with pytest.raises(TimeoutException):
check_url("gemini://localhost\n", PORT)
with pytest.raises(TimeoutException):
check_url("gemini://localhost\r", PORT)
with pytest.raises(TimeoutException):
check_url("gemini://localhost", PORT)
def test_check_input_response():
assert check_url("gemini://localhost?\r\n", PORT)
assert check_url("gemini://localhost?hello\r\n", PORT)
assert check_url("gemini://localhost?hello+world\r\n", PORT)
assert check_url("gemini://localhost?hello%20world\r\n", PORT)
def test_check_url_bad_port():
with pytest.raises(ProxyRequestRefusedException):
check_url("gemini://localhost:1968\r\n", PORT)
def test_check_url_no_gemini():
with pytest.raises(BadRequestException):
check_url("localhost\r\n", PORT)
with pytest.raises(ProxyRequestRefusedException):
check_url("https://localhost\r\n", PORT)
def test_check_url_root():
assert check_url("gemini://localhost\r\n", PORT)
assert check_url("gemini://localhost/\r\n", PORT)
assert check_url("gemini://localhost:1965\r\n", PORT)
assert check_url("gemini://localhost:1965/\r\n", PORT)