starred/servers-modelcontextprotocol
1
0
Fork 0
mirror of https://github.com/modelcontextprotocol/servers.git synced 2026-04-18 16:13:22 +02:00
Code Issues Packages Projects Releases Wiki Activity

fix time server

Browse Source
This commit is contained in:
David Soria Parra
2024-11-29 11:28:45 +00:00
parent 18f6ab6ce8
commit bccd33f7a1
5 changed files with 78 additions and 56 deletions
Download Patch File Download Diff File Expand all files Collapse all files

15
src/time/test/time_server_test.py
View File

@@ -1,9 +1,9 @@
import os
from freezegun import freeze_time
from mcp.shared.exceptions import McpError
import pytest
from mcp_server_time.server import TimeServer, serve
from mcp_server_time.server import TimeServer
@pytest.mark.parametrize(
@@ -82,7 +82,10 @@ def test_get_current_time(test_time, timezone, expected):
def test_get_current_time_with_invalid_timezone():
time_server = TimeServer()
with pytest.raises(ValueError, match=r"Unknown timezone: 'Invalid/Timezone'"):
with pytest.raises(
McpError,
match=r"Invalid timezone: 'No time zone found with key Invalid/Timezone'",
):
time_server.get_current_time("Invalid/Timezone")
@@ -93,13 +96,13 @@ def test_get_current_time_with_invalid_timezone():
"invalid_tz",
"12:00",
"Europe/London",
"Unknown source timezone: 'invalid_tz'",
"Invalid timezone: 'No time zone found with key invalid_tz'",
),
(
"Europe/Warsaw",
"12:00",
"invalid_tz",
"Unknown target timezone: 'invalid_tz'",
"Invalid timezone: 'No time zone found with key invalid_tz'",
),
(
"Europe/Warsaw",
@@ -111,7 +114,7 @@ def test_get_current_time_with_invalid_timezone():
)
def test_convert_time_errors(source_tz, time_str, target_tz, expected_error):
time_server = TimeServer()
with pytest.raises(ValueError, match=expected_error):
with pytest.raises((McpError, ValueError), match=expected_error):
time_server.convert_time(source_tz, time_str, target_tz)