X-Git-Url: https://gerrit.fd.io/r/gitweb?p=csit.git;a=blobdiff_plain;f=resources%2Flibraries%2Fpython%2FDUTSetup.py;h=78caeb79a46479ddad01d4043fc9b546ba0215a9;hp=e5e56784d5219854cc5cc412f3b1cdb749614041;hb=eca12f0b86528bbd02b3b5833f5ba022f18c1e0f;hpb=8f285166faf13156a4f7c70adac9a7e20549268f diff --git a/resources/libraries/python/DUTSetup.py b/resources/libraries/python/DUTSetup.py index e5e56784d5..78caeb79a4 100644 --- a/resources/libraries/python/DUTSetup.py +++ b/resources/libraries/python/DUTSetup.py @@ -10,46 +10,90 @@ # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. # See the License for the specific language governing permissions and # limitations under the License. + +"""DUT setup library.""" + from robot.api import logger -from topology import NodeType -from ssh import SSH -from constants import Constants -class DUTSetup(object): +from resources.libraries.python.topology import NodeType +from resources.libraries.python.ssh import SSH +from resources.libraries.python.constants import Constants +from resources.libraries.python.VatExecutor import VatExecutor - def __init__(self): - pass - def start_vpp_service_on_all_duts(self, nodes): +class DUTSetup(object): + """Contains methods for setting up DUTs.""" + @staticmethod + def start_vpp_service_on_all_duts(nodes): """Start up the VPP service on all nodes.""" ssh = SSH() for node in nodes.values(): if node['type'] == NodeType.DUT: ssh.connect(node) (ret_code, stdout, stderr) = \ - ssh.exec_command_sudo('service vpp restart') - if 0 != int(ret_code): + ssh.exec_command_sudo('service vpp restart', timeout=120) + if int(ret_code) != 0: logger.debug('stdout: {0}'.format(stdout)) logger.debug('stderr: {0}'.format(stderr)) raise Exception('DUT {0} failed to start VPP service'. - format(node['host'])) + format(node['host'])) + + @staticmethod + def vpp_show_version_verbose(node): + """Run "show version verbose" CLI command. + + :param node: Node to run command on. + :type node: dict + """ + vat = VatExecutor() + vat.execute_script("show_version_verbose.vat", node, json_out=False) - def setup_all_duts(self, nodes): + @staticmethod + def vpp_api_trace_save(node): + """Run "api trace save" CLI command. + + :param node: Node to run command on. + :type node: dict + """ + vat = VatExecutor() + vat.execute_script("api_trace_save.vat", node, json_out=False) + + @staticmethod + def vpp_api_trace_dump(node): + """Run "api trace custom-dump" CLI command. + + :param node: Node to run command on. + :type node: dict + """ + vat = VatExecutor() + vat.execute_script("api_trace_dump.vat", node, json_out=False) + + @staticmethod + def setup_all_duts(nodes): """Prepare all DUTs in given topology for test execution.""" for node in nodes.values(): if node['type'] == NodeType.DUT: - self.setup_dut(node) + DUTSetup.setup_dut(node) - def setup_dut(self, node): + @staticmethod + def setup_dut(node): + """Run script over SSH to setup the DUT node. + + :param node: DUT node to set up. + :type node: dict + + :raises Exception: If the DUT setup fails. + """ ssh = SSH() ssh.connect(node) (ret_code, stdout, stderr) = \ ssh.exec_command('sudo -Sn bash {0}/{1}/dut_setup.sh'.format( - Constants.REMOTE_FW_DIR, Constants.RESOURCES_LIB_SH)) + Constants.REMOTE_FW_DIR, Constants.RESOURCES_LIB_SH), timeout=120) logger.trace(stdout) - if 0 != int(ret_code): + logger.trace(stderr) + if int(ret_code) != 0: logger.debug('DUT {0} setup script failed: "{1}"'. - format(node['host'], stdout + stderr)) + format(node['host'], stdout + stderr)) raise Exception('DUT test setup script failed at node {}'. - format(node['host'])) + format(node['host']))