Fix: add uvicorn mock to resolve systemexit

sanggusti · sanggusti · commit 0222b791b4d3 · 2025-12-26T09:56:28.000Z
diff --git a/tests/unit/test_lit_server.py b/tests/unit/test_lit_server.py
@@ -854,6 +854,8 @@ def __init__(self, api_path="/predict"):
     ],
 )
 def test_workers_per_device_can_be_configured_per_route(monkeypatch, workers_cfg, expected_total_by_path):
+    monkeypatch.setattr("litserve.server.uvicorn", MagicMock())
+    
     sentiment = MultiRouteAPI(api_path="/sentiment")
     generate = MultiRouteAPI(api_path="/generate")
     server = LitServer([sentiment, generate], accelerator="cuda", devices=[0, 1], workers_per_device=workers_cfg)

Original file line number	Diff line number	Diff line change
`@@ -854,6 +854,8 @@ def __init__(self, api_path="/predict"):`
`854`	`854`	`],`
`855`	`855`	`)`
`856`	`856`	`def test_workers_per_device_can_be_configured_per_route(monkeypatch, workers_cfg, expected_total_by_path):`
	`857`	`+ monkeypatch.setattr("litserve.server.uvicorn", MagicMock())`
	`858`	`+`
`857`	`859`	`sentiment = MultiRouteAPI(api_path="/sentiment")`
`858`	`860`	`generate = MultiRouteAPI(api_path="/generate")`
`859`	`861`	`server = LitServer([sentiment, generate], accelerator="cuda", devices=[0, 1], workers_per_device=workers_cfg)`