Skip to content

Commit e798d20

Browse files
committed
ruff
1 parent aa7869a commit e798d20

File tree

1 file changed

+7
-7
lines changed

1 file changed

+7
-7
lines changed

tests/shared/test_sse.py

Lines changed: 7 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -1,21 +1,22 @@
11
import multiprocessing
22
import socket
33
import time
4+
from typing import AsyncGenerator, Generator
5+
46
import anyio
5-
from starlette.requests import Request
6-
import uvicorn
7+
import httpx
78
import pytest
9+
import uvicorn
810
from pydantic import AnyUrl
9-
import httpx
10-
from typing import AsyncGenerator, Generator
1111
from starlette.applications import Starlette
12+
from starlette.requests import Request
1213
from starlette.routing import Mount, Route
1314

14-
from mcp.shared.exceptions import McpError
1515
from mcp.client.session import ClientSession
1616
from mcp.client.sse import sse_client
1717
from mcp.server import Server
1818
from mcp.server.sse import SseServerTransport
19+
from mcp.shared.exceptions import McpError
1920
from mcp.types import (
2021
EmptyResult,
2122
ErrorData,
@@ -157,8 +158,7 @@ async def http_client(server, server_url) -> AsyncGenerator[httpx.AsyncClient, N
157158
@pytest.mark.anyio
158159
async def test_raw_sse_connection(http_client: httpx.AsyncClient) -> None:
159160
"""Test the SSE connection establishment simply with an HTTP client."""
160-
async with anyio.create_task_group() as tg:
161-
161+
async with anyio.create_task_group():
162162
async def connection_test() -> None:
163163
async with http_client.stream("GET", "/sse") as response:
164164
assert response.status_code == 200

0 commit comments

Comments
 (0)