Function system_restore

tests/integration/src/system_upgrade_tests.rs:403–477

()

#[test]
fn system_restore() {
    let TestEnv {
        env, canister_ids, ..
    } = setup_new_env();

    let system_info = get_system_info(&env, WALLET_ADMIN_USER, canister_ids.station);
    let upgrader_id = system_info.upgrader_id;

    for target in [
        SystemRestoreTargetDTO::RestoreStation,
        SystemRestoreTargetDTO::RestoreUpgrader,
    ] {
        let target_canister_id = match target {
            SystemRestoreTargetDTO::RestoreStation => canister_ids.station,
            SystemRestoreTargetDTO::RestoreUpgrader => upgrader_id,
        };
        let controller = match target {
            SystemRestoreTargetDTO::RestoreStation => upgrader_id,
            SystemRestoreTargetDTO::RestoreUpgrader => canister_ids.station,
        };
        let snapshot = env
            .take_canister_snapshot(target_canister_id, Some(controller), None)
            .unwrap();

        let system_restore = RequestOperationInput::SystemRestore(SystemRestoreOperationInput {
            target: target.clone(),
            snapshot_id: hex::encode(&snapshot.id),
        });

        match target {
            SystemRestoreTargetDTO::RestoreStation => {
                let system_restore_request = submit_request(
                    &env,
                    WALLET_ADMIN_USER,
                    canister_ids.station,
                    system_restore,
                );
                // wait until the station is restored (to a state before the restore request) at which point the restore request is not found anymore
                loop {
                    env.tick();
                    env.advance_time(Duration::from_secs(5));
                    match try_get_request(
                        &env,
                        WALLET_ADMIN_USER,
                        canister_ids.station,
                        system_restore_request.clone(),
                    ) {
                        Err(err) => assert!(
                            err.reject_message.contains(&format!(
                                "Canister {} is not running",
                                canister_ids.station
                            )) || err
                                .reject_message
                                .contains(&format!("Canister {} is stopped", canister_ids.station))
                        ),
                        Ok(Err(err)) => {
                            assert_eq!(err.code, "NOT_FOUND");
                            break;