diff --git a/tests/pfcwd/test_pfcwd_function.py b/tests/pfcwd/test_pfcwd_function.py index 20c1ac0c26a..b53e60bc728 100644 --- a/tests/pfcwd/test_pfcwd_function.py +++ b/tests/pfcwd/test_pfcwd_function.py @@ -692,7 +692,7 @@ def run_test(self, dut, port, action, mmu_action=None): def set_traffic_action(self, duthost, action): action = action if action != "dontcare" else "drop" - if duthost.facts["asic_type"] == "mellanox": + if duthost.facts["asic_type"] in ["mellanox", "cisco-8000"]: self.rx_action = "forward" else: self.rx_action = action diff --git a/tests/pfcwd/test_pfcwd_warm_reboot.py b/tests/pfcwd/test_pfcwd_warm_reboot.py index ac108d1a40a..6bf107deb36 100644 --- a/tests/pfcwd/test_pfcwd_warm_reboot.py +++ b/tests/pfcwd/test_pfcwd_warm_reboot.py @@ -292,7 +292,7 @@ def verify_wd_func(self, dut, detect=True): tx_action = "forward" wd_action = "forward" - if dut.facts['asic_type'] == 'mellanox': + if dut.facts['asic_type'] in ['mellanox', 'cisco-8000']: rx_action = "forward" logger.info("--- Verify PFCwd function for pfcwd action {}, Tx traffic {}, Rx traffic {} ---".format(wd_action, tx_action, rx_action))