diff --git a/tests/common/snappi_tests/snappi_fixtures.py b/tests/common/snappi_tests/snappi_fixtures.py index 447373308de..b8b9f9541b1 100755 --- a/tests/common/snappi_tests/snappi_fixtures.py +++ b/tests/common/snappi_tests/snappi_fixtures.py @@ -1775,20 +1775,24 @@ def tgen_port_info(request: pytest.FixtureRequest, snappi_port_selection, get_sn rx_port_count, testbed ) - return snappi_dut_base_config(duthosts, snappi_ports, snappi_api, setup=True) - - flatten_skeleton_parameter = request.param - speed, category = flatten_skeleton_parameter.split("-") + testbed_config, port_config_list, snappi_ports = snappi_dut_base_config( + duthosts, snappi_ports, snappi_api, setup=True) + yield (testbed_config, port_config_list, snappi_ports) + logger.info('Snappi cleanup after test') + setup_dut_ports(False, duthosts, testbed_config, port_config_list, snappi_ports) + else: + flatten_skeleton_parameter = request.param + speed, category = flatten_skeleton_parameter.split("-") - if float(speed) not in snappi_port_selection or category not in snappi_port_selection[float(speed)]: - pytest.skip(f"Unsupported combination for {flatten_skeleton_parameter}") + if float(speed) not in snappi_port_selection or category not in snappi_port_selection[float(speed)]: + pytest.skip(f"Unsupported combination for {flatten_skeleton_parameter}") - snappi_ports = snappi_port_selection[float(speed)][category] + snappi_ports = snappi_port_selection[float(speed)][category] - if not snappi_ports: - pytest.skip(f"Unsupported combination for {flatten_skeleton_parameter}") + if not snappi_ports: + pytest.skip(f"Unsupported combination for {flatten_skeleton_parameter}") - return snappi_dut_base_config(duthosts, snappi_ports, snappi_api, setup=True) + return snappi_dut_base_config(duthosts, snappi_ports, snappi_api, setup=True) def flatten_list(lst):