Add test for key/value stripping

Saurabh Kumar · Saurabh Kumar · commit ab268f09fae8 · 2016-08-13T19:47:23.000+05:30
closes theskumar#29
diff --git a/README.rst b/README.rst
@@ -212,6 +212,11 @@ would not have been possible without the support of these `awesome people <https
 Changelog
 =========
 
+dev
+----------
+- Handle escaped charaters and newlines in quoted values
+- Remove any spaces around unquoted key/value
+
 0.5.1
 ----------
 - Fix `find_dotenv` - it now start search from the file where this function is called from.
diff --git a/dotenv/main.py b/dotenv/main.py
@@ -1,4 +1,6 @@
 # -*- coding: utf-8 -*-
+from __future__ import absolute_import
+
 import os
 import sys
 import warnings
@@ -89,8 +91,10 @@ def parse_dotenv(dotenv_path):
             if not line or line.startswith('#') or '=' not in line:
                 continue
             k, v = line.split('=', 1)
-            k = k.strip()
-            v = v.strip()
+
+            # Remove any leading and trailing spaces in key, value
+            k, v = k.strip(), v.strip()
+
             if len(v) > 0:
                 quoted = v[0] == v[len(v) - 1] == '"'
 
diff --git a/tests/test_cli.py b/tests/test_cli.py
@@ -22,6 +22,18 @@ def test_get_key():
     assert success is None
 
 
+def test_key_value_without_quotes():
+    with open(dotenv_path, 'w') as f:
+        f.write("TEST = value \n")
+    assert dotenv.get_key(dotenv_path, 'TEST') == "value"
+    sh.rm(dotenv_path)
+
+    with open(dotenv_path, 'w') as f:
+        f.write('TEST = " with spaces " \n')
+    assert dotenv.get_key(dotenv_path, 'TEST') == " with spaces "
+    sh.rm(dotenv_path)
+
+
 def test_unset():
     sh.touch(dotenv_path)
     success, key_to_set, value_to_set = dotenv.set_key(dotenv_path, 'HELLO', 'WORLD')
