use crate::{
    get_error,
    integers::u32::{output_one, output_zero},
    parse_program,
};
use leo_types::InputValue;

use snarkos_curves::edwards_bls12::Fq;
use snarkos_models::gadgets::r1cs::TestConstraintSystem;

pub mod conditional;

// Ternary if {bool}? {expression} : {expression};

#[test]
fn test_ternary_basic() {
    let bytes = include_bytes!("ternary_basic.leo");
    let mut program_input_true = parse_program(bytes).unwrap();

    let mut program_input_false = program_input_true.clone();

    program_input_true.set_main_inputs(vec![Some(InputValue::Boolean(true))]);
    output_one(program_input_true);

    program_input_false.set_main_inputs(vec![Some(InputValue::Boolean(false))]);
    output_zero(program_input_false);
}

// Iteration for i {start}..{stop} { statements }

#[test]
fn test_iteration_basic() {
    let bytes = include_bytes!("iteration_basic.leo");
    let program = parse_program(bytes).unwrap();

    output_one(program);
}

// Assertion

#[test]
fn test_assertion_basic() {
    let bytes = include_bytes!("assertion_basic.leo");
    let program = parse_program(bytes).unwrap();

    let mut program_input_true = program.clone();
    let mut cs_satisfied = TestConstraintSystem::<Fq>::new();

    program_input_true.set_main_inputs(vec![Some(InputValue::Boolean(true))]);
    let _output = program_input_true.compile_constraints(&mut cs_satisfied).unwrap();

    assert!(cs_satisfied.is_satisfied());

    let mut program_input_false = program.clone();
    let mut cs_unsatisfied = TestConstraintSystem::<Fq>::new();

    program_input_false.set_main_inputs(vec![Some(InputValue::Boolean(false))]);
    let _output = program_input_false.compile_constraints(&mut cs_unsatisfied).unwrap();

    assert!(!cs_unsatisfied.is_satisfied());
}

#[test]
fn test_num_returns_fail() {
    let bytes = include_bytes!("num_returns_fail.leo");
    let program = parse_program(bytes).unwrap();

    let _ = get_error(program);
}