|
1 | | -from openstack_mcp_server.prompts import register_prompt |
| 1 | +from unittest.mock import MagicMock |
2 | 2 |
|
| 3 | +from fastmcp import FastMCP |
3 | 4 |
|
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 |
7 | 6 |
|
8 | | - mcp = MagicMock() |
9 | | - register_prompt(mcp) |
10 | | - mcp.prompt.assert_called() |
11 | 7 |
|
| 8 | +class TestPrompts: |
| 9 | + """Test cases for MCP prompts.""" |
12 | 10 |
|
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() |
16 | 16 |
|
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) |
19 | 21 |
|
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 |
22 | 24 |
|
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 |
25 | 27 |
|
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