Update test_ccmis.py

Fixing the test code failure case.
sonic-net · Aug 25, 2023 · 04e6048 · 04e6048
1 parent 5644385
commit 04e6048
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/tests/sonic_xcvr/test_ccmis.py b/tests/sonic_xcvr/test_ccmis.py
@@ -631,19 +631,23 @@ def test_get_transceiver_pm(self, mock_response, expected):
         (1, 1),
     ])
     def test_freeze_vdm_stats(self, mock_response, expected):
+        original_api = self.api.util_status_check_with_retry
         self.api.util_status_check_with_retry = MagicMock()
         self.api.util_status_check_with_retry.return_value = mock_response
         result = self.api.freeze_vdm_stats()
+        self.api.util_status_check_with_retry = original_api
         assert result == expected
 
     @pytest.mark.parametrize("mock_response, expected", [
         (0, 0),
         (1, 1),
     ])
     def test_unfreeze_vdm_stats(self, mock_response, expected):
+        original_api = self.api.util_status_check_with_retry
         self.api.util_status_check_with_retry = MagicMock()
         self.api.util_status_check_with_retry.return_value = mock_response
         result = self.api.unfreeze_vdm_stats()
+        self.api.util_status_check_with_retry = original_api
         assert result == expected
 
     @pytest.mark.parametrize("input_param, mock_response, expected",[