Skip to content

Commit 12af26b

Browse files
committed
Fix missing dependencies in test_schemas.py
Signed-off-by: Mihai Criveti <[email protected]>
1 parent 03f306c commit 12af26b

File tree

1 file changed

+15
-1
lines changed

1 file changed

+15
-1
lines changed

tests/unit/mcpgateway/test_schemas.py

Lines changed: 15 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -11,7 +11,8 @@
1111

1212
import json
1313
import os
14-
from datetime import datetime
14+
from datetime import datetime, timedelta
15+
from unittest.mock import Mock
1516

1617
import pytest
1718
from pydantic import ValidationError
@@ -44,6 +45,19 @@
4445
ToolResult,
4546
)
4647

48+
from mcpgateway.schemas import (
49+
EventMessage,
50+
AdminToolCreate,
51+
AdminGatewayCreate,
52+
ServerCreate,
53+
ServerUpdate,
54+
ServerRead,
55+
ServerMetrics,
56+
StatusToggleRequest,
57+
StatusToggleResponse,
58+
ListFilters,
59+
)
60+
4761
PROTOCOL_VERSION = os.getenv("PROTOCOL_VERSION", "2025-03-26")
4862

4963

0 commit comments

Comments
 (0)