Spaces:
Running
Running
Jeremiah Lowin commited on
Commit ·
1582f65
1
Parent(s): 36b7a70
Use update env var pattern
Browse files
tests/auth/providers/test_bearer_env.py
CHANGED
|
@@ -12,7 +12,7 @@ def test_load_bearer_env_from_env_var(monkeypatch):
|
|
| 12 |
mcp = FastMCP()
|
| 13 |
assert mcp.auth is None
|
| 14 |
|
| 15 |
-
monkeypatch.setenv("
|
| 16 |
monkeypatch.setenv("FASTMCP_AUTH_BEARER_PUBLIC_KEY", "test-public-key")
|
| 17 |
|
| 18 |
with temporary_settings(**Settings().model_dump()):
|
|
@@ -24,7 +24,7 @@ def test_load_bearer_env_from_env_var_requires_public_key_or_jwks_uri(monkeypatc
|
|
| 24 |
mcp = FastMCP()
|
| 25 |
assert mcp.auth is None
|
| 26 |
|
| 27 |
-
monkeypatch.setenv("
|
| 28 |
|
| 29 |
with temporary_settings(**Settings().model_dump()):
|
| 30 |
with pytest.raises(
|
|
@@ -34,7 +34,7 @@ def test_load_bearer_env_from_env_var_requires_public_key_or_jwks_uri(monkeypatc
|
|
| 34 |
|
| 35 |
|
| 36 |
def test_configure_bearer_env_from_env_var(monkeypatch):
|
| 37 |
-
monkeypatch.setenv("
|
| 38 |
monkeypatch.setenv("FASTMCP_AUTH_BEARER_PUBLIC_KEY", "test-public-key")
|
| 39 |
monkeypatch.setenv("FASTMCP_AUTH_BEARER_ISSUER", "http://test-issuer")
|
| 40 |
monkeypatch.setenv("FASTMCP_AUTH_BEARER_AUDIENCE", "test-audience")
|
|
@@ -52,7 +52,7 @@ def test_configure_bearer_env_from_env_var(monkeypatch):
|
|
| 52 |
|
| 53 |
|
| 54 |
def test_list_of_scopes_must_be_a_list(monkeypatch):
|
| 55 |
-
monkeypatch.setenv("
|
| 56 |
monkeypatch.setenv("FASTMCP_AUTH_BEARER_REQUIRED_SCOPES", "test-scope1")
|
| 57 |
|
| 58 |
with temporary_settings(**Settings().model_dump()):
|
|
@@ -61,7 +61,7 @@ def test_list_of_scopes_must_be_a_list(monkeypatch):
|
|
| 61 |
|
| 62 |
|
| 63 |
def test_configure_bearer_env_jwks_uri_from_env_var(monkeypatch):
|
| 64 |
-
monkeypatch.setenv("
|
| 65 |
monkeypatch.setenv("FASTMCP_AUTH_BEARER_JWKS_URI", "test-jwks-uri")
|
| 66 |
|
| 67 |
with temporary_settings(**Settings().model_dump()):
|
|
@@ -71,7 +71,7 @@ def test_configure_bearer_env_jwks_uri_from_env_var(monkeypatch):
|
|
| 71 |
|
| 72 |
|
| 73 |
def test_configure_bearer_env_public_key_and_jwks_uri_error(monkeypatch):
|
| 74 |
-
monkeypatch.setenv("
|
| 75 |
monkeypatch.setenv("FASTMCP_AUTH_BEARER_PUBLIC_KEY", "test-public-key")
|
| 76 |
monkeypatch.setenv("FASTMCP_AUTH_BEARER_JWKS_URI", "test-jwks-uri")
|
| 77 |
|
|
@@ -81,7 +81,7 @@ def test_configure_bearer_env_public_key_and_jwks_uri_error(monkeypatch):
|
|
| 81 |
|
| 82 |
|
| 83 |
def test_provided_auth_takes_precedence_over_env_vars(monkeypatch):
|
| 84 |
-
monkeypatch.setenv("
|
| 85 |
monkeypatch.setenv("FASTMCP_AUTH_BEARER_PUBLIC_KEY", "test-public-key")
|
| 86 |
|
| 87 |
with temporary_settings(**Settings().model_dump()):
|
|
|
|
| 12 |
mcp = FastMCP()
|
| 13 |
assert mcp.auth is None
|
| 14 |
|
| 15 |
+
monkeypatch.setenv("FASTMCP_DEFAULT_AUTH_PROVIDER", "bearer_env")
|
| 16 |
monkeypatch.setenv("FASTMCP_AUTH_BEARER_PUBLIC_KEY", "test-public-key")
|
| 17 |
|
| 18 |
with temporary_settings(**Settings().model_dump()):
|
|
|
|
| 24 |
mcp = FastMCP()
|
| 25 |
assert mcp.auth is None
|
| 26 |
|
| 27 |
+
monkeypatch.setenv("FASTMCP_DEFAULT_AUTH_PROVIDER", "bearer_env")
|
| 28 |
|
| 29 |
with temporary_settings(**Settings().model_dump()):
|
| 30 |
with pytest.raises(
|
|
|
|
| 34 |
|
| 35 |
|
| 36 |
def test_configure_bearer_env_from_env_var(monkeypatch):
|
| 37 |
+
monkeypatch.setenv("FASTMCP_DEFAULT_AUTH_PROVIDER", "bearer_env")
|
| 38 |
monkeypatch.setenv("FASTMCP_AUTH_BEARER_PUBLIC_KEY", "test-public-key")
|
| 39 |
monkeypatch.setenv("FASTMCP_AUTH_BEARER_ISSUER", "http://test-issuer")
|
| 40 |
monkeypatch.setenv("FASTMCP_AUTH_BEARER_AUDIENCE", "test-audience")
|
|
|
|
| 52 |
|
| 53 |
|
| 54 |
def test_list_of_scopes_must_be_a_list(monkeypatch):
|
| 55 |
+
monkeypatch.setenv("FASTMCP_DEFAULT_AUTH_PROVIDER", "bearer_env")
|
| 56 |
monkeypatch.setenv("FASTMCP_AUTH_BEARER_REQUIRED_SCOPES", "test-scope1")
|
| 57 |
|
| 58 |
with temporary_settings(**Settings().model_dump()):
|
|
|
|
| 61 |
|
| 62 |
|
| 63 |
def test_configure_bearer_env_jwks_uri_from_env_var(monkeypatch):
|
| 64 |
+
monkeypatch.setenv("FASTMCP_DEFAULT_AUTH_PROVIDER", "bearer_env")
|
| 65 |
monkeypatch.setenv("FASTMCP_AUTH_BEARER_JWKS_URI", "test-jwks-uri")
|
| 66 |
|
| 67 |
with temporary_settings(**Settings().model_dump()):
|
|
|
|
| 71 |
|
| 72 |
|
| 73 |
def test_configure_bearer_env_public_key_and_jwks_uri_error(monkeypatch):
|
| 74 |
+
monkeypatch.setenv("FASTMCP_DEFAULT_AUTH_PROVIDER", "bearer_env")
|
| 75 |
monkeypatch.setenv("FASTMCP_AUTH_BEARER_PUBLIC_KEY", "test-public-key")
|
| 76 |
monkeypatch.setenv("FASTMCP_AUTH_BEARER_JWKS_URI", "test-jwks-uri")
|
| 77 |
|
|
|
|
| 81 |
|
| 82 |
|
| 83 |
def test_provided_auth_takes_precedence_over_env_vars(monkeypatch):
|
| 84 |
+
monkeypatch.setenv("FASTMCP_DEFAULT_AUTH_PROVIDER", "bearer_env")
|
| 85 |
monkeypatch.setenv("FASTMCP_AUTH_BEARER_PUBLIC_KEY", "test-public-key")
|
| 86 |
|
| 87 |
with temporary_settings(**Settings().model_dump()):
|