ceos/fastmcp
1
0
Fork 0
mirror of https://github.com/PrefectHQ/fastmcp.git synced 2026-08-27 07:50:43 +02:00
Code Issues Projects Releases Packages Wiki Activity

Refactor transform list methods to pure function pattern (#2942)

Browse source
This commit is contained in:
Jeremiah Lowin 2026-01-19 16:21:35 -05:00 • committed by GitHub
commit 4d2feb0c29
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
14 changed files with 143 additions and 334 deletions
Download patch file Download diff file Expand all files Collapse all files

21
tests/server/transforms/test_enabled.py
View file

@ -236,10 +236,7 @@ class TestTransformChain:
"""list_tools applies marks to matching components."""
disable_internal = Enabled(False, tags=set({"internal"}))
async def base():
return tools
result = await disable_internal.list_tools(base)
result = await disable_internal.list_tools(tools)
assert len(result) == 3
assert is_enabled(result[0]) # public
@ -251,12 +248,8 @@ class TestTransformChain:
disable_internal = Enabled(False, tags=set({"internal"}))
enable_safe = Enabled(True, tags=set({"safe"}))
async def base():
return tools
async def after_disable():
return await disable_internal.list_tools(base)
# Apply transforms sequentially
after_disable = await disable_internal.list_tools(tools)
result = await enable_safe.list_tools(after_disable)
enabled = [t for t in result if is_enabled(t)]
@ -270,12 +263,8 @@ class TestTransformChain:
disable_all = Enabled(False, match_all=True)
enable_public = Enabled(True, tags=set({"public"}))
async def base():
return tools
async def after_disable():
return await disable_all.list_tools(base)
# Apply transforms sequentially
after_disable = await disable_all.list_tools(tools)
result = await enable_public.list_tools(after_disable)
enabled = [t for t in result if is_enabled(t)]