fix unit test

Author: YixuanWang-99

tests/unit/elastic_utils_test.py

@@ -107,19 +107,19 @@ def test_record_slice_state(self):
     self.fake_pathwaysutils.elastic.get_active_slice_indices.return_value = {0, 1}
 
     fake_recorder = Mock()
-    fake_recorder.record_slice_state = Mock()
+    fake_recorder.record_elastic_slice_counts = Mock()
 
     elastic_utils.record_slice_state(fake_recorder)
 
-    fake_recorder.record_slice_state.assert_called_once_with(
+    fake_recorder.record_elastic_slice_counts.assert_called_once_with(
         available_slices=2,
         active_slices=1,
         total_slices=2
     )
 
-    fake_recorder.record_slice_state.reset_mock()
+    fake_recorder.record_elastic_slice_counts.reset_mock()
     elastic_utils.record_slice_state(fake_recorder, active_slices_override=0)
-    fake_recorder.record_slice_state.assert_called_once_with(
+    fake_recorder.record_elastic_slice_counts.assert_called_once_with(
         available_slices=2,
         active_slices=0,
         total_slices=2
@@ -350,7 +350,7 @@ def test_record_elastic_event_start(self):
     fake_recorder.record_custom_badput_event_start_time.assert_called_once_with(
         custom_badput_event_type='elastic_slice_down'
     )
-    fake_recorder.record_slice_state.assert_called_once()
+    fake_recorder.record_elastic_slice_counts.assert_called_once()
     self.assertEqual(elastic_utils.pending_elastic_event_type, 'elastic_slice_down')
 
   def test_record_elastic_event_start_scale_up(self):
@@ -366,7 +366,7 @@ def test_record_elastic_event_start_scale_up(self):
     fake_recorder.record_custom_badput_event_start_time.assert_called_once_with(
         custom_badput_event_type='elastic_scale_up'
     )
-    fake_recorder.record_slice_state.assert_called_once()
+    fake_recorder.record_elastic_slice_counts.assert_called_once()
 
   def test_record_elastic_wait_end_and_reinit_start_noop_on_first_attempt(self):
     """Tests recording elastic event end and elastic reinit start."""
@@ -395,7 +395,7 @@ def test_record_elastic_wait_end_and_reinit_start(self):
     fake_recorder.record_custom_badput_event_start_time.assert_called_once_with(
         custom_badput_event_type='elastic_reinitialization'
     )
-    fake_recorder.record_slice_state.assert_called_once()
+    fake_recorder.record_elastic_slice_counts.assert_called_once()
     self.assertIs(elastic_utils.pending_reinit_recorder, fake_recorder)
     self.assertIsNone(elastic_utils.pending_elastic_event_type)
 
@@ -412,7 +412,7 @@ def test_record_elastic_reinit_end(self):
     fake_recorder.record_custom_badput_event_end_time.assert_called_once_with(
         custom_badput_event_type='elastic_reinitialization'
     )
-    fake_recorder.record_slice_state.assert_called_once()
+    fake_recorder.record_elastic_slice_counts.assert_called_once()
     self.assertIsNone(elastic_utils.pending_reinit_recorder)
 
   def test_record_elastic_reinit_end_on_cold_start(self):