lighthouse-pulse/lighthouse/environment/tests/environment_builder.rs

#![cfg(test)]

use environment::EnvironmentBuilder;
use eth2_network_config::{Eth2NetworkConfig, DEFAULT_HARDCODED_NETWORK};
use std::path::PathBuf;
use types::{Config, MainnetEthSpec};

fn builder() -> EnvironmentBuilder<MainnetEthSpec> {
    EnvironmentBuilder::mainnet()
        .multi_threaded_tokio_runtime()
        .expect("should set runtime")
        .null_logger()
        .expect("should set logger")
}

fn eth2_network_config() -> Option<Eth2NetworkConfig> {
    Eth2NetworkConfig::constant(DEFAULT_HARDCODED_NETWORK).expect("should decode mainnet params")
}

mod setup_eth2_config {
    use super::*;

    #[test]
    fn update_spec_with_yaml_config() {
        if let Some(mut eth2_network_config) = eth2_network_config() {
            let testnet_dir = PathBuf::from("./tests/testnet_dir");
            let config = testnet_dir.join("config.yaml");

            eth2_network_config.config =
                Config::from_file(config.as_path()).expect("should load yaml config");

            let environment = builder()
                .eth2_network_config(eth2_network_config)
                .expect("should setup eth2_config")
                .build()
                .expect("should build environment");

            assert_eq!(
                environment
                    .eth2_config
                    .spec
                    .min_genesis_active_validator_count,
                100000 // see testnet_dir/config.yaml
            );
            assert_eq!(
                environment.eth2_config.spec.inactivity_score_bias,
                2 // see testnet_dir/config.yaml
            );
        }
    }
}