mirror of
https://github.com/farion1231/cc-switch.git
synced 2026-04-10 05:00:17 +08:00
1e3a978ecb
The previous cleanup logic only sent a shutdown signal but didn't wait for the proxy server to actually stop. This caused a race condition where the app would exit before cleanup completed, leaving Live configs in an inconsistent state. Changes: - Add `server_handle` field to ProxyServer to track the spawned task - Modify `stop()` to wait for server task completion (5s timeout) - Add 100ms delay before process exit to ensure I/O flush - Export ProxyService and fix test files that were missing proxy_service field
89 lines
3.5 KiB
Rust
89 lines
3.5 KiB
Rust
use std::sync::Arc;
|
|
|
|
use cc_switch_lib::{import_provider_from_deeplink, parse_deeplink_url, AppState, Database, ProxyService};
|
|
|
|
#[path = "support.rs"]
|
|
mod support;
|
|
use support::{ensure_test_home, reset_test_fs, test_mutex};
|
|
|
|
#[test]
|
|
fn deeplink_import_claude_provider_persists_to_db() {
|
|
let _guard = test_mutex().lock().expect("acquire test mutex");
|
|
reset_test_fs();
|
|
let _home = ensure_test_home();
|
|
|
|
let url = "ccswitch://v1/import?resource=provider&app=claude&name=DeepLink%20Claude&homepage=https%3A%2F%2Fexample.com&endpoint=https%3A%2F%2Fapi.example.com%2Fv1&apiKey=sk-test-claude-key&model=claude-sonnet-4&icon=claude";
|
|
let request = parse_deeplink_url(url).expect("parse deeplink url");
|
|
|
|
let db = Arc::new(Database::memory().expect("create memory db"));
|
|
let proxy_service = ProxyService::new(db.clone());
|
|
let state = AppState { db: db.clone(), proxy_service };
|
|
|
|
let provider_id = import_provider_from_deeplink(&state, request.clone())
|
|
.expect("import provider from deeplink");
|
|
|
|
// Verify DB state
|
|
let providers = db.get_all_providers("claude").expect("get providers");
|
|
let provider = providers
|
|
.get(&provider_id)
|
|
.expect("provider created via deeplink");
|
|
|
|
assert_eq!(provider.name, request.name.clone().unwrap());
|
|
assert_eq!(provider.website_url.as_deref(), request.homepage.as_deref());
|
|
assert_eq!(provider.icon.as_deref(), Some("claude"));
|
|
let auth_token = provider
|
|
.settings_config
|
|
.pointer("/env/ANTHROPIC_AUTH_TOKEN")
|
|
.and_then(|v| v.as_str());
|
|
let base_url = provider
|
|
.settings_config
|
|
.pointer("/env/ANTHROPIC_BASE_URL")
|
|
.and_then(|v| v.as_str());
|
|
assert_eq!(auth_token, request.api_key.as_deref());
|
|
assert_eq!(base_url, request.endpoint.as_deref());
|
|
}
|
|
|
|
#[test]
|
|
fn deeplink_import_codex_provider_builds_auth_and_config() {
|
|
let _guard = test_mutex().lock().expect("acquire test mutex");
|
|
reset_test_fs();
|
|
let _home = ensure_test_home();
|
|
|
|
let url = "ccswitch://v1/import?resource=provider&app=codex&name=DeepLink%20Codex&homepage=https%3A%2F%2Fopenai.example&endpoint=https%3A%2F%2Fapi.openai.example%2Fv1&apiKey=sk-test-codex-key&model=gpt-4o&icon=openai";
|
|
let request = parse_deeplink_url(url).expect("parse deeplink url");
|
|
|
|
let db = Arc::new(Database::memory().expect("create memory db"));
|
|
let proxy_service = ProxyService::new(db.clone());
|
|
let state = AppState { db: db.clone(), proxy_service };
|
|
|
|
let provider_id = import_provider_from_deeplink(&state, request.clone())
|
|
.expect("import provider from deeplink");
|
|
|
|
let providers = db.get_all_providers("codex").expect("get providers");
|
|
let provider = providers
|
|
.get(&provider_id)
|
|
.expect("provider created via deeplink");
|
|
|
|
assert_eq!(provider.name, request.name.clone().unwrap());
|
|
assert_eq!(provider.website_url.as_deref(), request.homepage.as_deref());
|
|
assert_eq!(provider.icon.as_deref(), Some("openai"));
|
|
let auth_value = provider
|
|
.settings_config
|
|
.pointer("/auth/OPENAI_API_KEY")
|
|
.and_then(|v| v.as_str());
|
|
let config_text = provider
|
|
.settings_config
|
|
.get("config")
|
|
.and_then(|v| v.as_str())
|
|
.unwrap_or_default();
|
|
assert_eq!(auth_value, request.api_key.as_deref());
|
|
assert!(
|
|
config_text.contains(request.endpoint.as_deref().unwrap()),
|
|
"config.toml content should contain endpoint"
|
|
);
|
|
assert!(
|
|
config_text.contains("model = \"gpt-4o\""),
|
|
"config.toml content should contain model setting"
|
|
);
|
|
}
|