Skip to content

Commit 0c29e3c

Browse files
committed
style: align prompt test structure with tools test convention
Wrap test functions in TestPrompts class and move imports to module level, matching the pattern used in tests/tools/.
1 parent b5d7f91 commit 0c29e3c

File tree

1 file changed

+22
-20
lines changed

1 file changed

+22
-20
lines changed

tests/prompts/test_prompts.py

Lines changed: 22 additions & 20 deletions
Original file line numberDiff line numberDiff line change
@@ -1,29 +1,31 @@
1-
from openstack_mcp_server.prompts import register_prompt
1+
from unittest.mock import MagicMock
22

3+
from fastmcp import FastMCP
34

4-
def test_get_servers_by_security_group_prompt_registered():
5-
"""Test that the prompt is registered with the MCP instance."""
6-
from unittest.mock import MagicMock
5+
from openstack_mcp_server.prompts import register_prompt
76

8-
mcp = MagicMock()
9-
register_prompt(mcp)
10-
mcp.prompt.assert_called()
117

8+
class TestPrompts:
9+
"""Test cases for MCP prompts."""
1210

13-
def test_get_servers_by_security_group_prompt_content():
14-
"""Test that the prompt returns expected content."""
15-
from fastmcp import FastMCP
11+
def test_get_servers_by_security_group_prompt_registered(self):
12+
"""Test that the prompt is registered with the MCP instance."""
13+
mcp = MagicMock()
14+
register_prompt(mcp)
15+
mcp.prompt.assert_called()
1616

17-
mcp = FastMCP("test")
18-
register_prompt(mcp)
17+
def test_get_servers_by_security_group_prompt_content(self):
18+
"""Test that the prompt returns expected content."""
19+
mcp = FastMCP("test")
20+
register_prompt(mcp)
1921

20-
prompts = mcp._prompt_manager._prompts
21-
assert "get_servers_by_security_group" in prompts
22+
prompts = mcp._prompt_manager._prompts
23+
assert "get_servers_by_security_group" in prompts
2224

23-
prompt_obj = prompts["get_servers_by_security_group"]
24-
assert prompt_obj.fn is not None
25+
prompt_obj = prompts["get_servers_by_security_group"]
26+
assert prompt_obj.fn is not None
2527

26-
result = prompt_obj.fn(security_group_name="my-sg")
27-
assert "my-sg" in result
28-
assert "get_servers" in result
29-
assert "security_groups" in result
28+
result = prompt_obj.fn(security_group_name="my-sg")
29+
assert "my-sg" in result
30+
assert "get_servers" in result
31+
assert "security_groups" in result

0 commit comments

Comments
 (0)