vane 0.9.2

A flow-based reverse proxy with multi-layer routing and programmable pipelines.
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68

/* src/common/config/env_loader.rs */

use std::env;
use std::ffi::OsStr;

/// Gets an environment variable, returning a default value if it's not set.
pub fn get_env<T: AsRef<OsStr>>(key: T, default: String) -> String {
	env::var(key).unwrap_or(default)
}

/// Converts a string to lowercase.
#[must_use]
pub fn to_lowercase(s: &str) -> String {
	s.to_lowercase()
}

/// Checks if a string contains only numeric characters.
pub fn is_numeric(s: &str) -> bool {
	s.chars().all(char::is_numeric)
}

#[cfg(test)]
mod tests {
	use super::*;
	use serial_test::serial;
	use temp_env;

	/// Tests that get_env correctly retrieves a value when the environment variable is set.
	#[test]
	#[serial]
	fn test_get_env_retrieves_value() {
		temp_env::with_var("TEST_KEY_EXISTS", Some("my_value"), || {
			assert_eq!(
				get_env("TEST_KEY_EXISTS", "default".to_string()),
				"my_value"
			);
		});
	}

	/// Tests that get_env falls back to the default value when the variable is not set.
	#[test]
	#[serial]
	fn test_get_env_uses_default() {
		temp_env::with_var_unset("TEST_KEY_UNSET", || {
			assert_eq!(
				get_env("TEST_KEY_UNSET", "default_value".to_string()),
				"default_value"
			);
		});
	}

	/// Tests the string to_lowercase conversion utility.
	#[test]
	fn test_to_lowercase_converts_correctly() {
		assert_eq!(to_lowercase("Hello World"), "hello world");
		assert_eq!(to_lowercase("ALREADY_LOWER"), "already_lower");
		assert_eq!(to_lowercase("123!@#"), "123!@#");
	}

	/// Tests the numeric string validation utility.
	#[test]
	fn test_is_numeric_validates_correctly() {
		assert!(is_numeric("12345"));
		assert!(!is_numeric("123a45"));
		assert!(!is_numeric("123 45"));
		assert!(is_numeric("")); // An empty string is considered numeric by this logic.
	}
}