Fix UTs

sonic-net · Oct 22, 2021 · e048b3a · e048b3a
1 parent cd49fff
commit e048b3a
Showing 1 changed file with 6 additions and 6 deletions.
diff --git a/tests/test_qos_map.py b/tests/test_qos_map.py
@@ -160,10 +160,10 @@ def test_dscp_to_fc(self, dvs):
         ]
 
         for fvs in maps:
-            dscp_ps.set('AZURE', swsscommon.FieldValuePairs([fvs]))
+            self.dscp_ps.set('AZURE', swsscommon.FieldValuePairs([fvs]))
             time.sleep(1)
-            assert(asic_qos_map_count == len(asic_db.get_keys('ASIC_STATE:SAI_OBJECT_TYPE_QOS_MAP')))
-            dscp_ps._del("AZURE")
+            assert(self.asic_qos_map_count == len(self.asic_db.get_keys('ASIC_STATE:SAI_OBJECT_TYPE_QOS_MAP')))
+            self.dscp_ps._del("AZURE")
 
         # Delete a map that does not exist.  Nothing should happen
         self.dscp_ps._del("AZURE")
@@ -201,10 +201,10 @@ def test_exp_to_fc(self, dvs):
         ]
 
         for fvs in maps:
-            exp_ps.set('AZURE', swsscommon.FieldValuePairs([fvs]))
+            self.exp_ps.set('AZURE', swsscommon.FieldValuePairs([fvs]))
             time.sleep(1)
-            assert(asic_qos_map_count == len(asic_db.get_keys('ASIC_STATE:SAI_OBJECT_TYPE_QOS_MAP')))
-            exp_ps._del("AZURE")
+            assert(self.asic_qos_map_count == len(self.asic_db.get_keys('ASIC_STATE:SAI_OBJECT_TYPE_QOS_MAP')))
+            self.exp_ps._del("AZURE")
 
         # Update the map with valid values
         exp_map = [(str(i), str(i + 10)) for i in range(0, 8)]