Deprecate ServerState in the main module

encode · Feb 12, 2025 · 9d7a483 · 9d7a483
1 parent d68f89e
commit 9d7a483
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/tests/test_main.py b/tests/test_main.py
@@ -6,6 +6,7 @@
 import httpx
 import pytest
 
+import uvicorn.server
 from tests.utils import run_server
 from uvicorn import Server
 from uvicorn._types import ASGIReceiveCallable, ASGISendCallable, Scope
@@ -122,4 +123,4 @@ def test_deprecated_server_state_from_main() -> None:
     ):
         main = importlib.import_module("uvicorn.main")
         server_state_cls = getattr(main, "ServerState")
-    assert server_state_cls is not None
+    assert server_state_cls is uvicorn.server.ServerState