Add tests for parse_line

Saurabh Kumar · Saurabh Kumar · commit 4b8c7231f70c · 2018-03-16T22:59:40.000+05:30
diff --git a/tests/test_core.py b/tests/test_core.py
@@ -8,10 +8,27 @@
 import sh
 
 from dotenv import load_dotenv, find_dotenv, set_key, dotenv_values
+from dotenv.main import parse_line
 from dotenv.compat import StringIO
 from IPython.terminal.embed import InteractiveShellEmbed
 
 
+@pytest.mark.parametrize("test_input,expected", [
+    ("a=b", ("a", "b")),
+    (" a = b ", ("a", "b")),
+    ("export a=b", ("a", "b")),
+    (" export 'a'=b", ("'a'", "b")),
+    (" export 'a'=b", ("'a'", "b")),
+    ("# a=b", (None, None)),
+    ("# a=b", (None, None)),
+    ("a=b space ", ('a', 'b space')),
+    ("a='b space '", ('a', 'b space ')),
+    ('a="b space "', ('a', 'b space ')),
+])
+def test_parse_line(test_input, expected):
+    assert parse_line(test_input) == expected
+
+
 def test_warns_if_file_does_not_exist():
     with warnings.catch_warnings(record=True) as w:
         load_dotenv('.does_not_exist', verbose=True)
