Fix flake in rack_set_initialized_with_many_nexus_services test

oxidecomputer · Feb 20, 2023 · 5b9263f · 5b9263f
1 parent 1637370
commit 5b9263f
Showing 1 changed file with 14 additions and 2 deletions.
diff --git a/nexus/src/db/datastore/rack.rs b/nexus/src/db/datastore/rack.rs
@@ -665,13 +665,25 @@ mod test {
             observed_external_ips[&observed_nexus_services[0].external_ip_id]
                 .ip
                 .ip(),
-            IpAddr::V4(nexus_ip_start),
+            if let internal_params::ServiceKind::Nexus { external_address } =
+                services[0].kind
+            {
+                external_address
+            } else {
+                panic!("Unexpected service kind")
+            }
         );
         assert_eq!(
             observed_external_ips[&observed_nexus_services[1].external_ip_id]
                 .ip
                 .ip(),
-            IpAddr::V4(nexus_ip_end),
+            if let internal_params::ServiceKind::Nexus { external_address } =
+                services[1].kind
+            {
+                external_address
+            } else {
+                panic!("Unexpected service kind")
+            }
         );
 
         // Furthermore, we should be able to see that this IP addresses have been