update test_axon.py

Author: Roman

tests/unit_tests/test_axon.py

@@ -1,21 +1,18 @@
 import re
-import threading
 import time
 from dataclasses import dataclass
 from typing import Any, Optional, Tuple
 from unittest import IsolatedAsyncioTestCase
 from unittest.mock import AsyncMock, MagicMock, patch
 
-import aiohttp
 import fastapi
 import netaddr
 import pydantic
 import pytest
-import uvicorn
 from fastapi.testclient import TestClient
 from starlette.requests import Request
 
-from bittensor.core.axon import Axon, AxonMiddleware, FastAPIThreadedServer
+from bittensor.core.axon import AxonMiddleware, Axon
 from bittensor.core.errors import RunException
 from bittensor.core.settings import version_as_int
 from bittensor.core.stream import StreamingSynapse
@@ -768,45 +765,3 @@ async def forward_fn(synapse: streaming_synapse_cls):
                 "computed_body_hash": "a7ffc6f8bf1ed76651c14756a061d662f580ff4de43b49fa82d80a4b80f8434a",
             },
         )
-
-
-@pytest.mark.asyncio
-async def test_threaded_fastapi():
-    server_started = threading.Event()
-    server_stopped = threading.Event()
-
-    async def lifespan(app):
-        server_started.set()
-        yield
-        server_stopped.set()
-
-    app = fastapi.FastAPI(
-        lifespan=lifespan,
-    )
-    app.get("/")(lambda: "Hello World")
-
-    server = FastAPIThreadedServer(
-        uvicorn.Config(
-            app,
-        ),
-    )
-    server.start()
-
-    server_started.wait()
-
-    assert server.is_running is True
-
-    async with aiohttp.ClientSession(
-        base_url="http://127.0.0.1:8000",
-    ) as session:
-        async with session.get("/") as response:
-            assert await response.text() == '"Hello World"'
-
-        server.stop()
-
-        assert server.should_exit is True
-
-        server_stopped.wait()
-
-        with pytest.raises(aiohttp.ClientConnectorError):
-            await session.get("/")