Merge pull request #185 from matthiasbeyer/custom-env-seperator

Fix custom separator for group prefixes

Author: matthiasbeyer

src/env.rs

@@ -76,8 +76,16 @@ impl Source for Environment {
             _ => "",
         };
 
+        let group_separator = match self.separator {
+            Some(ref separator) => separator,
+            _ => "_",
+        };
+
         // Define a prefix pattern to test and exclude from keys
-        let prefix_pattern = self.prefix.as_ref().map(|prefix| prefix.clone() + "_");
+        let prefix_pattern = self
+            .prefix
+            .as_ref()
+            .map(|prefix| format!("{}{}", prefix.clone(), group_separator));
 
         for (key, value) in env::vars() {
             // Treat empty environment variables as unset

tests/env.rs

@@ -71,3 +71,14 @@ fn test_empty_value_is_ignored() {
 
     env::remove_var("C_A_B");
 }
+
+#[test]
+fn test_custom_separator_behavior() {
+    env::set_var("C.B.A", "abc");
+
+    let environment = Environment::with_prefix("C").separator(".");
+
+    assert!(environment.collect().unwrap().contains_key("b.a"));
+
+    env::remove_var("C.B.A");
+}