1010from fastmcp_extensions import (
1111 MCPServerConfig ,
1212 MCPServerConfigArg ,
13+ get_mcp_config ,
1314 mcp_server ,
14- resolve_config ,
1515)
1616
1717
@@ -112,7 +112,7 @@ def test_mcp_server_config_arg_attributes(
112112
113113
114114@pytest .mark .unit
115- def test_resolve_config_from_env_var () -> None :
115+ def test_get_mcp_config_from_env_var () -> None :
116116 """Test resolving config from environment variable."""
117117 config_args = [
118118 MCPServerConfigArg (
@@ -125,12 +125,12 @@ def test_resolve_config_from_env_var() -> None:
125125 app = mcp_server ("test-server" , server_config_args = config_args )
126126
127127 with patch .dict (os .environ , {"TEST_API_KEY" : "secret-key-123" }):
128- value = resolve_config (app , "api_key" )
128+ value = get_mcp_config (app , "api_key" )
129129 assert value == "secret-key-123"
130130
131131
132132@pytest .mark .unit
133- def test_resolve_config_from_http_header () -> None :
133+ def test_get_mcp_config_from_http_header () -> None :
134134 """Test resolving config from HTTP header (takes precedence over env var)."""
135135 config_args = [
136136 MCPServerConfigArg (
@@ -146,12 +146,12 @@ def test_resolve_config_from_http_header() -> None:
146146 "fastmcp_extensions.server.get_http_headers" ,
147147 return_value = {"X-API-Key" : "header-key" },
148148 ):
149- value = resolve_config (app , "api_key" )
149+ value = get_mcp_config (app , "api_key" )
150150 assert value == "header-key"
151151
152152
153153@pytest .mark .unit
154- def test_resolve_config_header_case_insensitive () -> None :
154+ def test_get_mcp_config_header_case_insensitive () -> None :
155155 """Test that HTTP header resolution is case-insensitive."""
156156 config_args = [
157157 MCPServerConfigArg (
@@ -167,21 +167,21 @@ def test_resolve_config_header_case_insensitive() -> None:
167167 "fastmcp_extensions.server.get_http_headers" ,
168168 return_value = {"x-api-key" : "lowercase-header-key" },
169169 ):
170- value = resolve_config (app , "api_key" )
170+ value = get_mcp_config (app , "api_key" )
171171 assert value == "lowercase-header-key"
172172
173173
174174@pytest .mark .unit
175- def test_resolve_config_unknown_name_raises_key_error () -> None :
175+ def test_get_mcp_config_unknown_name_raises_key_error () -> None :
176176 """Test that resolving unknown config name raises KeyError."""
177177 app = mcp_server ("test-server" )
178178
179179 with pytest .raises (KeyError , match = "Unknown config argument" ):
180- resolve_config (app , "nonexistent" )
180+ get_mcp_config (app , "nonexistent" )
181181
182182
183183@pytest .mark .unit
184- def test_resolve_config_required_missing_raises_value_error () -> None :
184+ def test_get_mcp_config_required_missing_raises_value_error () -> None :
185185 """Test that missing required config raises ValueError."""
186186 config_args = [
187187 MCPServerConfigArg (
@@ -197,11 +197,11 @@ def test_resolve_config_required_missing_raises_value_error() -> None:
197197 "fastmcp_extensions.server.get_http_headers" , return_value = None
198198 )
199199 with mock_headers , pytest .raises (ValueError , match = "Required config" ):
200- resolve_config (app , "api_key" )
200+ get_mcp_config (app , "api_key" )
201201
202202
203203@pytest .mark .unit
204- def test_resolve_config_optional_missing_returns_empty_string () -> None :
204+ def test_get_mcp_config_optional_missing_returns_empty_string () -> None :
205205 """Test that missing optional config returns empty string."""
206206 config_args = [
207207 MCPServerConfigArg (
@@ -214,7 +214,7 @@ def test_resolve_config_optional_missing_returns_empty_string() -> None:
214214 app = mcp_server ("test-server" , server_config_args = config_args )
215215
216216 with patch ("fastmcp_extensions.server.get_http_headers" , return_value = None ):
217- value = resolve_config (app , "optional_key" )
217+ value = get_mcp_config (app , "optional_key" )
218218 assert value == ""
219219
220220
@@ -237,7 +237,7 @@ def test_mcp_server_config_default_values() -> None:
237237
238238
239239@pytest .mark .unit
240- def test_resolve_config_with_string_default () -> None :
240+ def test_get_mcp_config_with_string_default () -> None :
241241 """Test resolving config with a string default value."""
242242 config_args = [
243243 MCPServerConfigArg (
@@ -250,12 +250,12 @@ def test_resolve_config_with_string_default() -> None:
250250 app = mcp_server ("test-server" , server_config_args = config_args )
251251
252252 with patch ("fastmcp_extensions.server.get_http_headers" , return_value = None ):
253- value = resolve_config (app , "api_key" )
253+ value = get_mcp_config (app , "api_key" )
254254 assert value == "default-value"
255255
256256
257257@pytest .mark .unit
258- def test_resolve_config_with_callable_default () -> None :
258+ def test_get_mcp_config_with_callable_default () -> None :
259259 """Test resolving config with a callable default value."""
260260 config_args = [
261261 MCPServerConfigArg (
@@ -268,12 +268,12 @@ def test_resolve_config_with_callable_default() -> None:
268268 app = mcp_server ("test-server" , server_config_args = config_args )
269269
270270 with patch ("fastmcp_extensions.server.get_http_headers" , return_value = None ):
271- value = resolve_config (app , "api_key" )
271+ value = get_mcp_config (app , "api_key" )
272272 assert value == "callable-default"
273273
274274
275275@pytest .mark .unit
276- def test_resolve_config_env_var_takes_precedence_over_default () -> None :
276+ def test_get_mcp_config_env_var_takes_precedence_over_default () -> None :
277277 """Test that env var takes precedence over default value."""
278278 config_args = [
279279 MCPServerConfigArg (
@@ -286,12 +286,12 @@ def test_resolve_config_env_var_takes_precedence_over_default() -> None:
286286 app = mcp_server ("test-server" , server_config_args = config_args )
287287
288288 with patch .dict (os .environ , {"TEST_API_KEY" : "env-value" }):
289- value = resolve_config (app , "api_key" )
289+ value = get_mcp_config (app , "api_key" )
290290 assert value == "env-value"
291291
292292
293293@pytest .mark .unit
294- def test_resolve_config_with_only_env_var () -> None :
294+ def test_get_mcp_config_with_only_env_var () -> None :
295295 """Test resolving config with only env_var set (no http_header_key)."""
296296 config_args = [
297297 MCPServerConfigArg (
@@ -303,12 +303,12 @@ def test_resolve_config_with_only_env_var() -> None:
303303 app = mcp_server ("test-server" , server_config_args = config_args )
304304
305305 with patch .dict (os .environ , {"TEST_API_KEY" : "env-only-value" }):
306- value = resolve_config (app , "api_key" )
306+ value = get_mcp_config (app , "api_key" )
307307 assert value == "env-only-value"
308308
309309
310310@pytest .mark .unit
311- def test_resolve_config_with_only_http_header () -> None :
311+ def test_get_mcp_config_with_only_http_header () -> None :
312312 """Test resolving config with only http_header_key set (no env_var)."""
313313 config_args = [
314314 MCPServerConfigArg (
@@ -323,5 +323,5 @@ def test_resolve_config_with_only_http_header() -> None:
323323 "fastmcp_extensions.server.get_http_headers" ,
324324 return_value = {"X-API-Key" : "header-only-value" },
325325 ):
326- value = resolve_config (app , "api_key" )
326+ value = get_mcp_config (app , "api_key" )
327327 assert value == "header-only-value"
0 commit comments