move get_log_level into Server class

This commit is contained in:
Stefan Allius
2025-05-05 23:47:06 +02:00
parent 089fb92a43
commit ab2d4ed831

View File

@@ -3,34 +3,41 @@ import pytest
import logging import logging
import os import os
from mock import patch from mock import patch
from server import get_log_level, app, ProxyState from app import app, ProxyState
from server import Server
pytest_plugins = ('pytest_asyncio',) pytest_plugins = ('pytest_asyncio',)
class FakeServer(Server):
def __init__(self):
pass # don't call the suoer(.__init__ for unit tests
def test_get_log_level(): def test_get_log_level():
s = FakeServer()
with patch.dict(os.environ, {}): with patch.dict(os.environ, {}):
log_lvl = get_log_level() log_lvl = s.get_log_level()
assert log_lvl == None assert log_lvl == None
with patch.dict(os.environ, {'LOG_LVL': 'DEBUG'}): with patch.dict(os.environ, {'LOG_LVL': 'DEBUG'}):
log_lvl = get_log_level() log_lvl = s.get_log_level()
assert log_lvl == logging.DEBUG assert log_lvl == logging.DEBUG
with patch.dict(os.environ, {'LOG_LVL': 'INFO'}): with patch.dict(os.environ, {'LOG_LVL': 'INFO'}):
log_lvl = get_log_level() log_lvl = s.get_log_level()
assert log_lvl == logging.INFO assert log_lvl == logging.INFO
with patch.dict(os.environ, {'LOG_LVL': 'WARN'}): with patch.dict(os.environ, {'LOG_LVL': 'WARN'}):
log_lvl = get_log_level() log_lvl = s.get_log_level()
assert log_lvl == logging.WARNING assert log_lvl == logging.WARNING
with patch.dict(os.environ, {'LOG_LVL': 'ERROR'}): with patch.dict(os.environ, {'LOG_LVL': 'ERROR'}):
log_lvl = get_log_level() log_lvl = s.get_log_level()
assert log_lvl == logging.ERROR assert log_lvl == logging.ERROR
with patch.dict(os.environ, {'LOG_LVL': 'UNKNOWN'}): with patch.dict(os.environ, {'LOG_LVL': 'UNKNOWN'}):
log_lvl = get_log_level() log_lvl = s.get_log_level()
assert log_lvl == None assert log_lvl == None
@pytest.mark.asyncio @pytest.mark.asyncio