tools/blackout/blobfs-checkerboard/src/lib.rs

// Copyright 2021 The Fuchsia Authors. All rights reserved.
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.

use {
    anyhow::Result, blackout_host::TestEnv, ffx_core::ffx_plugin,
    ffx_storage_blackout_blobfs_checkerboard_args::BlobfsCheckerboardCommand, std::time::Duration,
};

#[ffx_plugin("storage_dev")]
pub async fn blobfs_checkerboard(cmd: BlobfsCheckerboardCommand) -> Result<()> {
    let opts = blackout_host::CommonOpts {
        device_label: cmd.device_label,
        device_path: cmd.device_path,
        seed: cmd.seed,
        reboot: if cmd.dmc_reboot {
            blackout_host::RebootType::Dmc
        } else if cmd.relay.is_some() {
            blackout_host::RebootType::Hardware(cmd.relay.unwrap())
        } else {
            blackout_host::RebootType::Software
        },
        iterations: cmd.iterations,
        run_until_failure: cmd.run_until_failure,
    };
    let mut test = TestEnv::new(
        "blackout-blobfs-checkerboard-target",
        "blackout-blobfs-checkerboard-target-component",
        opts,
    )
    .await;
    test.setup_step()
        .load_step(Some(Duration::from_secs(30)))
        .reboot_step(cmd.bootserver)
        .verify_step(20, Duration::from_secs(10));
    test.run().await?;
    Ok(())
}