Function backup_snapshot

tests/integration/src/system_upgrade_tests.rs:526–696

()

#[test]
fn backup_snapshot() {
    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;

    let station_init_arg = SystemInstall::Upgrade(SystemUpgrade { name: None });
    let station_init_arg_bytes = Encode!(&station_init_arg).unwrap();
    let upgrader_init_arg = InitArg {
        target_canister: canister_ids.station,
    };
    let upgrader_init_arg_bytes = Encode!(&upgrader_init_arg).unwrap();

    let upgrade = |system_upgrade_operation_input: SystemUpgradeOperationInput| {
        let extra_ticks = match system_upgrade_operation_input.target {
            SystemUpgradeTargetDTO::UpgradeStation => STATION_UPGRADE_EXTRA_TICKS,
            SystemUpgradeTargetDTO::UpgradeUpgrader => 0,
        };
        execute_request_with_extra_ticks(
            &env,
            WALLET_ADMIN_USER,
            canister_ids.station,
            RequestOperationInput::SystemUpgrade(system_upgrade_operation_input),
            extra_ticks,
        )
    };

    let snapshot = |target: &SystemUpgradeTargetDTO| -> Option<Vec<u8>> {
        let (canister_id, caller) = match target {
            SystemUpgradeTargetDTO::UpgradeStation => (canister_ids.station, upgrader_id),
            SystemUpgradeTargetDTO::UpgradeUpgrader => (upgrader_id, canister_ids.station),
        };
        let snapshots: Vec<_> = env
            .list_canister_snapshots(canister_id, Some(caller))
            .unwrap();
        if snapshots.is_empty() {
            None
        } else {
            assert_eq!(snapshots.len(), 1);
            Some(snapshots[0].id.clone())
        }
    };

    let check_snapshots = |target: &SystemUpgradeTargetDTO, snapshot_id: Option<Vec<u8>>| {
        match target {
            SystemUpgradeTargetDTO::UpgradeStation => {
                let snapshots_via_upgrader =
                    update_candid_as::<_, (ApiResult<Vec<upgrader_api::Snapshot>>,)>(
                        &env,
                        upgrader_id,
                        WALLET_ADMIN_USER,
                        "canister_snapshots",
                        (),
                    )
                    .unwrap()
                    .0