diff --git a/tests/qos/qos_sai_base.py b/tests/qos/qos_sai_base.py index 6d77baccafb..b6b5b403f80 100644 --- a/tests/qos/qos_sai_base.py +++ b/tests/qos/qos_sai_base.py @@ -145,7 +145,7 @@ def __computeBufferThreshold(self, dut_asic, bufferProfile): else: db = "4" keystr = "BUFFER_POOL|" - if check_qos_db_fv_reference_with_table(duthost) == True: + if check_qos_db_fv_reference_with_table(dut_asic) == True: pool = bufferProfile["pool"].encode("utf-8").translate(None, "[]") else: pool = keystr + bufferProfile["pool"].encode("utf-8") @@ -171,7 +171,7 @@ def __updateVoidRoidParams(self, dut_asic, bufferProfile): Returns: Updates bufferProfile with VOID/ROID obtained from Redis db """ - if check_qos_db_fv_reference_with_table(duthost) == True: + if check_qos_db_fv_reference_with_table(dut_asic) == True: if self.isBufferInApplDb(dut_asic): bufferPoolName = bufferProfile["pool"].encode("utf-8").translate( None, "[]").replace("BUFFER_POOL_TABLE:",'' @@ -220,7 +220,7 @@ def __getBufferProfile(self, request, dut_asic, os_version, table, port, priorit keystr = "{0}|{1}|{2}".format(table, port, priorityGroup) bufkeystr = "BUFFER_POOL|" - if check_qos_db_fv_reference_with_table(duthost) == True: + if check_qos_db_fv_reference_with_table(dut_asic) == True: bufferProfileName = dut_asic.run_redis_cmd( argv = ["redis-cli", "-n", db, "HGET", keystr, "profile"] )[0].encode("utf-8").translate(None, "[]") @@ -289,7 +289,7 @@ def __getEcnWredParam(self, dut_asic, table, port): Returns: wredProfile (dict): Map of ECN/WRED attributes """ - if check_qos_db_fv_reference_with_table(duthost) == True: + if check_qos_db_fv_reference_with_table(dut_asic) == True: wredProfileName = dut_asic.run_redis_cmd( argv = [ "redis-cli", "-n", "4", "HGET", @@ -345,7 +345,7 @@ def __getSchedulerParam(self, dut_asic, port, queue): Returns: SchedulerParam (dict): Map of scheduler parameters """ - if check_qos_db_fv_reference_with_table(duthost) == True: + if check_qos_db_fv_reference_with_table(dut_asic) == True: schedProfile = dut_asic.run_redis_cmd( argv = [ "redis-cli", "-n", "4", "HGET",