1 
use std::path::PathBuf;
2 
use std::process::exit;
3 
use std::sync::Arc;
4 

            

        

            

                5
            

            

                
            

            

                
use clap::ValueHint;

            

        

            

                6
            

            

                
            

            

                
use conjure_cp_cli::utils::conjure::{

            

        

            

                7
            

            

                
            

            

                
    get_solutions, get_solutions_from_conjure, solutions_to_json,

            

        

            

                8
            

            

                
            

            

                
};

            

        

            

                9
            

            

                
            

            

                
use conjure_cp_cli::utils::testing::normalize_solutions_for_comparison;

            

        

            

                10
            

            

                
            

            

                

            

        

            

                11
            

            

                
            

            

                
use crate::cli::GlobalArgs;

            

        

            

                12
            

            

                
            

            

                
use crate::solve::{self, init_solver};

            

        

            

                13
            

            

                
            

            

                

            

        

            

                14
            

            

                
            

            

                
#[derive(Clone, Debug, clap::Args)]

            

        

            

                15
            

            

                
            

            

                
pub struct Args {

            

        

            

                16
            

            

                
            

            

                
    /// The input Essence file

            

        

            

                17
            

            

                
            

            

                
    #[arg(value_name = "INPUT_ESSENCE",value_hint=ValueHint::FilePath)]

            

        

            

                18
            

            

                
            

            

                
    pub input_file: PathBuf,

            

        

            

                19
            

            

                
            

            

                
}

            

        

            

                20
            

            

                
            

            

                

            

        

            

                21
            

            

                
            

            

                
pub fn run_test_solve_command(global_args: GlobalArgs, local_args: Args) -> anyhow::Result<()> {

            

        

            

                22
            

            

                
            

            

                
    // stealing most of the steps of the solve command, except the solver stuff.

            

        

            

                23
            

            

                
            

            

                
    let input_file = local_args.input_file;

            

        

            

                24
            

            

                
            

            

                

            

        

            

                25
            

            

                
            

            

                
    let context = solve::init_context(&global_args, input_file.clone())?;

            

        

            

                26
            

            

                
            

            

                
    let model = solve::parse(&global_args, Arc::clone(&context))?;

            

        

            

                27
            

            

                
            

            

                
    let rewritten_model = solve::rewrite(model, &global_args, Arc::clone(&context))?;

            

        

            

                28
            

            

                
            

            

                

            

        

            

                29
            

            

                
            

            

                
    let solver = init_solver(global_args.solver);

            

        

            

                30
            

            

                
            

            

                

            

        

            

                31
            

            

                
            

            

                
    // now we are stealing from the integration tester

            

        

            

                32
            

            

                
            

            

                

            

        

            

                33
            

            

                
            

            

                
    let our_solutions = get_solutions(

            

        

            

                34
            

            

                
            

            

                
        solver,

            

        

            

                35
            

            

                
            

            

                
        rewritten_model,

            

        

            

                36
            

            

                
            

            

                
        0,

            

        

            

                37
            

            

                
            

            

                
        &global_args.save_solver_input_file,

            

        

            

                38
            

            

                
            

            

                
    )?;

            

        

            

                39
            

            

                
            

            

                

            

        

            

                40
            

            

                
            

            

                
    let conjure_solutions =

            

        

            

                41
            

            

                
            

            

                
        get_solutions_from_conjure(input_file.to_str().unwrap(), Arc::clone(&context))?;

            

        

            

                42
            

            

                
            

            

                

            

        

            

                43
            

            

                
            

            

                
    let our_solutions = normalize_solutions_for_comparison(&our_solutions);

            

        

            

                44
            

            

                
            

            

                
    let conjure_solutions = normalize_solutions_for_comparison(&conjure_solutions);

            

        

            

                45
            

            

                
            

            

                

            

        

            

                46
            

            

                
            

            

                
    let mut our_solutions_json = solutions_to_json(&our_solutions);

            

        

            

                47
            

            

                
            

            

                
    let mut conjure_solutions_json = solutions_to_json(&conjure_solutions);

            

        

            

                48
            

            

                
            

            

                

            

        

            

                49
            

            

                
            

            

                
    our_solutions_json.sort_all_objects();

            

        

            

                50
            

            

                
            

            

                
    conjure_solutions_json.sort_all_objects();

            

        

            

                51
            

            

                
            

            

                

            

        

            

                52
            

            

                
            

            

                
    if our_solutions_json == conjure_solutions_json {

            

        

            

                53
            

            

                
            

            

                
        eprintln!("Success: solutions match!");

            

        

            

                54
            

            

                
            

            

                
        exit(0);

            

        

            

                55
            

            

                
            

            

                
    } else {

            

        

            

                56
            

            

                
            

            

                
        eprintln!("=== our solutions:");

            

        

            

                57
            

            

                
            

            

                
        eprintln!("{our_solutions_json}");

            

        

            

                58
            

            

                
            

            

                
        eprintln!("=== conjure's solutions:");

            

        

            

                59
            

            

                
            

            

                
        eprintln!("{conjure_solutions_json}");

            

        

            

                60
            

            

                
            

            

                
        eprintln!("Failure: solutions do not match!");

            

        

            

                61
            

            

                
            

            

                
        exit(1);

            

        

            

                62
            

            

                
            

            

                
    }

            

        

            

                63
            

            

                
            

            

                
}