Merge pull request #6 from sherifzain/truncate-values

Truncate values before evaluation

envargs/parser.py

@@ -22,7 +22,7 @@ def _load_values(values, fields):
                 yield dest, field.default
 
             else:
-                value = field.parse(value, location)
+                value = field.parse(value.strip(), location)
                 field.validate(value, location)
                 yield dest, value
         elif isinstance(field, dict):

tests/test_parsing.py

@@ -232,3 +232,29 @@ def test_nesting():
             'an_int': 10,
         },
     }
+
+
+def test_numeric_var_with_spaces():
+    """Test that numbers are parsed properly even if they have spaces."""
+    args = {
+        'an_int': Var(use=int, load_from='AN_INT',),
+    }
+
+    os.environ['AN_INT'] = '0   '
+
+    assert parse_env(args) == {
+        'an_int': 0,
+    }
+
+
+def test_string_var_with_spaces():
+    """Test that string values are truncated."""
+    args = {
+        'a_var': Var(use=str, load_from='A_VAR',),
+    }
+
+    os.environ['A_VAR'] = ' username\n'
+
+    assert parse_env(args) == {
+        'a_var': 'username',
+    }