Function create_traffic_generator

dpdk/dts/framework/testbed_model/tg_node.py:83–96

A factory function for creating traffic generator object from user config.

(
    tg_node: TGNode, traffic_generator_config: ScapyTrafficGeneratorConfig
)

def create_traffic_generator(
    tg_node: TGNode, traffic_generator_config: ScapyTrafficGeneratorConfig
) -> CapturingTrafficGenerator:
    """A factory function for creating traffic generator object from user config."""

    from .scapy import ScapyTrafficGenerator

    match traffic_generator_config.traffic_generator_type:
        case TrafficGeneratorType.SCAPY:
            return ScapyTrafficGenerator(tg_node, traffic_generator_config)
        case _:
            raise ConfigurationError(
                f"Unknown traffic generator: {traffic_generator_config.traffic_generator_type}"
            )