Max Isbey
48 lines
1.7 KiB
Python
48 lines
1.7 KiB
Python
import pytest
|
|
from pydantic import AnyHttpUrl
|
|
|
|
from mcp.server.auth.routes import validate_issuer_url
|
|
|
|
|
|
def test_validate_issuer_url_https_allowed():
|
|
validate_issuer_url(AnyHttpUrl("https://example.com/path"))
|
|
|
|
|
|
def test_validate_issuer_url_http_localhost_allowed():
|
|
validate_issuer_url(AnyHttpUrl("http://localhost:8080/path"))
|
|
|
|
|
|
def test_validate_issuer_url_http_127_0_0_1_allowed():
|
|
validate_issuer_url(AnyHttpUrl("http://127.0.0.1:8080/path"))
|
|
|
|
|
|
def test_validate_issuer_url_http_ipv6_loopback_allowed():
|
|
validate_issuer_url(AnyHttpUrl("http://[::1]:8080/path"))
|
|
|
|
|
|
def test_validate_issuer_url_http_non_loopback_rejected():
|
|
with pytest.raises(ValueError, match="Issuer URL must be HTTPS"):
|
|
validate_issuer_url(AnyHttpUrl("http://evil.com/path"))
|
|
|
|
|
|
def test_validate_issuer_url_http_127_prefix_domain_rejected():
|
|
"""A domain like 127.0.0.1.evil.com is not loopback."""
|
|
with pytest.raises(ValueError, match="Issuer URL must be HTTPS"):
|
|
validate_issuer_url(AnyHttpUrl("http://127.0.0.1.evil.com/path"))
|
|
|
|
|
|
def test_validate_issuer_url_http_127_prefix_subdomain_rejected():
|
|
"""A domain like 127.0.0.1something.example.com is not loopback."""
|
|
with pytest.raises(ValueError, match="Issuer URL must be HTTPS"):
|
|
validate_issuer_url(AnyHttpUrl("http://127.0.0.1something.example.com/path"))
|
|
|
|
|
|
def test_validate_issuer_url_fragment_rejected():
|
|
with pytest.raises(ValueError, match="fragment"):
|
|
validate_issuer_url(AnyHttpUrl("https://example.com/path#frag"))
|
|
|
|
|
|
def test_validate_issuer_url_query_rejected():
|
|
with pytest.raises(ValueError, match="query"):
|
|
validate_issuer_url(AnyHttpUrl("https://example.com/path?q=1"))
|