Divide test in two

theskumar · Nov 14, 2023 · 65c9a44 · 65c9a44
1 parent b5eeb4b
commit 65c9a44
Showing 1 changed file with 21 additions and 0 deletions.
diff --git a/tests/test_main.py b/tests/test_main.py
@@ -465,8 +465,29 @@ def test_required_variable_throws(string, message):
         ("XX=$TEST", "$TEST"),
         ("XX=${TEST}", "tt"),
         ("XX=\"${TEST}\"", "tt"),
+        ("XX='${TEST}'", "tt"),
         ("XX='$TEST'", "$TEST"),
+        ("XX='\\$\\{TEST\\}'", "\\$\\{TEST\\}"),
         ("XX=\\$\\{TEST\\}", "\\$\\{TEST\\}"),
+        ("XX=\"\\$\\{TEST\\}\"", "\\$\\{TEST\\}"),
+    ],
+)
+def test_document_expansions(string, expected_xx):
+    test_env = {"TEST": "tt"}
+    with mock.patch.dict(os.environ, test_env, clear=True):
+        stream = io.StringIO(string)
+        stream.seek(0)
+
+        result = dotenv.dotenv_values(stream=stream, interpolate=True)
+
+        assert result["XX"] == expected_xx
+
+
+@pytest.mark.parametrize(
+    "string,expected_xx",
+    [
+        ("XX=${TEST}", "tt"),
+        ("XX=\"${TEST}\"", "tt"),
         ("XX='${TEST}'", "${TEST}"),
     ],
 )