From 850f0a005050ff0e1701cbef91c1f3e919e77039 Mon Sep 17 00:00:00 2001
From: Jeremiah Lowin <153965+jlowin@users.noreply.github.com>
Date: Sat, 17 Jan 2026 14:46:21 -0500
Subject: [PATCH] fix: update all tests to use get_prompt instead of
 _get_prompt

---
 .../providers/test_local_provider_prompts.py   | 18 +++++++++---------
 1 file changed, 9 insertions(+), 9 deletions(-)

diff --git a/tests/server/providers/test_local_provider_prompts.py b/tests/server/providers/test_local_provider_prompts.py
index 666ecb62d..2ffea845b 100644
--- a/tests/server/providers/test_local_provider_prompts.py
+++ b/tests/server/providers/test_local_provider_prompts.py
@@ -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"