mirror of
https://github.com/farion1231/cc-switch.git
synced 2026-04-08 07:03:22 +08:00
67aa275599
This commit refactors all tests to work with the new database-based architecture, replacing the previous JSON config approach. Key changes: - Add Database export to lib.rs for test access - Create test helper functions in support.rs: - create_test_state(): Creates empty test state with fresh DB - create_test_state_with_config(): Migrates JSON config to DB - Fix environment isolation in provider_service tests: - provider_service_switch_missing_provider_returns_error - provider_service_switch_codex_missing_auth_returns_error - Replace ignored export tests with working alternatives: - export_sql_writes_to_target_path (tests Database::export_sql) - export_sql_returns_error_for_invalid_path (tests error handling) - Update error type matching to align with current implementation All tests now: - Use isolated test environments (test_mutex + reset_test_fs) - Access data via Database API instead of RwLock<MultiAppConfig> - Work with SQLite persistence layer - Pass without environment pollution or race conditions Fixes test compilation errors after database migration.
63 lines
2.1 KiB
Rust
63 lines
2.1 KiB
Rust
use std::path::{Path, PathBuf};
|
||
use std::sync::{Arc, Mutex, OnceLock};
|
||
|
||
use cc_switch_lib::{update_settings, AppSettings, AppState, Database, MultiAppConfig};
|
||
|
||
/// 为测试设置隔离的 HOME 目录，避免污染真实用户数据。
|
||
pub fn ensure_test_home() -> &'static Path {
|
||
static HOME: OnceLock<PathBuf> = OnceLock::new();
|
||
HOME.get_or_init(|| {
|
||
let base = std::env::temp_dir().join("cc-switch-test-home");
|
||
if base.exists() {
|
||
let _ = std::fs::remove_dir_all(&base);
|
||
}
|
||
std::fs::create_dir_all(&base).expect("create test home");
|
||
std::env::set_var("HOME", &base);
|
||
#[cfg(windows)]
|
||
std::env::set_var("USERPROFILE", &base);
|
||
base
|
||
})
|
||
.as_path()
|
||
}
|
||
|
||
/// 清理测试目录中生成的配置文件与缓存。
|
||
pub fn reset_test_fs() {
|
||
let home = ensure_test_home();
|
||
for sub in [".claude", ".codex", ".cc-switch", ".gemini"] {
|
||
let path = home.join(sub);
|
||
if path.exists() {
|
||
if let Err(err) = std::fs::remove_dir_all(&path) {
|
||
eprintln!("failed to clean {}: {}", path.display(), err);
|
||
}
|
||
}
|
||
}
|
||
let claude_json = home.join(".claude.json");
|
||
if claude_json.exists() {
|
||
let _ = std::fs::remove_file(&claude_json);
|
||
}
|
||
|
||
// 重置内存中的设置缓存，确保测试环境不受上一次调用影响
|
||
let _ = update_settings(AppSettings::default());
|
||
}
|
||
|
||
/// 全局互斥锁，避免多测试并发写入相同的 HOME 目录。
|
||
pub fn test_mutex() -> &'static Mutex<()> {
|
||
static MUTEX: OnceLock<Mutex<()>> = OnceLock::new();
|
||
MUTEX.get_or_init(|| Mutex::new(()))
|
||
}
|
||
|
||
/// 创建测试用的 AppState，包含一个空的数据库
|
||
pub fn create_test_state() -> Result<AppState, Box<dyn std::error::Error>> {
|
||
let db = Database::init()?;
|
||
Ok(AppState { db: Arc::new(db) })
|
||
}
|
||
|
||
/// 创建测试用的 AppState，并从 MultiAppConfig 迁移数据
|
||
pub fn create_test_state_with_config(
|
||
config: &MultiAppConfig,
|
||
) -> Result<AppState, Box<dyn std::error::Error>> {
|
||
let db = Database::init()?;
|
||
db.migrate_from_json(config)?;
|
||
Ok(AppState { db: Arc::new(db) })
|
||
}
|