fix the CMIS unittestfailure

Signed-off-by: Dante Su <dante.su@broadcom.com>
sonic-net · Dec 9, 2021 · de65013 · de65013
1 parent 8951bbf
commit de65013
Showing 1 changed file with 32 additions and 10 deletions.
diff --git a/sonic-xcvrd/tests/test_xcvrd.py b/sonic-xcvrd/tests/test_xcvrd.py
@@ -610,16 +610,38 @@ def test_CmisManagerTask_task_worker(self, mock_chassis):
             'ConfigStatusLane7': 'ConfigSuccess',
             'ConfigStatusLane8': 'ConfigSuccess'
         })
-        mock_xcvr_api.get_datapath_state = MagicMock(return_value={
-            'DP1State': 'DataPathInitialized',
-            'DP2State': 'DataPathInitialized',
-            'DP3State': 'DataPathInitialized',
-            'DP4State': 'DataPathInitialized',
-            'DP5State': 'DataPathInitialized',
-            'DP6State': 'DataPathInitialized',
-            'DP7State': 'DataPathInitialized',
-            'DP8State': 'DataPathInitialized'
-        })
+        mock_xcvr_api.get_datapath_state = MagicMock(side_effect=[
+            {
+                'DP1State': 'DataPathDeactivated',
+                'DP2State': 'DataPathDeactivated',
+                'DP3State': 'DataPathDeactivated',
+                'DP4State': 'DataPathDeactivated',
+                'DP5State': 'DataPathDeactivated',
+                'DP6State': 'DataPathDeactivated',
+                'DP7State': 'DataPathDeactivated',
+                'DP8State': 'DataPathDeactivated'
+            },
+            {
+                'DP1State': 'DataPathInitialized',
+                'DP2State': 'DataPathInitialized',
+                'DP3State': 'DataPathInitialized',
+                'DP4State': 'DataPathInitialized',
+                'DP5State': 'DataPathInitialized',
+                'DP6State': 'DataPathInitialized',
+                'DP7State': 'DataPathInitialized',
+                'DP8State': 'DataPathInitialized'
+            },
+            {
+                'DP1State': 'DataPathActivated',
+                'DP2State': 'DataPathActivated',
+                'DP3State': 'DataPathActivated',
+                'DP4State': 'DataPathActivated',
+                'DP5State': 'DataPathActivated',
+                'DP6State': 'DataPathActivated',
+                'DP7State': 'DataPathActivated',
+                'DP8State': 'DataPathActivated'
+            }
+        ])
 
         mock_sfp = MagicMock()
         mock_sfp.get_presence = MagicMock(return_value=True)