EasyCLIProxyAPI/src-tauri/src/tests/support.rs

use super::*;

pub(super) fn agent_test_home(name: &str) -> PathBuf {
    let path = std::env::temp_dir().join(format!(
        "cpa-gui-agent-{name}-{}-{}",
        std::process::id(),
        SystemTime::now()
            .duration_since(UNIX_EPOCH)
            .unwrap()
            .as_nanos()
    ));
    fs::create_dir_all(&path).unwrap();
    path
}

pub(super) fn test_agent_models(names: &[&str]) -> Vec<AgentModelOption> {
    names
        .iter()
        .map(|name| AgentModelOption {
            name: (*name).to_string(),
            alias: None,
            is_alias: false,
            context_window: Some(200_000),
        })
        .collect()
}

pub(super) fn test_codex_models(names: &[&str]) -> String {
    let payload = serde_json::json!({
        "models": names.iter().map(|name| serde_json::json!({ "id": name })).collect::<Vec<_>>()
    });
    let runtime = codex_catalog::parse_runtime_models(&payload).unwrap();
    codex_catalog::prepare_catalog(&runtime).unwrap().json
}

pub(super) fn test_codex_oauth_thinking_source(model: &str) -> ResolvedThinkingAliasSource {
    ResolvedThinkingAliasSource {
        source: ThinkingAliasSource {
            id: format!("codex-oauth:{model}"),
            model: model.to_string(),
            display_name: None,
            provider: "Codex OAuth".to_string(),
            kind: "codex-oauth".to_string(),
            protocol: "codex".to_string(),
        },
        location: ThinkingAliasSourceLocation::Oauth {
            channel: "codex",
            force_mapping: false,
        },
    }
}