pub fn session_to_block(n: u32) -> u32 {

    // let block_number = flashbox_runtime::Period::get() * n;

    let block_number = Babe::current_epoch().duration.saturated_into::<u32>() * n;

    // Add 1 because the block that emits the NewSession event cannot contain any extrinsics,

    // so this is the first block of the new session that can actually be used

    block_number + 1

}

pub fn authorities() -> Vec<babe_primitives::AuthorityId> {

    let _session_index = Session::current_index();

    Babe::authorities()

        .iter()

        .map(|(key, _)| key.clone())

        .collect()

}

pub fn authorities_for_container(para_id: ParaId) -> Option<Vec<NimbusId>> {

    let session_index = Session::current_index();

    TanssiAuthorityAssignment::collator_container_chain(session_index)

        .expect("authorities should be set")

        .container_chains

        .get(&para_id)

        .cloned()

}

pub fn run_to_session(n: u32) {

    run_to_block(session_to_block(n));

}

pub fn run_to_block(n: u32) {

    let current_block_number = System::block_number();

    assert!(

        current_block_number < n,

    while System::block_number() < n {

        run_block();

    }

}

pub fn insert_authorities_and_slot_digests(slot: u64) {

    let pre_digest = Digest {

        logs: vec![DigestItem::PreRuntime(

            BABE_ENGINE_ID,

            PreDigest::SecondaryPlain(SecondaryPlainPreDigest {

                slot: slot.into(),

                authority_index: 0,

            })

            .encode(),

        )],

    };

    System::reset_events();

    System::initialize(

        &(System::block_number() + 1),

        &System::parent_hash(),

        &pre_digest,

    );

}

    Start(u32),

    End(u32),

    fn assert_can_advance(&self, new_state: &RunBlockState) {

        match self {

            RunBlockState::Start(n) => {

                assert_eq!(

                    new_state,

                    &RunBlockState::End(*n),

            RunBlockState::End(n) => {

                assert_eq!(

                    new_state,

                    &RunBlockState::Start(*n + 1),

    }

fn advance_block_state_machine(new_state: RunBlockState) {

    if frame_support::storage::unhashed::exists(b"__mock_is_xcm_test") {

    }

    let old_state: RunBlockState =

        frame_support::storage::unhashed::get(b"__mock_debug_block_state").unwrap_or(

            // Initial state is expecting a call to start() with block number 1, so old state should be

            // end of block 0

            RunBlockState::End(0),

        );

    old_state.assert_can_advance(&new_state);

    frame_support::storage::unhashed::put(b"__mock_debug_block_state", &new_state);

}

pub fn start_block() {

    let block_number = System::block_number();

    advance_block_state_machine(RunBlockState::Start(block_number + 1));

    insert_authorities_and_slot_digests(current_slot() + 1);

    // Initialize the new block

    Babe::on_initialize(System::block_number());

    Session::on_initialize(System::block_number());

    Initializer::on_initialize(System::block_number());

}

pub fn end_block() {

    let block_number = System::block_number();

    advance_block_state_machine(RunBlockState::End(block_number));

    // Finalize the block

    Babe::on_finalize(System::block_number());

    Session::on_finalize(System::block_number());

    Initializer::on_finalize(System::block_number());

    TransactionPayment::on_finalize(System::block_number());

}

pub fn run_block() {

    end_block();

    start_block()

}

    fn from(

        value: (

            u32,

            ContainerChainGenesisData<MaxLengthTokenSymbol>,

            u32,

            u32,

        ),

    ) -> Self {

        Self {

            para_id: value.0,

            genesis_data: value.1,

            block_production_credits: value.2,

            collator_assignment_credits: value.3,

        }

    }

pub fn default_config() -> pallet_configuration::HostConfiguration {

    pallet_configuration::HostConfiguration {

        max_collators: 100,

        min_orchestrator_collators: 2,

        max_orchestrator_collators: 2,

        collators_per_container: 2,

        full_rotation_period: 0,

        ..Default::default()

    }

}

    fn default() -> Self {

        Self {

            balances: vec![

                // Alice gets 10k extra tokens for her mapping deposit

                (AccountId::from(ALICE), 210_000 * UNIT),

                (AccountId::from(BOB), 100_000 * UNIT),

            ],

            validators: vec![

                (AccountId::from(ALICE), 210 * UNIT),

                (AccountId::from(BOB), 100 * UNIT),

            ],

            collators: Default::default(),

            sudo: Default::default(),

            para_ids: Default::default(),

            config: default_config(),

            own_para_id: Default::default(),

        }

    }

    pub fn with_balances(mut self, balances: Vec<(AccountId, Balance)>) -> Self {

        self.balances = balances;

        self

    }

    pub fn with_collators(mut self, collators: Vec<(AccountId, Balance)>) -> Self {

        self.collators = collators;

        self

    }

    pub fn with_para_ids(mut self, para_ids: Vec<ParaRegistrationParams>) -> Self {

        self.para_ids = para_ids;

        self

    }

    pub fn with_config(mut self, config: pallet_configuration::HostConfiguration) -> Self {

        self.config = config;

        self

    }

    pub fn build_storage(self) -> sp_core::storage::Storage {

        let mut t = frame_system::GenesisConfig::<Runtime>::default()

            .build_storage()

            .unwrap();

        pallet_babe::GenesisConfig::<Runtime> {

            ..Default::default()

        }

        .assimilate_storage(&mut t)

        .unwrap();

        pallet_balances::GenesisConfig::<Runtime> {

            balances: self.balances,

        }

        .assimilate_storage(&mut t)

        .unwrap();

        // We need to initialize these pallets first. When initializing pallet-session,

        // these values will be taken into account for collator-assignment.

        pallet_registrar::GenesisConfig::<Runtime> {

            para_ids: self

                .para_ids

                .iter()

                .cloned()

                .map(|registered_para| {

                    (registered_para.para_id.into(), registered_para.genesis_data)

                })

                .collect(),

        }

        .assimilate_storage(&mut t)

        .unwrap();

        // TODO: add here pallet_services_payment::GenesisConfig

        pallet_configuration::GenesisConfig::<Runtime> {

            config: self.config,

            ..Default::default()

        }

        .assimilate_storage(&mut t)

        .unwrap();

        let mut keys: Vec<_> = Vec::new();

        if !self.validators.is_empty() {

            let validator_keys: Vec<_> = self

                .validators

                .clone()

                .into_iter()

                .map(|(account, _balance)| {

                    let authority_keys = get_authority_keys_from_seed(&account.to_string());

                    (

                        account.clone(),

                        account,

                        starlight_runtime::SessionKeys {

                            babe: authority_keys.babe.clone(),

                            grandpa: authority_keys.grandpa.clone(),

                            para_validator: authority_keys.para_validator.clone(),

                            para_assignment: authority_keys.para_assignment.clone(),

                            authority_discovery: authority_keys.authority_discovery.clone(),

                            beefy: authority_keys.beefy.clone(),

                            nimbus: authority_keys.nimbus.clone(),

                        },

                    )

                })

                .collect();

            keys.extend(validator_keys)

        if !self.collators.is_empty() {

            let invulnerables: Vec<AccountId> = self

                .collators

                .clone()

                .into_iter()

                .map(|(account, _balance)| account)

                .collect();

            pallet_invulnerables::GenesisConfig::<Runtime> {

                invulnerables: invulnerables.clone(),

            }

            .assimilate_storage(&mut t)

            .unwrap();

            // But we also initialize their keys in the session pallet

            // We discard those that had the key initialized already

            // from the validator list

            // in other words, for testing purposes we allow to inject a validator account

            // in the collator list

            let validator_unique_accounts: Vec<_> = self

                .validators

                .iter()

                .map(|(account, _)| account.clone())

                .collect();

            let collator_keys: Vec<_> = self

                .collators

                .into_iter()

                .filter_map(|(account, _balance)| {

                    if validator_unique_accounts.contains(&account) {

                        None

                })

                .collect();

            keys.extend(collator_keys)

        }

        pallet_session::GenesisConfig::<Runtime> { keys }

            .assimilate_storage(&mut t)

            .unwrap();

        pallet_sudo::GenesisConfig::<Runtime> { key: self.sudo }

            .assimilate_storage(&mut t)

            .unwrap();

        t

    }

    pub fn build(self) -> sp_io::TestExternalities {

        let t = self.build_storage();

        let mut ext = sp_io::TestExternalities::new(t);

        ext.execute_with(|| {

            // Start block 1

            start_block();

        });

        ext

    }

pub fn root_origin() -> <Runtime as frame_system::Config>::RuntimeOrigin {

    <Runtime as frame_system::Config>::RuntimeOrigin::root()

}

pub fn origin_of(account_id: AccountId) -> <Runtime as frame_system::Config>::RuntimeOrigin {

    <Runtime as frame_system::Config>::RuntimeOrigin::signed(account_id)

}

pub fn empty_genesis_data() -> ContainerChainGenesisData<MaxLengthTokenSymbol> {

    ContainerChainGenesisData {

        storage: Default::default(),

        name: Default::default(),

        id: Default::default(),

        fork_id: Default::default(),

        extensions: Default::default(),

        properties: Default::default(),

    }

}

pub fn current_slot() -> u64 {

    Babe::current_slot().into()

}