test

Author: jayshrivastava

tests/metrics_collection.rs

@@ -152,7 +152,8 @@ mod tests {
         assert!(value > 100);
         let value = node_metrics::<NetworkCoalesceExec>(&d_physical, "elapsed_compute", 1);
         assert!(value > 100);
-        let value = node_metrics::<NetworkCoalesceExec>(&d_physical, "time_to_first_byte", 1);
+        let value =
+            node_metrics::<NetworkCoalesceExec>(&d_physical, "first_message_network_latency", 1);
         assert!(value > 0);
 
         let value = node_metrics::<NetworkShuffleExec>(&d_physical, "bytes_transferred", 1);
@@ -161,7 +162,8 @@ mod tests {
         assert!(value > 100);
         let value = node_metrics::<NetworkShuffleExec>(&d_physical, "elapsed_compute", 1);
         assert!(value > 100);
-        let value = node_metrics::<NetworkShuffleExec>(&d_physical, "time_to_first_byte", 1);
+        let value =
+            node_metrics::<NetworkShuffleExec>(&d_physical, "first_message_network_latency", 1);
         assert!(value > 0);
 
         Ok(())