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
|
import pytest
from dotenv.variables import Literal, Variable, parse_variables
@pytest.mark.parametrize(
"value,expected",
[
("", []),
("a", [Literal(value="a")]),
("${a}", [Variable(name="a", default=None)]),
("${a:-b}", [Variable(name="a", default="b")]),
(
"${a}${b}",
[
Variable(name="a", default=None),
Variable(name="b", default=None),
],
),
(
"a${b}c${d}e",
[
Literal(value="a"),
Variable(name="b", default=None),
Literal(value="c"),
Variable(name="d", default=None),
Literal(value="e"),
],
),
]
)
def test_parse_variables(value, expected):
result = parse_variables(value)
assert list(result) == expected
|
