secrets_provider_tests/
env.rs

1use crate::retry::parse_bool_env;
2use serde_json::json;
3use std::env;
4use std::fmt::Write as _;
5use std::time::{SystemTime, UNIX_EPOCH};
6
7/// Parsed test environment configuration.
8#[derive(Debug, Clone)]
9pub struct TestEnv {
10    pub prefix: TestPrefix,
11    pub cleanup: bool,
12}
13
14impl TestEnv {
15    pub fn from_env(provider: &str) -> Self {
16        let prefix = TestPrefix::from_env(provider);
17
18        let keep = parse_bool_env("GREENTIC_TEST_KEEP");
19        let cleanup = if keep {
20            false
21        } else if let Ok(value) = env::var("GREENTIC_TEST_CLEANUP") {
22            parse_bool_env_value(&value, true)
23        } else {
24            true
25        };
26
27        Self { prefix, cleanup }
28    }
29}
30
31#[derive(Debug, Clone)]
32pub struct TestPrefix {
33    provider: String,
34    base: String,
35    counter: std::sync::Arc<std::sync::atomic::AtomicU64>,
36}
37
38impl TestPrefix {
39    pub fn from_env(provider: &str) -> Self {
40        if let Ok(explicit) = env::var("GREENTIC_TEST_PREFIX") {
41            return Self::new(provider, explicit);
42        }
43
44        let run_id = env::var("GITHUB_RUN_ID").ok();
45        let run_attempt = env::var("GITHUB_RUN_ATTEMPT").ok();
46        let repo = env::var("GITHUB_REPOSITORY").unwrap_or_else(|_| "local".to_string());
47
48        if let (Some(id), Some(attempt)) = (run_id, run_attempt) {
49            let base = format!("ci/{provider}/{repo}/{id}/{attempt}");
50            return Self::new(provider, base);
51        }
52
53        let now = SystemTime::now()
54            .duration_since(UNIX_EPOCH)
55            .unwrap_or_default()
56            .as_secs();
57        let pid = std::process::id();
58        let mut base = String::from("local/");
59        let _ = write!(&mut base, "{provider}/{now}/{pid}");
60        Self::new(provider, base)
61    }
62
63    fn new(provider: &str, base: String) -> Self {
64        Self {
65            provider: provider.to_string(),
66            base,
67            counter: std::sync::Arc::new(std::sync::atomic::AtomicU64::new(0)),
68        }
69    }
70
71    /// Returns a unique key prefix for the current test run.
72    pub fn base(&self) -> String {
73        self.base.clone()
74    }
75
76    /// Derive a unique secret key for a test case.
77    pub fn key(&self, suffix: &str) -> String {
78        let next = self
79            .counter
80            .fetch_add(1, std::sync::atomic::Ordering::SeqCst);
81        format!("{}/{suffix}-{next}", self.base)
82    }
83
84    /// Minimal JSON metadata used in debugging output.
85    pub fn to_metadata(&self) -> serde_json::Value {
86        json!({
87            "provider": self.provider,
88            "prefix": self.base,
89        })
90    }
91}
92
93fn parse_bool_env_value(value: &str, default_true: bool) -> bool {
94    match value {
95        "" => default_true,
96        v if v.eq_ignore_ascii_case("1") || v.eq_ignore_ascii_case("true") => true,
97        v if v.eq_ignore_ascii_case("0") || v.eq_ignore_ascii_case("false") => false,
98        _ => default_true,
99    }
100}
101
102#[cfg(test)]
103mod tests {
104    use super::*;
105
106    #[test]
107    fn builds_local_prefix_when_no_ci_env() {
108        let prefix = TestPrefix::new("dev", "local/test/123".to_string());
109        assert!(prefix.base().starts_with("local/test/123"));
110        let k1 = prefix.key("a");
111        let k2 = prefix.key("a");
112        assert_ne!(k1, k2);
113    }
114}
secrets_provider_tests/env.rs

secrets_provider_tests/
env.rs
