mirror of
https://github.com/BerriAI/litellm.git
synced 2026-10-31 06:56:59 +00:00
91d4db7d44
The workspace moves to reqwest 0.13, so litellm-http no longer carries a second reqwest just for rmcp. The mcp feature, its pool and its proxy conversion go away, and the gateway hands rmcp a pooled NoRedirect client that already carries the TLS, proxy and timeout settings The Client test helpers collapse into for_test, a no-redirect client without TLS or proxy settings, and wrap_for_test for the few tests that need a hand-built reqwest client Co-Authored-By: Claude Opus 5.5 <noreply@anthropic.com>
62 lines
2.1 KiB
Rust
62 lines
2.1 KiB
Rust
#[cfg(test)]
|
|
mod tests {
|
|
use rstest::rstest;
|
|
|
|
use litellm_secrets::source::{EnvironmentSecrets, SecretSource};
|
|
|
|
#[rstest]
|
|
#[case::lowercase_true("LITELLM_ENVIRONMENT_SECRETS_TRUE", "true", None)]
|
|
#[case::padded_false("LITELLM_ENVIRONMENT_SECRETS_FALSE", " FALSE ", None)]
|
|
#[case::text("LITELLM_ENVIRONMENT_SECRETS_TEXT", "secret", Some("secret"))]
|
|
#[tokio::test]
|
|
async fn python_environment_values_are_absent_like_get_secret_str(
|
|
#[case] name: &'static str,
|
|
#[case] value: &str,
|
|
#[case] expected: Option<&str>,
|
|
) {
|
|
unsafe { std::env::set_var(name, value) };
|
|
let secret = EnvironmentSecrets::python_compatible(litellm_http::Client::for_test())
|
|
.resolve(&[name])
|
|
.await
|
|
.unwrap()
|
|
.get(name);
|
|
unsafe { std::env::remove_var(name) };
|
|
assert_eq!(secret.as_deref(), expected);
|
|
}
|
|
}
|
|
|
|
#[tokio::test]
|
|
async fn dynamic_names_use_the_same_resolver_and_snapshots_never_do_fresh_lookups() {
|
|
use litellm_secrets::source::SecretSource;
|
|
use litellm_secrets::{OidcResolver, SecretManagerState, SecretResolver};
|
|
use std::sync::{
|
|
Arc,
|
|
atomic::{AtomicUsize, Ordering},
|
|
};
|
|
|
|
let calls = Arc::new(AtomicUsize::new(0));
|
|
let reads = calls.clone();
|
|
let source = SecretResolver::new(
|
|
Arc::new(SecretManagerState::default()),
|
|
Arc::new(move |name: &str| {
|
|
reads.fetch_add(1, Ordering::SeqCst);
|
|
(name != "missing").then(|| name.to_owned())
|
|
}),
|
|
OidcResolver::new(litellm_http::Client::for_test()),
|
|
);
|
|
let snapshot = source.resolve(&["declared", "missing"]).await.unwrap();
|
|
let name = format!("runtime-{}", "key");
|
|
assert_eq!(snapshot.get("declared").as_deref(), Some("declared"));
|
|
assert_eq!(snapshot.get("missing"), None);
|
|
assert_eq!(snapshot.get(&name), None);
|
|
assert_eq!(calls.load(Ordering::SeqCst), 2);
|
|
assert_eq!(
|
|
SecretSource::get_secret_str(&source, &name)
|
|
.await
|
|
.unwrap()
|
|
.unwrap()
|
|
.expose(),
|
|
name
|
|
);
|
|
assert_eq!(calls.load(Ordering::SeqCst), 3);
|
|
}
|