mirror of
https://github.com/PrefectHQ/fastmcp.git
synced 2026-08-25 15:04:18 +02:00
82 lines
3.1 KiB
Python
82 lines
3.1 KiB
Python
import pytest
|
|
from pydantic import AnyHttpUrl, ValidationError
|
|
|
|
from fastmcp import FastMCP
|
|
from fastmcp.server.auth.providers.bearer import BearerAuthProvider
|
|
from fastmcp.server.auth.providers.bearer_env import EnvBearerAuthProvider
|
|
|
|
|
|
def test_load_bearer_env_from_env_var(monkeypatch):
|
|
mcp = FastMCP()
|
|
assert mcp.auth is None
|
|
|
|
monkeypatch.setenv("FASTMCP_SERVER_DEFAULT_AUTH_PROVIDER", "bearer_env")
|
|
monkeypatch.setenv("FASTMCP_AUTH_BEARER_PUBLIC_KEY", "test-public-key")
|
|
|
|
mcp_with_auth = FastMCP()
|
|
assert isinstance(mcp_with_auth.auth, EnvBearerAuthProvider)
|
|
|
|
|
|
def test_load_bearer_env_from_env_var_requires_public_key_or_jwks_uri(monkeypatch):
|
|
mcp = FastMCP()
|
|
assert mcp.auth is None
|
|
|
|
monkeypatch.setenv("FASTMCP_SERVER_DEFAULT_AUTH_PROVIDER", "bearer_env")
|
|
|
|
with pytest.raises(
|
|
ValueError, match="Either public_key or jwks_uri must be provided"
|
|
):
|
|
FastMCP()
|
|
|
|
|
|
def test_configure_bearer_env_from_env_var(monkeypatch):
|
|
monkeypatch.setenv("FASTMCP_SERVER_DEFAULT_AUTH_PROVIDER", "bearer_env")
|
|
monkeypatch.setenv("FASTMCP_AUTH_BEARER_PUBLIC_KEY", "test-public-key")
|
|
monkeypatch.setenv("FASTMCP_AUTH_BEARER_ISSUER", "http://test-issuer")
|
|
monkeypatch.setenv("FASTMCP_AUTH_BEARER_AUDIENCE", "test-audience")
|
|
monkeypatch.setenv(
|
|
"FASTMCP_AUTH_BEARER_REQUIRED_SCOPES", '["test-scope1", "test-scope2"]'
|
|
)
|
|
|
|
mcp = FastMCP()
|
|
assert isinstance(mcp.auth, EnvBearerAuthProvider)
|
|
assert mcp.auth.public_key == "test-public-key"
|
|
assert mcp.auth.issuer_url == AnyHttpUrl("http://test-issuer")
|
|
assert mcp.auth.audience == "test-audience"
|
|
assert mcp.auth.required_scopes == ["test-scope1", "test-scope2"]
|
|
|
|
|
|
def test_list_of_scopes_must_be_a_list(monkeypatch):
|
|
monkeypatch.setenv("FASTMCP_SERVER_DEFAULT_AUTH_PROVIDER", "bearer_env")
|
|
monkeypatch.setenv("FASTMCP_AUTH_BEARER_REQUIRED_SCOPES", "test-scope1")
|
|
|
|
with pytest.raises(ValidationError, match="Input should be a valid list"):
|
|
FastMCP()
|
|
|
|
|
|
def test_configure_bearer_env_jwks_uri_from_env_var(monkeypatch):
|
|
monkeypatch.setenv("FASTMCP_SERVER_DEFAULT_AUTH_PROVIDER", "bearer_env")
|
|
monkeypatch.setenv("FASTMCP_AUTH_BEARER_JWKS_URI", "test-jwks-uri")
|
|
|
|
mcp = FastMCP()
|
|
assert isinstance(mcp.auth, EnvBearerAuthProvider)
|
|
assert mcp.auth.jwks_uri == "test-jwks-uri"
|
|
|
|
|
|
def test_configure_bearer_env_public_key_and_jwks_uri_error(monkeypatch):
|
|
monkeypatch.setenv("FASTMCP_SERVER_DEFAULT_AUTH_PROVIDER", "bearer_env")
|
|
monkeypatch.setenv("FASTMCP_AUTH_BEARER_PUBLIC_KEY", "test-public-key")
|
|
monkeypatch.setenv("FASTMCP_AUTH_BEARER_JWKS_URI", "test-jwks-uri")
|
|
|
|
with pytest.raises(ValueError, match="Provide either public_key or jwks_uri"):
|
|
FastMCP()
|
|
|
|
|
|
def test_provided_auth_takes_precedence_over_env_vars(monkeypatch):
|
|
monkeypatch.setenv("FASTMCP_SERVER_DEFAULT_AUTH_PROVIDER", "bearer_env")
|
|
monkeypatch.setenv("FASTMCP_AUTH_BEARER_PUBLIC_KEY", "test-public-key")
|
|
|
|
mcp = FastMCP(auth=BearerAuthProvider(public_key="test-public-key-2"))
|
|
assert isinstance(mcp.auth, BearerAuthProvider)
|
|
assert not isinstance(mcp.auth, EnvBearerAuthProvider)
|
|
assert mcp.auth.public_key == "test-public-key-2"
|