use cambrian::meta::AlgoConfigBuilder;
use cambrian::{self, meta, spec_util};
use cambrian::{sync_launch, termination::TerminationCriterion};
use float_cmp::approx_eq;

fn extract_anon_map_size(value: &serde_json::Value) -> usize {
    let mapping = match value {
        serde_json::Value::Object(mapping) => mapping,
        _ => unreachable!(),
    };

    mapping.len()
}

#[test]
fn anon_map() {
    let spec_str = "
    type: anon map
    valueType:
        type: bool
        init: false
    initSize: 0
    ";

    let spec = spec_util::from_yaml_str(spec_str).unwrap();

    let target_size = 3usize;

    let obj_func = meta::make_obj_func(move |value| {
        let anon_map_size = extract_anon_map_size(&value);
        let result =
            f64::from(i32::try_from(anon_map_size).unwrap() - i32::try_from(target_size).unwrap())
                .abs();
        Some(result)
    });

    let algo_config = AlgoConfigBuilder::new().build().unwrap();

    let termination_criteria = vec![TerminationCriterion::NumObjFuncEval(100)];

    let result = sync_launch::launch(
        spec,
        obj_func,
        algo_config,
        termination_criteria,
        None,
        true,
        None,
    )
    .unwrap();

    let obj_func_val = result.best_seen.obj_func_val;
    let anon_map_size = extract_anon_map_size(&result.best_seen.value);
    assert!(approx_eq!(f64, obj_func_val, 0.0));
    assert_eq!(anon_map_size, target_size);
}