zeph-core 0.18.6

// SPDX-FileCopyrightText: 2026 Andrei G <bug-ops>
// SPDX-License-Identifier: MIT OR Apache-2.0

// std::env::set_var / remove_var are unsafe in Rust 2024 edition; all callers are #[serial].
#![allow(unsafe_code)]
#![allow(clippy::default_trait_access)]

use std::path::{Path, PathBuf};

use super::*;
use crate::config::{Config, ProviderEntry, ProviderKind};
use zeph_llm::claude::ClaudeProvider;
use zeph_llm::ollama::OllamaProvider;

#[test]
fn vault_args_defaults_in_test_context() {
    let config = Config::load(Path::new("/nonexistent")).unwrap();
    let args = parse_vault_args(&config, None, None, None);
    assert_eq!(args.backend, "env");
    assert!(args.key_path.is_none());
    assert!(args.vault_path.is_none());
}

#[test]
fn vault_args_uses_config_backend_as_fallback() {
    let mut config = Config::load(Path::new("/nonexistent")).unwrap();
    config.vault.backend = "age".into();
    let args = parse_vault_args(&config, None, None, None);
    assert_eq!(args.backend, "age");
}

#[test]
fn vault_args_env_overrides_config() {
    let mut config = Config::load(Path::new("/nonexistent")).unwrap();
    config.vault.backend = "age".into();
    unsafe { std::env::set_var("ZEPH_VAULT_BACKEND", "env") };
    let args = parse_vault_args(&config, None, None, None);
    unsafe { std::env::remove_var("ZEPH_VAULT_BACKEND") };
    assert_eq!(args.backend, "env");
}

#[test]
fn vault_args_struct_construction() {
    let args = VaultArgs {
        backend: "age".into(),
        key_path: Some("/tmp/key".into()),
        vault_path: Some("/tmp/vault".into()),
    };
    assert_eq!(args.backend, "age");
    assert_eq!(args.key_path.as_deref(), Some("/tmp/key"));
    assert_eq!(args.vault_path.as_deref(), Some("/tmp/vault"));
}

#[test]
fn vault_args_cli_overrides_env_and_config() {
    let mut config = Config::load(Path::new("/nonexistent")).unwrap();
    config.vault.backend = "env".into();
    unsafe { std::env::set_var("ZEPH_VAULT_BACKEND", "env") };
    let args = parse_vault_args(
        &config,
        Some("age"),
        Some(Path::new("/cli/key")),
        Some(Path::new("/cli/vault")),
    );
    unsafe { std::env::remove_var("ZEPH_VAULT_BACKEND") };
    assert_eq!(args.backend, "age");
    assert_eq!(args.key_path.as_deref(), Some("/cli/key"));
    assert_eq!(args.vault_path.as_deref(), Some("/cli/vault"));
}

#[test]
fn vault_args_env_key_and_path_fallback() {
    let config = Config::load(Path::new("/nonexistent")).unwrap();
    unsafe { std::env::set_var("ZEPH_VAULT_KEY", "/env/key") };
    unsafe { std::env::set_var("ZEPH_VAULT_PATH", "/env/vault") };
    let args = parse_vault_args(&config, None, None, None);
    unsafe { std::env::remove_var("ZEPH_VAULT_KEY") };
    unsafe { std::env::remove_var("ZEPH_VAULT_PATH") };
    assert_eq!(args.key_path.as_deref(), Some("/env/key"));
    assert_eq!(args.vault_path.as_deref(), Some("/env/vault"));
}

#[test]
fn resolve_config_path_cli_override() {
    let path = resolve_config_path(Some(Path::new("/custom/config.toml")));
    assert_eq!(path, PathBuf::from("/custom/config.toml"));
}

#[test]
fn resolve_config_path_default() {
    let path = resolve_config_path(None);
    // Without ZEPH_CONFIG env, falls back to default
    if std::env::var("ZEPH_CONFIG").is_err() {
        assert_eq!(path, PathBuf::from("config/default.toml"));
    }
}

#[test]
fn vault_args_struct_env_backend() {
    let args = VaultArgs {
        backend: "env".into(),
        key_path: None,
        vault_path: None,
    };
    assert_eq!(args.backend, "env");
    assert!(args.key_path.is_none());
    assert!(args.vault_path.is_none());
}

#[test]
fn create_provider_ollama() {
    let config = Config::load(Path::new("/nonexistent")).unwrap();
    let provider = create_provider(&config).unwrap();
    assert!(matches!(provider, AnyProvider::Ollama(_)));
    assert_eq!(provider.name(), "ollama");
}

#[test]
fn create_provider_claude_without_api_key_errors() {
    let mut config = Config::load(Path::new("/nonexistent")).unwrap();
    config.llm.providers = vec![ProviderEntry {
        provider_type: ProviderKind::Claude,
        model: Some("claude-sonnet-4-6".into()),
        max_tokens: Some(4096),
        ..ProviderEntry::default()
    }];
    config.secrets.claude_api_key = None;

    let result = create_provider(&config);
    assert!(result.is_err());
    assert!(
        result
            .unwrap_err()
            .to_string()
            .contains("ZEPH_CLAUDE_API_KEY not found")
    );
}

#[tokio::test]
async fn health_check_ollama_unreachable() {
    let provider = AnyProvider::Ollama(OllamaProvider::new(
        "http://127.0.0.1:1",
        "test".into(),
        "embed".into(),
    ));
    health_check(&provider).await;
}

#[tokio::test]
async fn health_check_claude_noop() {
    let provider = AnyProvider::Claude(ClaudeProvider::new("key".into(), "model".into(), 1024));
    health_check(&provider).await;
}

#[test]
fn effective_embedding_model_defaults_to_llm() {
    let config = Config::load(Path::new("/nonexistent")).unwrap();
    assert_eq!(effective_embedding_model(&config), "qwen3-embedding");
}

#[test]
fn effective_embedding_model_uses_pool_embed_entry() {
    let mut config = Config::load(Path::new("/nonexistent")).unwrap();
    config.llm.providers = vec![ProviderEntry {
        provider_type: ProviderKind::OpenAi,
        model: Some("gpt-5.2".into()),
        max_tokens: Some(4096),
        embedding_model: Some("text-embedding-3-small".into()),
        embed: true,
        ..ProviderEntry::default()
    }];
    assert_eq!(effective_embedding_model(&config), "text-embedding-3-small");
}

#[test]
fn effective_embedding_model_falls_back_when_embed_missing() {
    let mut config = Config::load(Path::new("/nonexistent")).unwrap();
    config.llm.providers = vec![ProviderEntry {
        provider_type: ProviderKind::OpenAi,
        model: Some("gpt-5.2".into()),
        max_tokens: Some(4096),
        embedding_model: None,
        ..ProviderEntry::default()
    }];
    assert_eq!(effective_embedding_model(&config), "qwen3-embedding");
}

#[test]
fn create_provider_openai_missing_api_key_errors() {
    let mut config = Config::load(Path::new("/nonexistent")).unwrap();
    config.llm.providers = vec![ProviderEntry {
        provider_type: ProviderKind::OpenAi,
        base_url: Some("https://api.openai.com/v1".into()),
        model: Some("gpt-4o".into()),
        max_tokens: Some(4096),
        embedding_model: None,
        ..ProviderEntry::default()
    }];
    config.secrets.openai_api_key = None;
    let result = create_provider(&config);
    assert!(result.is_err());
    assert!(
        result
            .unwrap_err()
            .to_string()
            .contains("ZEPH_OPENAI_API_KEY not found")
    );
}

#[cfg(feature = "candle")]
#[test]
fn select_device_cpu_default() {
    let device = select_device("cpu").unwrap();
    assert!(matches!(device, zeph_llm::candle_provider::Device::Cpu));
}

#[cfg(feature = "candle")]
#[test]
fn select_device_unknown_defaults_to_cpu() {
    let device = select_device("unknown").unwrap();
    assert!(matches!(device, zeph_llm::candle_provider::Device::Cpu));
}

#[cfg(all(feature = "candle", not(feature = "metal")))]
#[test]
fn select_device_metal_without_feature_errors() {
    let result = select_device("metal");
    assert!(result.is_err());
    assert!(result.unwrap_err().to_string().contains("metal feature"));
}

#[cfg(all(feature = "candle", not(feature = "cuda")))]
#[test]
fn select_device_cuda_without_feature_errors() {
    let result = select_device("cuda");
    assert!(result.is_err());
    assert!(result.unwrap_err().to_string().contains("cuda feature"));
}

#[cfg(feature = "candle")]
#[test]
fn select_device_auto_fallback() {
    let device = select_device("auto").unwrap();
    assert!(matches!(
        device,
        zeph_llm::candle_provider::Device::Cpu
            | zeph_llm::candle_provider::Device::Cuda(_)
            | zeph_llm::candle_provider::Device::Metal(_)
    ));
}

#[cfg(feature = "candle")]
#[test]
fn create_provider_candle_without_config_errors() {
    let mut config = Config::load(Path::new("/nonexistent")).unwrap();
    config.llm.providers = vec![ProviderEntry {
        provider_type: ProviderKind::Candle,
        ..ProviderEntry::default()
    }];
    config.llm.candle = None;
    let result = create_provider(&config);
    assert!(result.is_err());
    assert!(
        result
            .unwrap_err()
            .to_string()
            .contains("candle provider requires 'candle' section in [[llm.providers]]")
    );
}

#[cfg(feature = "candle")]
#[tokio::test]
async fn health_check_candle_logs_device() {
    use zeph_llm::candle_provider::CandleProvider;

    let source = zeph_llm::candle_provider::loader::ModelSource::HuggingFace {
        repo_id: "test/model".to_string(),
        filename: Some("model.gguf".to_string()),
    };
    let template = zeph_llm::candle_provider::template::ChatTemplate::parse_str(
        "{{ bos_token }}{{ messages[0].content }}",
    );
    let gen_config = zeph_llm::candle_provider::generate::GenerationConfig {
        temperature: 0.7,
        top_p: Some(0.9),
        top_k: Some(50),
        max_tokens: 512,
        seed: 42,
        repeat_penalty: 1.1,
        repeat_last_n: 64,
    };
    let device = zeph_llm::candle_provider::Device::Cpu;

    let candle_result = CandleProvider::new(
        &source,
        template,
        gen_config,
        Some("embed/model"),
        None,
        device,
    );

    if let Ok(candle) = candle_result {
        let provider = AnyProvider::Candle(candle);
        health_check(&provider).await;
    }
}

#[test]
fn create_mcp_manager_with_http_transport() {
    use std::collections::HashMap;

    let mut config = Config::load(Path::new("/nonexistent")).unwrap();
    config.mcp.servers = vec![crate::config::McpServerConfig {
        id: "test".into(),
        url: Some("http://localhost:3000".into()),
        command: None,
        args: vec![],
        env: HashMap::new(),
        headers: HashMap::new(),
        oauth: None,
        timeout: 30,
        policy: Default::default(),
        trust_level: Default::default(),
        tool_allowlist: None,
        expected_tools: vec![],
        roots: vec![],
        tool_metadata: HashMap::new(),
        elicitation_enabled: None,
        env_isolation: None,
    }];

    let manager = create_mcp_manager(&config, false);
    let debug = format!("{manager:?}");
    assert!(debug.contains("server_count: 1"));
}

#[test]
fn create_mcp_manager_with_stdio_transport() {
    use std::collections::HashMap;

    let mut config = Config::load(Path::new("/nonexistent")).unwrap();
    config.mcp.servers = vec![crate::config::McpServerConfig {
        id: "test".into(),
        url: None,
        command: Some("node".into()),
        args: vec!["server.js".into()],
        env: HashMap::new(),
        headers: HashMap::new(),
        oauth: None,
        timeout: 30,
        policy: Default::default(),
        trust_level: Default::default(),
        tool_allowlist: None,
        expected_tools: vec![],
        roots: vec![],
        tool_metadata: HashMap::new(),
        elicitation_enabled: None,
        env_isolation: None,
    }];

    let manager = create_mcp_manager(&config, false);
    let debug = format!("{manager:?}");
    assert!(debug.contains("server_count: 1"));
}

#[test]
fn create_mcp_manager_empty_servers() {
    let mut config = Config::load(Path::new("/nonexistent")).unwrap();
    config.mcp.servers = vec![];

    let manager = create_mcp_manager(&config, false);
    let debug = format!("{manager:?}");
    assert!(debug.contains("server_count: 0"));
}

#[tokio::test]
async fn create_mcp_registry_when_semantic_disabled() {
    let config_path = Path::new("/nonexistent");
    let mut config = Config::load(config_path).unwrap();
    config.memory.semantic.enabled = false;

    let provider = AnyProvider::Ollama(OllamaProvider::new(
        "http://localhost:11434",
        "test".into(),
        "embed".into(),
    ));

    let mcp_tools = vec![];
    let registry = create_mcp_registry(&config, &provider, &mcp_tools, "test-model", None).await;
    assert!(registry.is_none());
}

#[test]
fn managed_skills_dir_returns_skills_subdir() {
    let dir = managed_skills_dir();
    assert!(
        dir.ends_with("skills"),
        "managed_skills_dir should end in 'skills', got: {dir:?}"
    );
}

#[test]
fn app_builder_managed_skills_dir_matches_free_fn() {
    assert_eq!(AppBuilder::managed_skills_dir(), managed_skills_dir());
}

#[test]
fn skill_paths_includes_managed_dir() {
    let config = Config::load(Path::new("/nonexistent")).unwrap();
    let builder = AppBuilder {
        config,
        config_path: PathBuf::from("/nonexistent/config.toml"),
        vault: Box::new(EnvVaultProvider),
        age_vault: None,
        qdrant_ops: None,
    };
    let paths = builder.skill_paths();
    let managed = managed_skills_dir();
    assert!(
        paths.contains(&managed),
        "skill_paths() should include managed_skills_dir, got: {paths:?}"
    );
}

#[test]
fn skill_paths_does_not_duplicate_managed_dir() {
    let managed = managed_skills_dir();
    let mut config = Config::load(Path::new("/nonexistent")).unwrap();
    config.skills.paths = vec![managed.to_string_lossy().into_owned()];
    let builder = AppBuilder {
        config,
        config_path: PathBuf::from("/nonexistent/config.toml"),
        vault: Box::new(EnvVaultProvider),
        age_vault: None,
        qdrant_ops: None,
    };
    let paths = builder.skill_paths();
    let count = paths.iter().filter(|p| p == &&managed).count();
    assert_eq!(
        count, 1,
        "managed dir should appear exactly once, got: {paths:?}"
    );
}

#[tokio::test]
async fn create_skill_matcher_when_semantic_disabled() {
    let tmp = std::env::temp_dir().join("zeph_test_skill_matcher_bootstrap.db");
    let _ = std::fs::remove_file(&tmp);
    let tmp_path = tmp.to_string_lossy().to_string();

    let mut config = Config::load(Path::new("/nonexistent")).unwrap();
    config.memory.semantic.enabled = false;
    config.memory.sqlite_path = tmp_path.clone();

    let provider = AnyProvider::Ollama(OllamaProvider::new(
        "http://localhost:11434",
        "test".into(),
        "embed".into(),
    ));

    let memory = SemanticMemory::with_sqlite_backend_and_pool_size(
        &tmp_path,
        provider.clone(),
        &config.llm.embedding_model,
        config.memory.semantic.vector_weight,
        config.memory.semantic.keyword_weight,
        1,
    )
    .await
    .unwrap();

    let meta: Vec<&SkillMeta> = vec![];
    let result = create_skill_matcher(&config, &provider, &meta, &memory, "test-model", None).await;
    assert!(result.is_none());

    let _ = std::fs::remove_file(&tmp);
}

#[test]
fn appbuilder_qdrant_ops_invalid_url_returns_err() {
    let mut config = Config::load(Path::new("/nonexistent")).unwrap();
    config.memory.vector_backend = crate::config::VectorBackend::Qdrant;
    config.memory.qdrant_url = "not a valid url".into();

    let result = zeph_memory::QdrantOps::new(&config.memory.qdrant_url);
    assert!(
        result.is_err(),
        "QdrantOps::new with invalid URL must fail (CRIT-04)"
    );
}

#[test]
fn appbuilder_qdrant_ops_valid_url_succeeds() {
    let result = zeph_memory::QdrantOps::new("http://localhost:6334");
    assert!(result.is_ok(), "QdrantOps::new with valid URL must succeed");
}

// ── build_feedback_classifier ─────────────────────────────────────────────────
//
// Full integration tests require a live provider (Ollama/OpenAI). These tests only verify
// the early-return behavior when `detector_mode != Model` — no LLM call is made.

fn make_builder_with_detector_mode(mode: crate::config::DetectorMode) -> AppBuilder {
    let mut config = Config::load(Path::new("/nonexistent")).unwrap();
    config.skills.learning.detector_mode = mode;
    AppBuilder {
        config,
        config_path: PathBuf::from("/nonexistent/config.toml"),
        vault: Box::new(EnvVaultProvider),
        age_vault: None,
        qdrant_ops: None,
    }
}

fn make_mock_primary() -> zeph_llm::any::AnyProvider {
    zeph_llm::any::AnyProvider::Ollama(zeph_llm::ollama::OllamaProvider::new(
        "http://localhost:11434",
        "llama3".into(),
        String::new(),
    ))
}

#[test]
fn build_feedback_classifier_regex_mode_returns_none() {
    let b = make_builder_with_detector_mode(crate::config::DetectorMode::Regex);
    let primary = make_mock_primary();
    let result = b.build_feedback_classifier(&primary);
    assert!(
        result.is_none(),
        "Regex mode must not build feedback classifier"
    );
}

#[test]
fn build_feedback_classifier_judge_mode_returns_none() {
    let b = make_builder_with_detector_mode(crate::config::DetectorMode::Judge);
    let primary = make_mock_primary();
    let result = b.build_feedback_classifier(&primary);
    assert!(
        result.is_none(),
        "Judge mode must not build feedback classifier"
    );
}

// ── create_embedding_provider ─────────────────────────────────────────────────

#[test]
fn create_embedding_provider_prefers_embed_flag() {
    let mut config = Config::load(Path::new("/nonexistent")).unwrap();
    // Two providers: first is Ollama (primary), second is OpenAI with embed=true.
    config.llm.providers = vec![
        ProviderEntry {
            provider_type: ProviderKind::Ollama,
            model: Some("qwen3:8b".into()),
            embedding_model: Some("nomic-embed-text".into()),
            embed: false,
            ..ProviderEntry::default()
        },
        ProviderEntry {
            provider_type: ProviderKind::Ollama,
            name: Some("embed".into()),
            model: Some("qwen3:8b".into()),
            embedding_model: Some("qwen3-embedding".into()),
            embed: true,
            ..ProviderEntry::default()
        },
    ];
    let primary = AnyProvider::Ollama(OllamaProvider::new(
        "http://localhost:11434",
        "qwen3:8b".into(),
        "nomic-embed-text".into(),
    ));
    let embed_provider = create_embedding_provider(&config, &primary);
    // Should resolve to an Ollama provider (the embed=true entry).
    assert!(matches!(embed_provider, AnyProvider::Ollama(_)));
}

#[test]
fn create_embedding_provider_falls_back_to_embedding_model_entry() {
    let mut config = Config::load(Path::new("/nonexistent")).unwrap();
    // Only one provider with embedding_model set but embed=false.
    config.llm.providers = vec![ProviderEntry {
        provider_type: ProviderKind::Ollama,
        name: Some("main".into()),
        model: Some("qwen3:8b".into()),
        embedding_model: Some("nomic-embed-text".into()),
        embed: false,
        ..ProviderEntry::default()
    }];
    let primary = AnyProvider::Ollama(OllamaProvider::new(
        "http://localhost:11434",
        "qwen3:8b".into(),
        "nomic-embed-text".into(),
    ));
    let embed_provider = create_embedding_provider(&config, &primary);
    assert!(matches!(embed_provider, AnyProvider::Ollama(_)));
}

#[test]
fn create_embedding_provider_falls_back_to_primary_when_no_embedding_entry() {
    let mut config = Config::load(Path::new("/nonexistent")).unwrap();
    // Provider with no embedding_model and embed=false.
    config.llm.providers = vec![ProviderEntry {
        provider_type: ProviderKind::Ollama,
        model: Some("qwen3:8b".into()),
        embedding_model: None,
        embed: false,
        ..ProviderEntry::default()
    }];
    let primary = AnyProvider::Ollama(OllamaProvider::new(
        "http://localhost:11434",
        "qwen3:8b".into(),
        "nomic-embed-text".into(),
    ));
    let embed_provider = create_embedding_provider(&config, &primary);
    // Falls back to primary clone — must still be Ollama.
    assert!(matches!(embed_provider, AnyProvider::Ollama(_)));
    assert_eq!(embed_provider.name(), primary.name());
}

// ── auto_budget_tokens fallback (#2773) ──────────────────────────────────────

#[test]
fn auto_budget_tokens_returns_configured_value() {
    let mut config = Config::load(Path::new("/nonexistent")).unwrap();
    config.memory.auto_budget = false;
    config.memory.context_budget_tokens = 65536;
    let builder = super::AppBuilder::for_test(config);
    let provider = AnyProvider::Mock(zeph_llm::mock::MockProvider::default());
    assert_eq!(builder.auto_budget_tokens(&provider), 65536);
}

#[test]
fn auto_budget_tokens_auto_budget_no_window_falls_back_to_128k() {
    let mut config = Config::load(Path::new("/nonexistent")).unwrap();
    config.memory.auto_budget = true;
    config.memory.context_budget_tokens = 0;
    let builder = super::AppBuilder::for_test(config);
    // MockProvider::context_window() returns None — triggers fallback
    let provider = AnyProvider::Mock(zeph_llm::mock::MockProvider::default());
    assert_eq!(builder.auto_budget_tokens(&provider), 128_000);
}

#[test]
fn auto_budget_tokens_explicit_zero_falls_back_to_128k() {
    let mut config = Config::load(Path::new("/nonexistent")).unwrap();
    config.memory.auto_budget = false;
    config.memory.context_budget_tokens = 0;
    let builder = super::AppBuilder::for_test(config);
    let provider = AnyProvider::Mock(zeph_llm::mock::MockProvider::default());
    assert_eq!(builder.auto_budget_tokens(&provider), 128_000);
}