"""This module implements initialization and cleanup of DPDK environment."""
-from resources.libraries.python.ssh import SSH
-from resources.libraries.python.constants import Constants
+from robot.api import logger
+
+from resources.libraries.python.ssh import SSH, exec_cmd_no_error
+from resources.libraries.python.Constants import Constants
from resources.libraries.python.topology import NodeType, Topology
ret_code, _, _ = ssh.exec_command_sudo(cmd, timeout=600)
if ret_code != 0:
- raise RuntimeError('Failed to cleanup the dpdk at node {name}'
- .format(name=dut_node['host']))
+ raise RuntimeError('Failed to cleanup the dpdk at node {name}'.
+ format(name=dut_node['host']))
+
+ @staticmethod
+ def install_dpdk_test(node):
+ """
+ Prepare the DPDK test environment
+
+ :param node: Dictionary created from topology
+ :type node: dict
+ :returns: nothing
+ :raises RuntimeError: If command returns nonzero return code.
+ """
+ arch = Topology.get_node_arch(node)
+
+ command = ('{fwdir}/tests/dpdk/dpdk_scripts/install_dpdk.sh {arch}'.
+ format(fwdir=Constants.REMOTE_FW_DIR, arch=arch))
+ message = 'Install the DPDK failed!'
+ exec_cmd_no_error(node, command, timeout=600, message=message)
+
+ command = ('cat {fwdir}/dpdk*/VERSION'.
+ format(fwdir=Constants.REMOTE_FW_DIR))
+ message = 'Get DPDK version failed!'
+ stdout, _ = exec_cmd_no_error(node, command, message=message)
+
+ logger.info('DPDK Version: {version}'.format(version=stdout))
+
+ @staticmethod
+ def install_dpdk_test_on_all_duts(nodes):
+ """
+ Prepare the DPDK test environment on all DUTs.
+
+ :param nodes: Nodes from topology file.
+ :type nodes: dict
+ :returns: nothing
+ """
+ for node in nodes.values():
+ if node['type'] == NodeType.DUT:
+ DPDKTools.install_dpdk_test(node)