python-utils-toolkit/tests/test_env.py at main · Ghraven/python-utils-toolkit · 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
"""Tests for utils/env.py"""
import os
import pytest
from utils.env import Env


@pytest.fixture(autouse=True)
def clean_env():
    """Remove test keys before and after each test."""
    test_keys = ["TEST_STR", "TEST_INT", "TEST_FLOAT", "TEST_BOOL", "TEST_LIST"]
    for k in test_keys:
        os.environ.pop(k, None)
    yield
    for k in test_keys:
        os.environ.pop(k, None)


def test_str_from_env():
    os.environ["TEST_STR"] = "hello"
    env = Env(dotenv_path=".nonexistent")
    assert env.str("TEST_STR") == "hello"


def test_str_default():
    env = Env(dotenv_path=".nonexistent")
    assert env.str("TEST_STR", default="fallback") == "fallback"


def test_str_missing_raises():
    env = Env(dotenv_path=".nonexistent")
    with pytest.raises(KeyError):
        env.str("TEST_STR")


def test_int_from_env():
    os.environ["TEST_INT"] = "42"
    env = Env(dotenv_path=".nonexistent")
    assert env.int("TEST_INT") == 42


def test_int_default():
    env = Env(dotenv_path=".nonexistent")
    assert env.int("TEST_INT", default=8080) == 8080


def test_int_invalid_raises():
    os.environ["TEST_INT"] = "not_a_number"
    env = Env(dotenv_path=".nonexistent")
    with pytest.raises(ValueError):
        env.int("TEST_INT")


def test_float_from_env():
    os.environ["TEST_FLOAT"] = "3.14"
    env = Env(dotenv_path=".nonexistent")
    assert env.float("TEST_FLOAT") == pytest.approx(3.14)


def test_bool_true_values():
    env = Env(dotenv_path=".nonexistent")
    for val in ["1", "true", "True", "yes", "on"]:
        os.environ["TEST_BOOL"] = val
        assert env.bool("TEST_BOOL") is True


def test_bool_false_values():
    env = Env(dotenv_path=".nonexistent")
    for val in ["0", "false", "no", "off", ""]:
        os.environ["TEST_BOOL"] = val
        assert env.bool("TEST_BOOL") is False


def test_list_from_env():
    os.environ["TEST_LIST"] = "a,b,c"
    env = Env(dotenv_path=".nonexistent")
    assert env.list("TEST_LIST") == ["a", "b", "c"]


def test_list_missing_returns_empty():
    env = Env(dotenv_path=".nonexistent")
    assert env.list("TEST_LIST") == []


def test_loads_from_dotenv_file(tmp_path):
    dotenv = tmp_path / ".env"
    dotenv.write_text("TEST_STR=from_file\n", encoding="utf-8")
    os.environ.pop("TEST_STR", None)
    env = Env(dotenv_path=dotenv)
    assert env.str("TEST_STR") == "from_file"


def test_dotenv_does_not_override_existing(tmp_path):
    dotenv = tmp_path / ".env"
    dotenv.write_text("TEST_STR=from_file\n", encoding="utf-8")
    os.environ["TEST_STR"] = "from_env"
    env = Env(dotenv_path=dotenv)
    assert env.str("TEST_STR") == "from_env"