From 48c357843bac5af76547415f37ea22b46f74020a Mon Sep 17 00:00:00 2001
From: Sean Klein <sean@oxide.computer>
Date: Tue, 2 Jul 2024 14:37:05 -0700
Subject: [PATCH] Patch firmware tests

---
 sled-storage/src/manager.rs | 14 +++++---------
 1 file changed, 5 insertions(+), 9 deletions(-)

diff --git a/sled-storage/src/manager.rs b/sled-storage/src/manager.rs
index 91ea0d86fb..e081bc5034 100644
--- a/sled-storage/src/manager.rs
+++ b/sled-storage/src/manager.rs
@@ -999,21 +999,17 @@ mod tests {
 
         // Now let's verify we saw the correct firmware update.
         for rd in &raw_disks {
-            let disk = all_disks_gen2
-                .iter_managed()
-                .find_map(|(identity, disk)| {
+            let firmware = all_disks_gen2
+                .iter_all()
+                .find_map(|(identity, _, _, fw)| {
                     if identity == rd.identity() {
-                        Some(disk)
+                        Some(fw)
                     } else {
                         None
                     }
                 })
                 .expect("disk exists");
-            assert_eq!(
-                disk.firmware(),
-                rd.firmware(),
-                "didn't see firmware update"
-            );
+            assert_eq!(firmware, rd.firmware(), "didn't see firmware update");
         }
 
         harness.cleanup().await;