Add QinQ strip and VLAN extend methods to TestPmdShell class.
Signed-off-by: Dean Marx <dm...@iol.unh.edu>
---
dts/framework/remote_session/testpmd_shell.py | 52 +++++++++++++++++++
1 file changed, 52 insertions(+)
diff --git a/dts/framework/remote_session/testpmd_shell.py
b/dts/framework/remote_session/testpmd_shell.py
index ac99bd09ef..aaab267a54 100644
--- a/dts/framework/remote_session/testpmd_shell.py
+++ b/dts/framework/remote_session/testpmd_shell.py
@@ -2099,6 +2099,58 @@ def set_vlan_filter(self, port: int, enable: bool,
verify: bool = True) -> None:
filter on port {port}"""
)
+ def set_vlan_extend(self, port: int, enable: bool, verify: bool = True) ->
None:
+ """Set vlan extend.
+
+ Args:
+ port: The port number to enable VLAN extend on.
+ enable: Enable extend on `port` if :data:`True`, otherwise disable
it.
+ verify: If :data:`True`, the output of the command and show port
info
+ is scanned to verify that vlan extend was set successfully.
+
+ Raises:
+ InteractiveCommandExecutionError: If `verify` is :data:`True` and
extend
+ fails to update.
+ """
+ extend_cmd_output = self.send_command(f"vlan set extend {'on' if
enable else 'off'} {port}")
+ if verify:
+ vlan_settings = self.show_port_info(port_id=port).vlan_offload
+ if enable ^ (vlan_settings is not None and VLANOffloadFlag.EXTEND
in vlan_settings):
+ self._logger.debug(
+ f"""Failed to {"enable" if enable else "disable"}
+ extend on port {port}:
\n{extend_cmd_output}"""
+ )
+ raise InteractiveCommandExecutionError(
+ f"""Failed to {"enable" if enable else "disable"} extend
on port {port}"""
+ )
+
+ def set_qinq_strip(self, port: int, enable: bool, verify: bool = True) ->
None:
+ """Set QinQ strip.
+
+ Args:
+ port: The port number to enable QinQ strip on.
+ enable: Enable stripping on `port` if :data:`True`, otherwise
disable it.
+ verify: If :data:`True`, the output of the command and show port
info
+ is scanned to verify that QinQ strip was set successfully.
+
+ Raises:
+ InteractiveCommandExecutionError: If `verify` is :data:`True` and
QinQ strip
+ fails to update.
+ """
+ qinq_cmd_output = self.send_command(
+ f"vlan set qinq_strip {'on' if enable else 'off'} {port}"
+ )
+ if verify:
+ vlan_settings = self.show_port_info(port_id=port).vlan_offload
+ if enable ^ (vlan_settings is not None and
VLANOffloadFlag.QINQ_STRIP in vlan_settings):
+ self._logger.debug(
+ f"""Failed to {"enable" if enable else "disable"}
+ QinQ strip on port {port}:
\n{qinq_cmd_output}"""
+ )
+ raise InteractiveCommandExecutionError(
+ f"""Failed to {"enable" if enable else "disable"} QinQ
strip on port {port}"""
+ )
+
def set_mac_address(self, port: int, mac_address: str, verify: bool =
True) -> None:
"""Set port's MAC address.
--
2.49.0
