diff options
Diffstat (limited to 'scripts/automation/trex_control_plane/common/trex_streams.py')
| -rwxr-xr-x | scripts/automation/trex_control_plane/common/trex_streams.py | 6 |
1 files changed, 4 insertions, 2 deletions
diff --git a/scripts/automation/trex_control_plane/common/trex_streams.py b/scripts/automation/trex_control_plane/common/trex_streams.py index 9b10995b..0766aa7b 100755 --- a/scripts/automation/trex_control_plane/common/trex_streams.py +++ b/scripts/automation/trex_control_plane/common/trex_streams.py @@ -1,7 +1,7 @@ #!/router/bin/python import external_packages -from client_utils.packet_builder import CTRexPktBuilder +from client_utils.packet_builder_interface import CTrexPktBuilderInterface from collections import OrderedDict, namedtuple from client_utils.yaml_utils import * import trex_stl_exceptions @@ -422,7 +422,7 @@ class STLStream(object): if not isinstance(mode, STLTXMode): raise STLArgumentError('mode', mode) - if not isinstance(packet, CTRexPktBuilder): + if not isinstance(packet, CTrexPktBuilderInterface): raise STLArgumentError('packet', packet) if not isinstance(enabled, bool): @@ -452,6 +452,8 @@ class STLStream(object): # mode self.fields['mode'] = mode.to_json() + packet.compile() + # packet and VM self.fields['packet'] = packet.dump_pkt() self.fields['vm'] = packet.get_vm_data() |