// Copyright (c) The Diem Core Contributors
// Copyright (c) The Move Contributors
// SPDX-License-Identifier: Apache-2.0

use move_cli::base::test::{run_move_unit_tests, UnitTestResult};
use move_core_types::account_address::AccountAddress;
use move_stdlib::{
    natives::{all_natives, nursery_natives, GasParameters, NurseryGasParameters},
    path_in_crate,
};
use move_unit_test::UnitTestingConfig;
use tempfile::tempdir;

fn run_tests_for_pkg(path_to_pkg: impl Into<String>, include_nursery_natives: bool) {
    let pkg_path = path_in_crate(path_to_pkg);

    let mut natives = all_natives(
        AccountAddress::from_hex_literal("0x1").unwrap(),
        GasParameters::zeros(),
    );
    if include_nursery_natives {
        natives.extend(nursery_natives(
            AccountAddress::from_hex_literal("0x1").unwrap(),
            NurseryGasParameters::zeros(),
        ))
    }

    let result = run_move_unit_tests(
        &pkg_path,
        move_package::BuildConfig {
            test_mode: true,
            install_dir: Some(tempdir().unwrap().path().to_path_buf()),
            ..Default::default()
        },
        UnitTestingConfig::default_with_bound(Some(100_000)),
        natives,
        /* compute_coverage */ false,
        &mut std::io::stdout(),
    )
    .unwrap();
    if result != UnitTestResult::Success {
        panic!("aborting because of Move unit test failures");
    }
}

#[test]
fn move_unit_tests() {
    run_tests_for_pkg(".", false);
    run_tests_for_pkg("nursery", true);
}