test code correction

sonic-net · Nov 28, 2023 · 79d89b0 · 79d89b0
1 parent fbf8c83
commit 79d89b0
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/tests/sonic_xcvr/test_ccmis.py b/tests/sonic_xcvr/test_ccmis.py
@@ -642,7 +642,7 @@ def test_get_vdm_freeze_status(self, mock_response, expected):
     def test_get_vdm_unfreeze_status(self, mock_response, expected):
         self.api.xcvr_eeprom.read = MagicMock()
         self.api.xcvr_eeprom.read.return_value = mock_response
-        result = self.api.get_vdm_unfreeze_stats()
+        result = self.api.get_vdm_unfreeze_status()
         assert result == expected
 
     @pytest.mark.parametrize("mock_response, expected", [

diff --git a/tests/sonic_xcvr/test_sfp_optoe_base.py b/tests/sonic_xcvr/test_sfp_optoe_base.py
@@ -33,7 +33,7 @@ def test_freeze_vdm_stats(self, mock_response1, mock_response2, expected):
     @pytest.mark.parametrize("mock_response1, mock_response2, expected", [
         (0, ccmis_api, 0),
         (1, ccmis_api, 1),
-        (None, None, None),
+        (None, None, False),
     ])
     def test_unfreeze_vdm_stats(self, mock_response1, mock_response2, expected):
         self.sfp_optoe_api.get_xcvr_api = MagicMock()
@@ -47,7 +47,7 @@ def test_unfreeze_vdm_stats(self, mock_response1, mock_response2, expected):
     @pytest.mark.parametrize("mock_response1, mock_response2, expected", [
         (0, ccmis_api, 0),
         (1, ccmis_api, 1),
-        (None, None, None),
+        (None, None, False),
     ])
     def test_get_vdm_freeze_status(self, mock_response1, mock_response2, expected):
         self.sfp_optoe_api.get_xcvr_api = MagicMock()
@@ -61,7 +61,7 @@ def test_get_vdm_freeze_status(self, mock_response1, mock_response2, expected):
     @pytest.mark.parametrize("mock_response1, mock_response2, expected", [
         (0, ccmis_api, 0),
         (1, ccmis_api, 1), 
-        (None, None, None),
+        (None, None, False),
     ])
     def test_get_vdm_unfreeze_status(self, mock_response1, mock_response2, expected):
         self.sfp_optoe_api.get_xcvr_api = MagicMock()