vllm-project--vllm-omni
wehub-resource-sync
85 行
2.7 KiB
Python
85 行
2.7 KiB
Python
# SPDX-License-Identifier: Apache-2.0
|
|
# SPDX-FileCopyrightText: Copyright contributors to the vLLM project
|
|
"""Unit tests for ServerSettings env-var parameterization."""
|
|
|
|
import pytest
|
|
|
|
from vllm_omni.config.server_settings import ServerSettings
|
|
|
|
pytestmark = [pytest.mark.core_model, pytest.mark.cpu]
|
|
|
|
ENV_VARS = [
|
|
"VLLM_OMNI_SERVER_STORAGE__PATH",
|
|
"VLLM_OMNI_SERVER_STORAGE__FILE_CONCURRENCY",
|
|
"VLLM_OMNI_SERVER_STORAGE__FILE_TTL",
|
|
"VLLM_OMNI_SERVER_STORAGE__TTL_SWEEP_INTERVAL",
|
|
"VLLM_OMNI_STORAGE_PATH",
|
|
"VLLM_OMNI_STORAGE_MAX_CONCURRENCY",
|
|
]
|
|
|
|
|
|
@pytest.fixture(autouse=True)
|
|
def clean_env(monkeypatch):
|
|
for name in ENV_VARS:
|
|
monkeypatch.delenv(name, raising=False)
|
|
|
|
|
|
def test_defaults_when_no_env_set():
|
|
storage = ServerSettings().storage
|
|
|
|
assert storage.path == "/tmp/storage"
|
|
assert storage.file_concurrency == 4
|
|
assert storage.file_ttl is None
|
|
assert storage.ttl_sweep_interval is None
|
|
|
|
|
|
def test_nested_env_vars_populate_storage(monkeypatch):
|
|
monkeypatch.setenv("VLLM_OMNI_SERVER_STORAGE__PATH", "/var/storage")
|
|
monkeypatch.setenv("VLLM_OMNI_SERVER_STORAGE__FILE_CONCURRENCY", "8")
|
|
monkeypatch.setenv("VLLM_OMNI_SERVER_STORAGE__FILE_TTL", "120")
|
|
monkeypatch.setenv("VLLM_OMNI_SERVER_STORAGE__TTL_SWEEP_INTERVAL", "30")
|
|
|
|
storage = ServerSettings().storage
|
|
|
|
assert storage.path == "/var/storage"
|
|
assert storage.file_concurrency == 8
|
|
assert storage.file_ttl == 120
|
|
assert storage.ttl_sweep_interval == 30
|
|
|
|
|
|
def test_ttl_sweep_interval_defaults_when_ttl_set(monkeypatch):
|
|
monkeypatch.setenv("VLLM_OMNI_SERVER_STORAGE__FILE_TTL", "60")
|
|
|
|
storage = ServerSettings().storage
|
|
|
|
assert storage.file_ttl == 60
|
|
assert storage.ttl_sweep_interval == 300
|
|
|
|
|
|
def test_deprecated_path_alias_populates_and_warns(monkeypatch):
|
|
monkeypatch.setenv("VLLM_OMNI_STORAGE_PATH", "/legacy/path")
|
|
|
|
with pytest.warns(DeprecationWarning, match="VLLM_OMNI_STORAGE_PATH"):
|
|
storage = ServerSettings().storage
|
|
|
|
assert storage.path == "/legacy/path"
|
|
|
|
|
|
def test_deprecated_concurrency_alias_populates_and_warns(monkeypatch):
|
|
monkeypatch.setenv("VLLM_OMNI_STORAGE_MAX_CONCURRENCY", "16")
|
|
|
|
with pytest.warns(DeprecationWarning, match="VLLM_OMNI_STORAGE_MAX_CONCURRENCY"):
|
|
storage = ServerSettings().storage
|
|
|
|
assert storage.file_concurrency == 16
|
|
|
|
|
|
def test_new_env_var_wins_over_deprecated_alias(monkeypatch):
|
|
monkeypatch.setenv("VLLM_OMNI_SERVER_STORAGE__PATH", "/new/path")
|
|
monkeypatch.setenv("VLLM_OMNI_STORAGE_PATH", "/legacy/path")
|
|
|
|
with pytest.warns(DeprecationWarning, match="VLLM_OMNI_STORAGE_PATH"):
|
|
storage = ServerSettings().storage
|
|
|
|
assert storage.path == "/new/path"
|