Patch firmware tests

oxidecomputer · Jul 2, 2024 · 48c3578 · 48c3578
1 parent 8f2301d
commit 48c3578
Showing 1 changed file with 5 additions and 9 deletions.
diff --git 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;