commit c638f799a04228c28f0c7baaaaaf850a199ff5da
author David Brown <david.brown@linaro.org> Tue Jan 10 12:34:33 2017 -0700
committer David Brown <david.brown@linaro.org> Tue Jan 10 16:43:07 2017 -0700
tree d565108444de18c284fa96e661e89c8388f97783
parent 1a44316c85bbe9de8d6db7080a73781906ea42b7

sim: Test multiple reverts

Make sure reboots after an image is reverted never results in the images
being swapped again.

sim/src/main.rs

diff --git a/sim/src/main.rs b/sim/src/main.rs
index 1f17f17..5691ac2 100644
--- a/sim/src/main.rs
+++ b/sim/src/main.rs
@@ -143,9 +143,11 @@
            bad, total_count,
            bad as f32 * 100.0 / total_count as f32);
 
-    info!("Try revert");
-    let fl2 = try_revert(&flash, &areadesc);
-    assert!(verify_image(&fl2, 0x020000, &primary));
+    for count in 2 .. 5 {
+        info!("Try revert: {}", count);
+        let fl2 = try_revert(&flash, &areadesc, count);
+        assert!(verify_image(&fl2, 0x020000, &primary));
+    }
 
     info!("Try norevert");
     let fl2 = try_norevert(&flash, &areadesc);
@@ -194,12 +196,13 @@
     (fl, cnt2)
 }
 
-fn try_revert(flash: &Flash, areadesc: &AreaDesc) -> Flash {
+fn try_revert(flash: &Flash, areadesc: &AreaDesc, count: usize) -> Flash {
     let mut fl = flash.clone();
     c::set_flash_counter(0);
 
-    assert_eq!(c::boot_go(&mut fl, &areadesc), 0);
-    assert_eq!(c::boot_go(&mut fl, &areadesc), 0);
+    for _ in 0 .. count {
+        assert_eq!(c::boot_go(&mut fl, &areadesc), 0);
+    }
     fl
 }