fastapi-clean-example/tests/unit/main/config/test_loader.py at 216fab53a73ab62b374a6f1f4cc1eb6f78e8a8ac · ivan-borovets/fastapi-clean-example · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
import pytest

from app.main.config.loader import (
    load_app_settings,
    load_cookie_settings,
    load_jwt_settings,
    load_password_hasher_settings,
    load_postgres_settings,
    load_session_settings,
    load_sqla_settings,
)
from app.main.config.logging_ import LoggingLevel


@pytest.mark.parametrize(
    "logging_level",
    [
        LoggingLevel.DEBUG,
        LoggingLevel.INFO,
        LoggingLevel.WARNING,
        LoggingLevel.ERROR,
        LoggingLevel.CRITICAL,
    ],
)
def test_load_app_settings_reads_env_vars(monkeypatch: pytest.MonkeyPatch, logging_level: LoggingLevel) -> None:
    monkeypatch.setenv("APP_SERVICE_NAME", "test-service")
    monkeypatch.setenv("APP_VERSION", "test-version")
    monkeypatch.setenv("APP_ROOT_PATH", "test-path")
    monkeypatch.setenv("APP_DEBUG_MODE", "1")
    monkeypatch.setenv("APP_LOGGING_LEVEL", logging_level)

    sut = load_app_settings()

    assert sut.SERVICE_NAME == "test-service"
    assert sut.VERSION == "test-version"
    assert sut.ROOT_PATH == "test-path"
    assert sut.DEBUG_MODE is True
    assert sut.LOGGING_LEVEL == logging_level


def test_load_postgres_settings_reads_env_vars(monkeypatch: pytest.MonkeyPatch) -> None:
    monkeypatch.setenv("POSTGRES_DB", "test-db")
    monkeypatch.setenv("POSTGRES_HOST", "test-host")
    monkeypatch.setenv("POSTGRES_PORT", "123456789")
    monkeypatch.setenv("POSTGRES_USER", "test-user")
    monkeypatch.setenv("POSTGRES_PASSWORD", "test-password")

    sut = load_postgres_settings()

    assert sut.DB == "test-db"
    assert sut.HOST == "test-host"
    assert sut.PORT == 123456789
    assert sut.USER == "test-user"
    assert sut.PASSWORD == "test-password"


def test_load_sqla_settings_reads_env_vars(monkeypatch: pytest.MonkeyPatch) -> None:
    monkeypatch.setenv("SQLA_ECHO", "true")
    monkeypatch.setenv("SQLA_ECHO_POOL", "true")
    monkeypatch.setenv("SQLA_POOL_SIZE", "123456789")
    monkeypatch.setenv("SQLA_MAX_OVERFLOW", "987654321")

    sut = load_sqla_settings()

    assert sut.ECHO is True
    assert sut.ECHO_POOL is True
    assert sut.POOL_SIZE == 123456789
    assert sut.MAX_OVERFLOW == 987654321


def test_load_password_hasher_settings_reads_env_vars(monkeypatch: pytest.MonkeyPatch) -> None:
    monkeypatch.setenv("PASSWORD_PEPPER", "test-pepper-test-pepper-test-pepper")
    monkeypatch.setenv("PASSWORD_WORK_FACTOR", "123456789")
    monkeypatch.setenv("PASSWORD_MAX_THREADS", "987654321")
    monkeypatch.setenv("PASSWORD_SEMAPHORE_WAIT_TIMEOUT_S", "1.23456789")

    sut = load_password_hasher_settings()

    assert sut.PEPPER == "test-pepper-test-pepper-test-pepper"
    assert sut.WORK_FACTOR == 123456789
    assert sut.MAX_THREADS == 987654321
    assert sut.SEMAPHORE_WAIT_TIMEOUT_S == 1.23456789


def test_load_jwt_settings_reads_env_vars(monkeypatch: pytest.MonkeyPatch) -> None:
    monkeypatch.setenv("JWT_SECRET", "test-secret-test-secret-test-secret")
    monkeypatch.setenv("JWT_ALGORITHM", "HS384")

    sut = load_jwt_settings()

    assert sut.SECRET == "test-secret-test-secret-test-secret"
    assert sut.ALGORITHM == "HS384"


def test_load_session_settings_reads_env_vars(monkeypatch: pytest.MonkeyPatch) -> None:
    monkeypatch.setenv("SESSION_TTL_MIN", "123465789")
    monkeypatch.setenv("SESSION_REFRESH_THRESHOLD_RATIO", "0.123456789")

    sut = load_session_settings()

    assert sut.TTL_MIN == 123465789
    assert sut.REFRESH_THRESHOLD_RATIO == 0.123456789


def test_load_cookie_settings_reads_env_vars(monkeypatch: pytest.MonkeyPatch) -> None:
    monkeypatch.setenv("COOKIE_NAME", "test-name")
    monkeypatch.setenv("COOKIE_PATH", "test-path")
    monkeypatch.setenv("COOKIE_HTTPONLY", "1")
    monkeypatch.setenv("COOKIE_SECURE", "false")
    monkeypatch.setenv("COOKIE_SAMESITE", "strict")

    sut = load_cookie_settings()

    assert sut.NAME == "test-name"
    assert sut.PATH == "test-path"
    assert sut.HTTPONLY is True
    assert sut.SECURE is False
    assert sut.SAMESITE == "strict"