axond 0.3.3

Axond — a stateless, single-binary, self-hosted AI gateway: one place for provider keys, model routing, usage, and telemetry.
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

//! Datastore DSNs for the tests that need a real Redis or Postgres.
//!
//! Absent configuration skips those tests, so the suite stays runnable with no
//! datastore. `AXOND_TEST_REQUIRE_SERVICES=1` inverts that: a missing variable
//! panics, so CI cannot report a green run for tests that never executed.

fn resolve(variable: &str, value: Option<String>, require_services: bool) -> Option<String> {
    match value {
        Some(value) => Some(value),
        None if require_services => panic!(
            "{variable} is required when AXOND_TEST_REQUIRE_SERVICES=1; \
             CI requires the test service to be available"
        ),
        None => None,
    }
}

pub(crate) fn redis_url() -> Option<String> {
    resolve(
        "AXOND_TEST_REDIS_URL",
        std::env::var("AXOND_TEST_REDIS_URL").ok(),
        std::env::var("AXOND_TEST_REQUIRE_SERVICES").as_deref() == Ok("1"),
    )
}

pub(crate) fn postgres_dsn() -> Option<String> {
    resolve(
        "AXOND_TEST_POSTGRES_DSN",
        std::env::var("AXOND_TEST_POSTGRES_DSN").ok(),
        std::env::var("AXOND_TEST_REQUIRE_SERVICES").as_deref() == Ok("1"),
    )
}

#[cfg(test)]
mod tests {
    use super::resolve;

    #[test]
    fn missing_service_is_optional_by_default() {
        assert_eq!(resolve("SERVICE_URL", None, false), None);
    }

    #[test]
    fn configured_service_is_returned_without_requirement() {
        assert_eq!(
            resolve("SERVICE_URL", Some("http://localhost".to_owned()), false),
            Some("http://localhost".to_owned())
        );
    }

    #[test]
    #[should_panic(expected = "SERVICE_URL is required")]
    fn missing_required_service_panics() {
        resolve("SERVICE_URL", None, true);
    }
}