Skip to content

Commit e048b3a

Browse files
committed
Fix UTs
1 parent cd49fff commit e048b3a

File tree

1 file changed

+6
-6
lines changed

1 file changed

+6
-6
lines changed

tests/test_qos_map.py

+6-6
Original file line numberDiff line numberDiff line change
@@ -160,10 +160,10 @@ def test_dscp_to_fc(self, dvs):
160160
]
161161

162162
for fvs in maps:
163-
dscp_ps.set('AZURE', swsscommon.FieldValuePairs([fvs]))
163+
self.dscp_ps.set('AZURE', swsscommon.FieldValuePairs([fvs]))
164164
time.sleep(1)
165-
assert(asic_qos_map_count == len(asic_db.get_keys('ASIC_STATE:SAI_OBJECT_TYPE_QOS_MAP')))
166-
dscp_ps._del("AZURE")
165+
assert(self.asic_qos_map_count == len(self.asic_db.get_keys('ASIC_STATE:SAI_OBJECT_TYPE_QOS_MAP')))
166+
self.dscp_ps._del("AZURE")
167167

168168
# Delete a map that does not exist. Nothing should happen
169169
self.dscp_ps._del("AZURE")
@@ -201,10 +201,10 @@ def test_exp_to_fc(self, dvs):
201201
]
202202

203203
for fvs in maps:
204-
exp_ps.set('AZURE', swsscommon.FieldValuePairs([fvs]))
204+
self.exp_ps.set('AZURE', swsscommon.FieldValuePairs([fvs]))
205205
time.sleep(1)
206-
assert(asic_qos_map_count == len(asic_db.get_keys('ASIC_STATE:SAI_OBJECT_TYPE_QOS_MAP')))
207-
exp_ps._del("AZURE")
206+
assert(self.asic_qos_map_count == len(self.asic_db.get_keys('ASIC_STATE:SAI_OBJECT_TYPE_QOS_MAP')))
207+
self.exp_ps._del("AZURE")
208208

209209
# Update the map with valid values
210210
exp_map = [(str(i), str(i + 10)) for i in range(0, 8)]

0 commit comments

Comments
 (0)