promptforge-gateway 0.1.0

use super::*;

const SAMPLE: &str = r#"
[server]
bind = "127.0.0.1:8081"
key = "t"

[[endpoint]]
id = "anthropic"
protocol = "openai"
base_url = "https://api.anthropic.com/v1"
api_key = ""

[[model]]
name = "m1"
description = "a small test model"
context = 8192
upstream = "u1"
endpoints = ["anthropic"]
"#;

#[test]
fn parses_a_valid_config() {
    let config = Config::from_toml_str(SAMPLE).unwrap();
    assert_eq!(config.endpoints.len(), 1);
    assert_eq!(config.models[0].name, "m1");
    assert_eq!(config.models[0].description, "a small test model");
    assert_eq!(config.models[0].context, 8192);
    assert_eq!(config.models[0].thinking, ThinkingMode::Never);
    assert_eq!(config.models[0].upstream, "u1");
}

#[test]
fn rejects_model_missing_description() {
    let toml = r#"
[server]
bind = "127.0.0.1:8081"
key = "t"

[[endpoint]]
id = "anthropic"
protocol = "openai"
base_url = "http://a"
api_key = ""

[[model]]
name = "m"
context = 8192
upstream = "u"
endpoints = ["anthropic"]
"#;
    assert!(matches!(
        Config::parse_toml(toml),
        Err(ConfigError::Parse { .. })
    ));
}

#[test]
fn rejects_model_missing_context() {
    let toml = r#"
[server]
bind = "127.0.0.1:8081"
key = "t"

[[endpoint]]
id = "anthropic"
protocol = "openai"
base_url = "http://a"
api_key = ""

[[model]]
name = "m"
description = "prose"
upstream = "u"
endpoints = ["anthropic"]
"#;
    assert!(matches!(
        Config::parse_toml(toml),
        Err(ConfigError::Parse { .. })
    ));
}

#[test]
fn rejects_empty_server_key() {
    let toml = r#"
[server]
bind = "127.0.0.1:8081"
key = ""

[[endpoint]]
id = "e"
protocol = "openai"
base_url = "http://a"
api_key = ""

[[model]]
name = "m"
description = "prose"
context = 8192
upstream = "u"
endpoints = ["e"]
"#;
    assert!(matches!(
        Config::parse_toml(toml),
        Err(ConfigError::Validation(_))
    ));
}

#[test]
fn rejects_web_search_default_count_over_max() {
    let toml = r#"
[server]
bind = "127.0.0.1:8081"
key = "t"

[[endpoint]]
id = "e"
protocol = "openai"
base_url = "http://a"
api_key = ""

[[model]]
name = "m"
description = "prose"
context = 8192
upstream = "u"
endpoints = ["e"]

[tools.web_search]
provider = "brave"
api_key = "k"
base_url = "https://api.search.brave.com/res/v1"
default_count = 30
max_count = 10
"#;
    assert!(matches!(
        Config::parse_toml(toml),
        Err(ConfigError::Validation(_))
    ));
}

#[test]
fn rejects_remote_device_with_lanes() {
    let toml = r#"
[server]
bind = "127.0.0.1:8081"
key = "t"

[[device]]
id = "gpu"
type = "remote"

[[device.lane]]
id = "l0"
concurrency = 1

[[endpoint]]
id = "e"
protocol = "openai"
base_url = "http://a"
api_key = ""

[[model]]
name = "m"
description = "prose"
context = 8192
upstream = "u"
endpoints = ["e"]
"#;
    assert!(matches!(
        Config::parse_toml(toml),
        Err(ConfigError::Validation(_))
    ));
}

#[test]
fn rejects_web_search_non_http_base_url() {
    let toml = r#"
[server]
bind = "127.0.0.1:8081"
key = "t"

[[endpoint]]
id = "e"
protocol = "openai"
base_url = "http://a"
api_key = ""

[[model]]
name = "m"
description = "prose"
context = 8192
upstream = "u"
endpoints = ["e"]

[tools.web_search]
provider = "brave"
api_key = "k"
base_url = "ftp://nope"
"#;
    assert!(matches!(
        Config::parse_toml(toml),
        Err(ConfigError::Validation(_))
    ));
}

#[test]
fn rejects_plaintext_http_local_model_source() {
    let sha = "a".repeat(64);
    let toml = format!(
        r#"
[server]
bind = "127.0.0.1:8081"
key = "t"

[[local_model]]
name = "m"
description = "a local model"
source = "http://example.com/m.gguf"
sha256 = "{sha}"
context = 4096
"#
    );
    assert!(matches!(
        Config::parse_toml(&toml),
        Err(ConfigError::Validation(_))
    ));
}

#[test]
fn rejects_unknown_model_key() {
    let toml = r#"
[server]
bind = "127.0.0.1:8081"
key = "t"

[[endpoint]]
id = "anthropic"
protocol = "openai"
base_url = "http://a"
api_key = ""

[[model]]
name = "m"
description = "prose"
context = 8192
upstream = "u"
endpoints = ["anthropic"]
mystery = true
"#;
    assert!(matches!(
        Config::parse_toml(toml),
        Err(ConfigError::Parse { .. })
    ));
}

#[test]
fn parses_thinking_modes() {
    let toml = r#"
[server]
bind = "127.0.0.1:8081"
key = "t"

[[endpoint]]
id = "anthropic"
protocol = "openai"
base_url = "http://a"
api_key = ""

[[model]]
name = "m"
description = "prose"
context = 8192
thinking = "switchable"
upstream = "u"
endpoints = ["anthropic"]
"#;
    let config = Config::from_toml_str(toml).unwrap();
    assert_eq!(config.models[0].thinking, ThinkingMode::Switchable);
}

#[test]
fn interpolates_and_escapes() {
    // SAFETY-free: reading is fine; this test sets no env vars.
    assert_eq!(interpolate("a$$b").unwrap(), "a$b");
    assert_eq!(interpolate("no vars here").unwrap(), "no vars here");
}

#[test]
fn interpolation_ignores_comments_and_keys() {
    // CFG-007: a `${VAR}` inside a comment is not interpolated, so an unset
    // variable there must not fail the load. Raw-text interpolation would have
    // tried to resolve it and errored.
    let toml = r#"
# a comment mentioning ${PROMPTFORGE_DEFINITELY_UNSET_VAR_XYZ}
[server]
bind = "127.0.0.1:8081"
key = "t"

[[endpoint]]
id = "e"
protocol = "openai"
base_url = "http://127.0.0.1:9"
api_key = ""

[[model]]
name = "m"
description = "a $$-priced model"
context = 8192
upstream = "u"
endpoints = ["e"]
"#;
    let config = Config::from_toml_str(toml).expect("comment vars are not interpolated");
    // `$$` in a string value still unescapes to a single `$`.
    assert_eq!(config.models[0].description, "a $-priced model");
}

#[test]
fn unresolved_variable_is_an_error() {
    let missing = "${PROMPTFORGE_DEFINITELY_UNSET_VAR_XYZ}";
    assert!(matches!(
        interpolate(missing),
        Err(ConfigError::UnresolvedVar(_))
    ));
}

#[test]
fn unclosed_interpolation_is_an_error() {
    assert!(matches!(
        interpolate("${OPEN"),
        Err(ConfigError::Interpolation(_))
    ));
}

mod validation;