test_variables.py

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=[Literal(value="b")])]),
        ("${a:=b}", [Variable(name="a", default=[Literal(value="b")])]),
        (
            "${a:-a${b:-c${d}e}f}",
            [
                Variable(name="a", default=[
                    Literal(value="a"),
                    Variable(name="b", default=[
                        Literal(value="c"),
                        Variable(name="d", default=None),
                        Literal(value="e")
                    ]),
                    Literal(value="f")
                ])
            ]
        ),
        (
            "${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