llguidance 1.7.6

use rstest::*;
use serde_json::{json, Value};

use llg_test_utils::{json_err_test, json_schema_check, NumericBounds};

#[test]
fn null_schema() {
    let schema = &json!({"type":"null"});
    json_schema_check(schema, &json!(null), true);
}

#[rstest]
#[case::boolean(&json!(true))]
#[case::integer(&json!(1))]
#[case::string(&json!("Hello"))]
fn null_schema_failures(#[case] sample_value: &Value) {
    let schema = &json!({"type":"null"});
    json_schema_check(schema, sample_value, false);
}

// ============================================================================

#[rstest]
#[case::bool_false(&json!(false))]
#[case::bool_true(&json!(true))]
fn boolean(#[case] sample_value: &Value) {
    let schema = &json!({"type":"boolean"});
    json_schema_check(schema, sample_value, true);
}

#[rstest]
#[case::int_0(&json!(0))]
#[case::int_1(&json!(1))]
#[case::str_false(&json!("false"))]
#[case::str_true(&json!("true"))]
fn boolean_failures(#[case] sample_value: &Value) {
    let schema = &json!({"type":"boolean"});
    json_schema_check(schema, sample_value, false);
}

// ============================================================================

#[rstest]
#[case::one(&json!(1))]
#[case::minus_1(&json!(-1))]
#[case::zero(&json!(0))]
#[case::large(&json!(10001))]
#[case::negative_large(&json!(-20002))]
fn integer(#[case] sample_value: &Value) {
    let schema = &json!({"type":"integer"});
    json_schema_check(schema, sample_value, true);
}

#[rstest]
#[case::float(&json!(1.0))]
#[case::string_one(&json!("1"))]
#[case::negative_float(&json!(-1.0))]
#[case::string_zero(&json!("0"))]
fn integer_failures(#[case] sample_value: &Value) {
    let schema = &json!({"type":"integer"});
    json_schema_check(schema, sample_value, false);
}

#[rstest]
fn integer_lower_bound(
    #[values(NumericBounds::Inclusive, NumericBounds::Exclusive)] bound_type: NumericBounds,
    #[values(-11, -2, -1, 0, 1, 2, 11)] lower_bound: i64,
) {
    assert!(lower_bound.abs() < 20);
    let iterate_range = 200;
    let schema = match bound_type {
        NumericBounds::Inclusive => {
            json!({"type":"integer", "minimum": lower_bound})
        }
        NumericBounds::Exclusive => {
            json!({"type":"integer", "exclusiveMinimum": lower_bound})
        }
    };
    for i in -iterate_range..=iterate_range {
        let sample_value = json!(i);
        let should_pass = match bound_type {
            NumericBounds::Inclusive => i >= lower_bound,
            NumericBounds::Exclusive => i > lower_bound,
        };
        json_schema_check(&schema, &sample_value, should_pass);
    }
}

#[rstest]
fn integer_upper_bound(
    #[values(NumericBounds::Inclusive, NumericBounds::Exclusive)] bound_type: NumericBounds,
    #[values(-11, -2, -1, 0, 1, 2, 11)] upper_bound: i64,
) {
    assert!(upper_bound.abs() < 20);
    let iterate_range = 101;
    let schema = match bound_type {
        NumericBounds::Inclusive => {
            json!({"type":"integer", "maximum": upper_bound})
        }
        NumericBounds::Exclusive => {
            json!({"type":"integer", "exclusiveMaximum": upper_bound})
        }
    };
    for i in -iterate_range..=iterate_range {
        let sample_value = json!(i);
        let should_pass = match bound_type {
            NumericBounds::Inclusive => i <= upper_bound,
            NumericBounds::Exclusive => i < upper_bound,
        };
        json_schema_check(&schema, &sample_value, should_pass);
    }
}

#[rstest]
#[case(-10, -2)]
#[case(-10, -1)]
#[case(-10, 0)]
#[case(-10, 2)]
#[case(-3, -1)]
#[case(-3, 0)]
#[case(-3, 2)]
#[case(0, 2)]
#[case(0, 10)]
#[case(1, 3)]
#[case(1, 10)]
#[case(2, 10)]
fn integer_both_bounds(
    #[values(NumericBounds::Inclusive, NumericBounds::Exclusive)] lower_bound_type: NumericBounds,
    #[values(NumericBounds::Inclusive, NumericBounds::Exclusive)] upper_bound_type: NumericBounds,
    #[case] lower_bound: i32,
    #[case] upper_bound: i32,
) {
    assert!(lower_bound.abs() < 20);
    assert!(upper_bound.abs() < 20);

    let lb_str = match lower_bound_type {
        NumericBounds::Inclusive => "minimum",
        NumericBounds::Exclusive => "exclusiveMinimum",
    };
    let ub_str = match upper_bound_type {
        NumericBounds::Inclusive => "maximum",
        NumericBounds::Exclusive => "exclusiveMaximum",
    };
    let schema = json!({"type":"integer", lb_str:lower_bound, ub_str:upper_bound});

    let iterate_range = 101;
    for i in -iterate_range..=iterate_range {
        let sample_value = json!(i);
        let should_pass = match (&lower_bound_type, &upper_bound_type) {
            (NumericBounds::Inclusive, NumericBounds::Inclusive) => {
                i >= lower_bound && i <= upper_bound
            }
            (NumericBounds::Inclusive, NumericBounds::Exclusive) => {
                i >= lower_bound && i < upper_bound
            }
            (NumericBounds::Exclusive, NumericBounds::Inclusive) => {
                i > lower_bound && i <= upper_bound
            }
            (NumericBounds::Exclusive, NumericBounds::Exclusive) => {
                i > lower_bound && i < upper_bound
            }
        };
        json_schema_check(&schema, &sample_value, should_pass);
    }
}

#[rstest]
fn integer_limits_incompatible(
    #[values("minimum", "exclusiveMinimum")] min_type: &str,
    #[values("maximum", "exclusiveMaximum")] max_type: &str,
) {
    let schema = &json!({
        "type": "integer",
        min_type: 1,
        max_type: -1
    });
    json_err_test(
        schema,
        "Unsatisfiable schema: minimum (1) is greater than maximum (-1)",
    );
}

#[rstest]
fn integer_limits_empty() {
    json_err_test(
        &json!({
            "type": "integer",
            "exclusiveMinimum": 0, "exclusiveMaximum": 1
        }),
        "Failed to generate regex for integer range",
    );
}

#[rstest]
fn integer_multipleof(#[values(0, 1, 3, 12, 1818, 1819)] test_value: i64) {
    // See also issue 222: want to add some negative values
    const MULTIPLE_OF: i64 = 3;
    let schema = &json!({"type":"integer", "multipleOf": MULTIPLE_OF});
    json_schema_check(schema, &json!(test_value), test_value % MULTIPLE_OF == 0);
}

/*
Not clear if this should work
#[rstest]
fn integer_multipleof_zero(#[values(0, 3, 12, 1818)] test_value: i64) {
    let schema = &json!({"type":"integer", "multipleOf": 0});
    json_schema_check(
        schema,
        &json!(test_value),
        if test_value == 0 { true } else { false },
    );
}
*/

#[rstest]
fn integer_both_minima(#[values(2, 3, 4)] test_value: i64) {
    let schema = &json!({"type":"integer", "minimum": 2, "exclusiveMinimum": 1});
    json_schema_check(schema, &json!(test_value), true);
}

#[rstest]
fn integer_both_maxima(#[values(-1,0,1)] test_value: i64) {
    let schema = &json!({"type":"integer", "maximum": 4, "exclusiveMaximum": 2});
    json_schema_check(schema, &json!(test_value), true);
}

// ============================================================================

#[rstest]
#[case::zero(&json!(0))]
#[case::zero_float(&json!(0.0))]
#[case::one(&json!(1))]
#[case::one_float(&json!(1.0))]
#[case::minus_1(&json!(-1))]
#[case::minus_1_float(&json!(-1.0))]
#[case::large(&json!(10001.1))]
#[case::negative_large(&json!(-20002.231))]
#[case::positive_exponent(&json!(8.231e2))]
#[case::negative_exponent(&json!(8.231e-2))]
#[case(&json!(-1.61e28))]
#[case(&json!(-8.4e-8))]
fn number(#[case] sample_value: &Value) {
    let schema = &json!({"type":"number"});
    json_schema_check(schema, sample_value, true);
}

#[rstest]
#[case::string_one_float(&json!("1.0"))]
#[case::string_one(&json!("1"))]
#[case::boolean(&json!(false))]
#[case::string_alpha(&json!("Hello"))]
fn number_failures(#[case] sample_value: &Value) {
    let schema = &json!({"type":"number"});
    json_schema_check(schema, sample_value, false);
}

#[rstest]
fn number_lower_bound(
    #[values(NumericBounds::Inclusive, NumericBounds::Exclusive)] bound_type: NumericBounds,
    #[values(-4.2e9, -11.0, -2.0, -1.0, -3.4e-8, 0.0, 2.3e-7, 1.0, 2.0, 11.0, 3.4e8)]
    lower_bound: f64,
    #[values(-2.0e23, -100.0, -11.0, -5.0, -1.000001, -1.0, -0.9999999, -9.2e-2, -2e-3, 0.0, 1e-4, 9.2e-2, 0.99999, 1.0, 1.000001, 10.0, 12.0, 4.5e14)]
    test_value: f64,
) {
    /*
       NOTE:
       Change the '1e-4' in the test_value array to '1e-8 (which it should be for the 2.3e-7 bound), and get
       a parser failure.
       This appears to be because 1e-4 gets turned into 0.0001, whereas 1e-8 is left in exponential form.

       There is a similar issue with the '-2e-3' value, which really should be '-2e-9' for the -3.4e-8 bound
    */
    let schema = match bound_type {
        NumericBounds::Inclusive => {
            json!({"type":"number", "minimum": lower_bound})
        }
        NumericBounds::Exclusive => {
            json!({"type":"number", "exclusiveMinimum": lower_bound})
        }
    };
    let expected_pass = match bound_type {
        NumericBounds::Inclusive => test_value >= lower_bound,
        NumericBounds::Exclusive => test_value > lower_bound,
    };
    let test_json = json!(test_value);
    json_schema_check(&schema, &test_json, expected_pass);
}

#[rstest]
fn number_upper_bound(
    #[values(NumericBounds::Inclusive, NumericBounds::Exclusive)] bound_type: NumericBounds,
    // #[values(-4.2e9, -11.0, -2.0, -1.0, -3.4e-8, 0.0, 2.3e-7, 1.0, 2.0, 11.0, 3.4e8)]
    #[values(-100.0, 0.0, 1.0, 100.0)] upper_bound: f64,
    // #[values(-2.0e3, -100.0, -11.0, -5.0, -1.000001, -1.0, -0.9999999, -9.2e-2, -2e-9, 0.0, 1e-8, 9.2e-2, 0.99999, 1.0, 1.000001, 10.0, 12.0, 4.5e14)]
    #[values(-100.0001, -100.0, -99.999, 0.9999, 1.0, 1.00001, 99.999, 100.0, 100.00001)]
    test_value: f64,
) {
    /*
       Seeing similar issues to the number_lower_bound test. The commented values are
       what should be run.

       Even with this reduced set, adding "0.0" to the test_value array causes a failure
    */
    let schema = match bound_type {
        NumericBounds::Inclusive => {
            json!({"type":"number", "maximum": upper_bound})
        }
        NumericBounds::Exclusive => {
            json!({"type":"number", "exclusiveMaximum": upper_bound})
        }
    };
    let expected_pass = match bound_type {
        NumericBounds::Inclusive => test_value <= upper_bound,
        NumericBounds::Exclusive => test_value < upper_bound,
    };
    let test_json = json!(test_value);
    json_schema_check(&schema, &test_json, expected_pass);
}

#[rstest]
#[case(&json!(0))]
#[case(&json!(-100))]
#[case(&json!(100))]
fn number_limits_inc_inc(#[case] sample_value: &Value) {
    let schema = &json!({"type":"number", "minimum": -100, "maximum": 100});
    json_schema_check(schema, sample_value, true);
}

#[rstest]
#[case(&json!(-100.0000001))]
#[case(&json!(100.000001))]
#[case(&json!(2.0e2))]
fn number_limits_inc_inc_failures(#[case] sample_value: &Value) {
    let schema = &json!({"type":"number", "minimum": -100, "maximum": 100});
    json_schema_check(schema, sample_value, false);
}

#[rstest]
#[case(&json!(0))]
#[case(&json!(-0.999999))]
#[case(&json!(-1e-2))]
#[case(&json!(100))]
fn number_limits_excl_inc(#[case] sample_value: &Value) {
    let schema = &json!({"type":"number", "exclusiveMinimum": -1, "maximum": 100});
    json_schema_check(schema, sample_value, true);
}

#[rstest]
#[case(&json!(-1))]
#[case(&json!(-1.0))]
#[case(&json!(100.000001))]
#[case(&json!(2.0e2))]
fn number_limits_exclu_inc_failures(#[case] sample_value: &Value) {
    let schema = &json!({"type":"number", "exclusiveMinimum": -1, "maximum": 100});
    json_schema_check(schema, sample_value, false);
}

#[rstest]
#[case(&json!(-2))]
#[case(&json!(-100))]
#[case(&json!(-1.00001))]
#[case(&json!(-1.00001e0))]
fn number_limits_inc_excl(#[case] sample_value: &Value) {
    let schema = &json!({"type":"number", "minimum": -100, "exclusiveMaximum": -1});
    json_schema_check(schema, sample_value, true);
}

#[rstest]
#[case(&json!(-100.0000001))]
#[case(&json!(-1))]
#[case(&json!(-1.0))]
#[case(&json!(-2.12e6))]
#[case(&json!(-4.6e-6))]
fn number_limits_inc_excl_failures(#[case] sample_value: &Value) {
    let schema = &json!({"type":"number", "minimum": -100, "exclusiveMaximum": -1});
    json_schema_check(schema, sample_value, false);
}

#[rstest]
fn number_limits_incompatible(
    #[values("minimum", "exclusiveMinimum")] min_type: &str,
    #[values("maximum", "exclusiveMaximum")] max_type: &str,
) {
    let schema = &json!({
        "type": "number",
        min_type: -0.1,
        max_type: -1.0
    });
    json_err_test(
        schema,
        "Unsatisfiable schema: minimum (-0.1) is greater than maximum (-1)",
    );
}

#[rstest]
// Issue 222 #[case(-5.0, false)]
// Issue 222 #[case(-3.0, true)]
// Issue 222 #[case(0.0, true)]
// Issue 222 #[case(3.0, true)]
#[case(3.5, false)]
// Issue 222 #[case(12.0, true)]
// Issue 222 #[case(3e22, true)]
fn number_multipleof(#[case] test_value: f64, #[case] expected_pass: bool) {
    // See also issue 222
    const MULTIPLE_OF: f64 = 3.0;
    let schema = &json!({"type":"number", "multipleOf": MULTIPLE_OF});
    json_schema_check(schema, &json!(test_value), expected_pass);
}

#[rstest]
// Issue 222 #[case(-35.0, true)]
// Issue 222 #[case(-30.0, false)]
// Issue 222 #[case(-7.0, true)]
#[case(0.0, true)]
#[case(3.5, true)]
#[case(4.0, false)]
#[case(325.5, true)]
#[case(326.5, false)]
// Issue 222 #[case(3.5e22, true)]
fn number_multipleof_noninteger(#[case] test_value: f64, #[case] expected_pass: bool) {
    // See also issue 222
    const MULTIPLE_OF: f64 = 3.5;
    let schema = &json!({"type":"number", "multipleOf": MULTIPLE_OF});
    json_schema_check(schema, &json!(test_value), expected_pass);
}

/*
Not clear if this should work
#[rstest]
fn number_multipleof_zero(#[values(0.0, 3.0, 12.0, 1818.0)] test_value: f64) {
    const MULTIPLE_OF: f64 = 0.0;
    let schema = &json!({"type":"number", "multipleOf": MULTIPLE_OF});
    json_schema_check(
        schema,
        &json!(test_value),
        if test_value == 0.0 { true } else { false },
    );
}
*/

// ============================================================================

#[rstest]
#[case::empty(&json!(""))]
#[case::hello(&json!("Hello"))]
#[case::number_string(&json!("123"))]
#[case::special_chars(&json!("!@#$%^&*{}()_+"))]
#[case::single_quote(&json!("'"))]
#[case::double_quote(&json!("\""))]
#[case::unbalanced_brace(&json!("}"))]
#[case::multiline_string(&json!(
    r"Hello\nWorld
            
            With some extra line breaks etc.
            "
))]
fn string(#[case] sample_value: &Value) {
    let schema = &json!({"type":"string"});
    json_schema_check(schema, sample_value, true);
}

#[rstest]
#[case::integer(&json!(1))]
#[case::boolean(&json!(true))]
#[case::null(&json!(null))]
fn string_failures(#[case] sample_value: &Value) {
    let schema = &json!({"type":"string"});
    json_schema_check(schema, sample_value, false);
}

#[rstest]
#[case(&json!("aB"))]
#[case(&json!("aC"))]
#[case(&json!("aZ"))]
fn string_regex(#[case] sample_value: &Value) {
    let schema = &json!({"type":"string", "pattern": r"a[A-Z]"});
    json_schema_check(schema, sample_value, true);
}

#[rstest]
#[case(&json!("aa"))]
#[case(&json!("a1"))]
#[case(&json!("Hello World!"))]
fn string_regex_failures(#[case] sample_value: &Value) {
    let schema = &json!({"type":"string", "pattern": r"a[A-Z]"});
    json_schema_check(schema, sample_value, false);
}

#[rstest]
#[case(&json!("abc"))]
#[case(&json!("abcd"))]
#[case(&json!("abcde"))]
fn string_length_many(#[case] sample_value: &Value) {
    let schema = &json!({"type":"string", "minLength": 3, "maxLength": 5});
    json_schema_check(schema, sample_value, true);
}

#[rstest]
#[case(&json!(""))]
#[case(&json!("ab"))]
#[case(&json!("abcdef"))]
fn string_length_many_failures(#[case] sample_value: &Value) {
    let schema = &json!({"type":"string", "minLength": 3, "maxLength": 5});
    json_schema_check(schema, sample_value, false);
}

#[rstest]
#[case(&json!("abc"))]
#[case(&json!("def"))]
fn string_length_single(#[case] sample_value: &Value) {
    let schema = &json!({"type":"string", "minLength": 3, "maxLength": 3});
    json_schema_check(schema, sample_value, true);
}

#[rstest]
#[case(&json!(""))]
#[case(&json!("ab"))]
#[case(&json!("abcd"))]
fn string_length_single_failures(#[case] sample_value: &Value) {
    let schema = &json!({"type":"string", "minLength": 3, "maxLength": 3});
    json_schema_check(schema, sample_value, false);
}

#[rstest]
#[case(&json!(""))]
fn string_length_empty(#[case] sample_value: &Value) {
    let schema = &json!({"type":"string", "minLength": 0, "maxLength": 0});
    json_schema_check(schema, sample_value, true);
}

#[rstest]
#[case(&json!("a"))]
#[case(&json!("abc"))]
fn string_length_empy_failures(#[case] sample_value: &Value) {
    let schema = &json!({"type":"string", "minLength": 0, "maxLength": 0});
    json_schema_check(schema, sample_value, false);
}

#[test]
fn string_length_unsatisfiable() {
    json_err_test(
        &json!({"type":"string", "minLength": 2, "maxLength": 1}),
        "Unsatisfiable schema: minLength (2) is greater than maxLength (1)",
    );
}