commit 6aee2ed42b0295c61f6453773a209fff03077465
author Alphan Ulusoy <alphan@google.com> Fri Sep 09 17:14:55 2022 -0400
committer Alphan Ulusoy <alphan@google.com> Mon Sep 12 10:27:24 2022 -0400
tree 585f394ae412808ed16ec4db45ed6a780ebaf87a
parent b5973ce4e78d8e9572e6526c5f34da35bf583780

[test] Minor cleanup in e2e_boostrap_entry

Signed-off-by: Alphan Ulusoy <alphan@google.com>

sw/host/tests/rom/e2e_bootstrap_entry/src/main.rs

diff --git a/sw/host/tests/rom/e2e_bootstrap_entry/src/main.rs b/sw/host/tests/rom/e2e_bootstrap_entry/src/main.rs
index ffa3d48..c53ec8d 100644
--- a/sw/host/tests/rom/e2e_bootstrap_entry/src/main.rs
+++ b/sw/host/tests/rom/e2e_bootstrap_entry/src/main.rs
@@ -292,28 +292,21 @@
 fn main() -> Result<()> {
     let opts = Opts::from_args();
     opts.init.init_logging();
-
     let transport = opts.init.init_target()?;
+
     execute_test!(test_bootstrap_enabled_not_requested, &opts, &transport);
     execute_test!(test_bootstrap_enabled_requested, &opts, &transport);
     execute_test!(test_jedec_id, &opts, &transport);
     execute_test!(test_sfdp, &opts, &transport);
     execute_test!(test_write_enable_disable, &opts, &transport);
-    // `kErrorBootstrapEraseAddress` (01425303) is defined in `error.h`.
-    execute_test!(
-        test_bootstrap_shutdown,
-        &opts,
-        &transport,
-        SpiFlash::SECTOR_ERASE,
-        "01425303"
-    );
-    // `kErrorBootstrapProgramAddress` (02425303) is defined in `error.h`.
-    execute_test!(
-        test_bootstrap_shutdown,
-        &opts,
-        &transport,
-        SpiFlash::PAGE_PROGRAM,
-        "02425303"
-    );
+    for (cmd, bfv) in [
+        // `kErrorBootstrapEraseAddress` (01425303) is defined in `error.h`.
+        (SpiFlash::SECTOR_ERASE, "01425303"),
+        // `kErrorBootstrapProgramAddress` (02425303) is defined in `error.h`.
+        (SpiFlash::PAGE_PROGRAM, "02425303"),
+    ] {
+        execute_test!(test_bootstrap_shutdown, &opts, &transport, cmd, bfv);
+    }
+
     Ok(())
 }