test: add unit tests for Novita provider defaults, alias, and driver creation

- test_provider_defaults_novita: verify base_url, api_key_env, key_required
- test_provider_defaults_novita_ai_alias: verify "novita-ai" alias resolves
- test_novita_provider_with_env_key: driver creation succeeds with env key
- test_novita_provider_no_key_errors: driver creation fails without key

Co-Authored-By: Claude Opus 4.6 <noreply@anthropic.com>
This commit is contained in:
Alex-wuhu
2026-03-31 14:26:49 +08:00
co-authored by Claude Opus 4.6
parent 365bec868c
commit 4714efb9e4
@@ -744,6 +744,52 @@ mod tests {
assert!(d.key_required);
}
#[test]
fn test_provider_defaults_novita() {
let d = provider_defaults("novita").unwrap();
assert_eq!(d.base_url, "https://api.novita.ai/openai/v1");
assert_eq!(d.api_key_env, "NOVITA_API_KEY");
assert!(d.key_required);
}
#[test]
fn test_provider_defaults_novita_ai_alias() {
let d = provider_defaults("novita-ai").unwrap();
assert_eq!(d.base_url, "https://api.novita.ai/openai/v1");
assert_eq!(d.api_key_env, "NOVITA_API_KEY");
assert!(d.key_required);
}
#[test]
fn test_novita_provider_with_env_key() {
let unique_key = "test-novita-key-12345";
std::env::set_var("NOVITA_API_KEY", unique_key);
let config = DriverConfig {
provider: "novita".to_string(),
api_key: None,
base_url: None,
skip_permissions: true,
};
let driver = create_driver(&config);
assert!(
driver.is_ok(),
"Novita provider with env var should succeed"
);
std::env::remove_var("NOVITA_API_KEY");
}
#[test]
fn test_novita_provider_no_key_errors() {
let config = DriverConfig {
provider: "novita".to_string(),
api_key: None,
base_url: None,
skip_permissions: true,
};
let driver = create_driver(&config);
assert!(driver.is_err());
}
#[test]
fn test_nvidia_provider_with_env_key() {
// NVIDIA NIM is a known provider — set API key and verify driver creation succeeds.