diff --git a/src-tauri/tests/deeplink_import.rs b/src-tauri/tests/deeplink_import.rs index d70a06019..2768f8953 100644 --- a/src-tauri/tests/deeplink_import.rs +++ b/src-tauri/tests/deeplink_import.rs @@ -1,46 +1,36 @@ -use std::sync::RwLock; +use std::sync::Arc; -use cc_switch_lib::{ - import_provider_from_deeplink, parse_deeplink_url, AppState, AppType, MultiAppConfig, -}; +use cc_switch_lib::{import_provider_from_deeplink, parse_deeplink_url, AppState, Database}; #[path = "support.rs"] mod support; use support::{ensure_test_home, reset_test_fs, test_mutex}; #[test] -fn deeplink_import_claude_provider_persists_to_config() { +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 _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"; + 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 mut config = MultiAppConfig::default(); - config.ensure_app(&AppType::Claude); + let db = Arc::new(Database::memory().expect("create memory db")); - let state = AppState { - config: RwLock::new(config), - }; + let state = AppState { db: db.clone() }; let provider_id = import_provider_from_deeplink(&state, request.clone()) .expect("import provider from deeplink"); - // 验证内存状态 - let guard = state.config.read().expect("read config"); - let manager = guard - .get_manager(&AppType::Claude) - .expect("claude manager should exist"); - let provider = manager - .providers + // 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); - assert_eq!( - provider.website_url.as_deref(), - Some(request.homepage.as_str()) - ); + + 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") @@ -49,50 +39,34 @@ fn deeplink_import_claude_provider_persists_to_config() { .settings_config .pointer("/env/ANTHROPIC_BASE_URL") .and_then(|v| v.as_str()); - assert_eq!(auth_token, Some(request.api_key.as_str())); - assert_eq!(base_url, Some(request.endpoint.as_str())); - drop(guard); - - // 验证配置已持久化 - let config_path = home.join(".cc-switch").join("config.json"); - assert!( - config_path.exists(), - "importing provider from deeplink should persist config.json" - ); + 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 _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"; + 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 mut config = MultiAppConfig::default(); - config.ensure_app(&AppType::Codex); + let db = Arc::new(Database::memory().expect("create memory db")); - let state = AppState { - config: RwLock::new(config), - }; + let state = AppState { db: db.clone() }; let provider_id = import_provider_from_deeplink(&state, request.clone()) .expect("import provider from deeplink"); - let guard = state.config.read().expect("read config"); - let manager = guard - .get_manager(&AppType::Codex) - .expect("codex manager should exist"); - let provider = manager - .providers + 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); - assert_eq!( - provider.website_url.as_deref(), - Some(request.homepage.as_str()) - ); + + 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") @@ -102,20 +76,13 @@ fn deeplink_import_codex_provider_builds_auth_and_config() { .get("config") .and_then(|v| v.as_str()) .unwrap_or_default(); - assert_eq!(auth_value, Some(request.api_key.as_str())); + assert_eq!(auth_value, request.api_key.as_deref()); assert!( - config_text.contains(request.endpoint.as_str()), + 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" ); - drop(guard); - - let config_path = home.join(".cc-switch").join("config.json"); - assert!( - config_path.exists(), - "importing provider from deeplink should persist config.json" - ); }