ceos/fastmcp
1
0
Fork 0
mirror of https://github.com/PrefectHQ/fastmcp.git synced 2026-08-22 13:34:17 +02:00
Code Issues Projects Releases Packages Wiki Activity

fix: update all tests to use get_prompt instead of _get_prompt

Browse source
This commit is contained in:
Jeremiah Lowin 2026-01-17 14:46:21 -05:00
commit 850f0a0050
1 changed files with 9 additions and 9 deletions
Download patch file Download diff file Expand all files Collapse all files

18
tests/server/providers/test_local_provider_prompts.py
View file

@ -377,8 +377,8 @@ class TestPromptEnabled:
prompts = await mcp.get_prompts()
assert len(prompts) == 0
# _get_prompt() applies visibility transform, returns None for disabled
prompt = await mcp._get_prompt("sample_prompt")
# get_prompt() applies visibility transform, returns None for disabled
prompt = await mcp.get_prompt("sample_prompt")
assert prompt is None
async def test_get_prompt_and_disable(self):
@ -388,15 +388,15 @@ class TestPromptEnabled:
def sample_prompt() -> str:
return "Hello, world!"
prompt = await mcp._get_prompt("sample_prompt")
prompt = await mcp.get_prompt("sample_prompt")
assert prompt is not None
mcp.disable(keys=["prompt:sample_prompt@"])
prompts = await mcp.get_prompts()
assert len(prompts) == 0
# _get_prompt() applies visibility transform, returns None for disabled
prompt = await mcp._get_prompt("sample_prompt")
# get_prompt() applies visibility transform, returns None for disabled
prompt = await mcp.get_prompt("sample_prompt")
assert prompt is None
async def test_cant_get_disabled_prompt(self):
@ -408,8 +408,8 @@ class TestPromptEnabled:
mcp.disable(keys=["prompt:sample_prompt@"])
# _get_prompt() applies visibility transform, returns None for disabled
prompt = await mcp._get_prompt("sample_prompt")
# get_prompt() applies visibility transform, returns None for disabled
prompt = await mcp.get_prompt("sample_prompt")
assert prompt is None
@ -459,7 +459,7 @@ class TestPromptTags:
result = await prompt.render({})
assert result.messages[0].content.text == "1"
prompt = await mcp._get_prompt("prompt_2")
prompt = await mcp.get_prompt("prompt_2")
assert prompt is None
async def test_read_prompt_excludes_tags(self):
@ -468,6 +468,6 @@ class TestPromptTags:
prompt = await mcp._get_prompt("prompt_1")
assert prompt is None
prompt = await mcp._get_prompt("prompt_2")
prompt = await mcp.get_prompt("prompt_2")
result = await prompt.render({})
assert result.messages[0].content.text == "2"