From 6128836ceaab2ee7669c86188d8c3f68513e582d Mon Sep 17 00:00:00 2001 From: chiourung_huang Date: Wed, 23 Oct 2024 02:54:27 +0000 Subject: [PATCH] Fix unit test error --- sonic-xcvrd/tests/test_xcvrd.py | 6 ++++++ 1 file changed, 6 insertions(+) diff --git a/sonic-xcvrd/tests/test_xcvrd.py b/sonic-xcvrd/tests/test_xcvrd.py index ca421665a..64994ba68 100644 --- a/sonic-xcvrd/tests/test_xcvrd.py +++ b/sonic-xcvrd/tests/test_xcvrd.py @@ -1879,6 +1879,7 @@ def test_CmisManagerTask_reset_port_active_apsel_to_db_error_cases(self, mock_fi host_lanes_mask = 0xff # Case: table does not exist + task.xcvr_table_helper = XcvrTableHelper(DEFAULT_NAMESPACE) task.xcvr_table_helper.get_intf_tbl = MagicMock(return_value=None) task.reset_port_active_apsel_to_db(lport, host_lanes_mask) assert mock_field_value_pairs.call_count == 0 @@ -2404,10 +2405,13 @@ def test_CmisManagerTask_task_worker_fastboot(self, mock_chassis, mock_get_statu task.get_configured_laser_freq_from_db = MagicMock(return_value=193100) task.configure_tx_output_power = MagicMock(return_value=1) task.configure_laser_frequency = MagicMock(return_value=1) + task.reset_port_active_apsel_to_db = MagicMock() task.task_stopping_event.is_set = MagicMock(side_effect=[False, False, True]) task.task_worker() + assert mock_xcvr_api.tx_disable_channel.call_count == 1 + assert task.reset_port_active_apsel_to_db.call_count == 1 assert get_cmis_state_from_state_db('Ethernet0', task.xcvr_table_helper.get_status_tbl(task.port_mapping.get_asic_id_for_logical_port('Ethernet0'))) == CMIS_STATE_READY @patch('xcvrd.xcvrd.XcvrTableHelper.get_status_tbl') @@ -2539,10 +2543,12 @@ def test_CmisManagerTask_task_worker_host_tx_ready_false(self, mock_chassis, moc task.get_configured_laser_freq_from_db = MagicMock(return_value=193100) task.configure_tx_output_power = MagicMock(return_value=1) task.configure_laser_frequency = MagicMock(return_value=1) + task.reset_port_active_apsel_to_db = MagicMock() task.task_stopping_event.is_set = MagicMock(side_effect=[False, False, True]) task.task_worker() + assert task.reset_port_active_apsel_to_db.call_count == 1 assert mock_xcvr_api.tx_disable_channel.call_count == 1 assert get_cmis_state_from_state_db('Ethernet0', task.xcvr_table_helper.get_status_tbl(task.port_mapping.get_asic_id_for_logical_port('Ethernet0'))) == CMIS_STATE_READY