X-Git-Url: https://gerrit.fd.io/r/gitweb?p=csit.git;a=blobdiff_plain;f=resources%2Flibraries%2Fpython%2FMemif.py;h=d38f5000a7d784f84622390e3bc111a794da195a;hp=5c38ec46f5e8fa0c8f24516b3243597503067f30;hb=6da5a6920171682bd5bf6a77517bedfef91cbd0e;hpb=33fb34665214bbbd0a4b3154169b21c2da01f69b diff --git a/resources/libraries/python/Memif.py b/resources/libraries/python/Memif.py index 5c38ec46f5..d38f5000a7 100644 --- a/resources/libraries/python/Memif.py +++ b/resources/libraries/python/Memif.py @@ -1,4 +1,4 @@ -# Copyright (c) 2019 Cisco and/or its affiliates. +# Copyright (c) 2021 Cisco and/or its affiliates. # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. # You may obtain a copy of the License at: @@ -15,11 +15,11 @@ from enum import IntEnum + from robot.api import logger from resources.libraries.python.topology import NodeType, Topology -from resources.libraries.python.PapiExecutor import PapiExecutor -from resources.libraries.python.L2Util import L2Util +from resources.libraries.python.PapiExecutor import PapiSocketExecutor class MemifRole(IntEnum): @@ -28,7 +28,7 @@ class MemifRole(IntEnum): SLAVE = 1 -class Memif(object): +class Memif: """Memif interface class""" def __init__(self): @@ -43,14 +43,18 @@ class Memif(object): :returns: List of memif interfaces extracted from Papi response. :rtype: list """ - with PapiExecutor(node) as papi_exec: - details = papi_exec.add("memif_dump").get_details() + cmd = u"memif_dump" + with PapiSocketExecutor(node) as papi_exec: + details = papi_exec.add(cmd).get_details() for memif in details: - memif["if_name"] = memif["if_name"].rstrip('\x00') - memif["hw_addr"] = L2Util.bin_to_mac(memif["hw_addr"]) + memif[u"hw_addr"] = str(memif[u"hw_addr"]) + memif[u"role"] = memif[u"role"].value + memif[u"mode"] = memif[u"mode"].value + memif[u"flags"] = memif[u"flags"].value \ + if hasattr(memif[u"flags"], u"value") else int(memif[u"flags"]) - logger.debug("MEMIF details:\n{details}".format(details=details)) + logger.debug(f"MEMIF details:\n{details}") return details @@ -70,15 +74,14 @@ class Memif(object): includes only retval. :rtype: dict """ - cmd = 'memif_socket_filename_add_del' - err_msg = 'Failed to create memif socket on host {host}'.format( - host=node['host']) + cmd = u"memif_socket_filename_add_del" + err_msg = f"Failed to create memif socket on host {node[u'host']}" args = dict( - is_add=int(is_add), + is_add=is_add, socket_id=int(sid), - socket_filename=str('/tmp/' + filename) + socket_filename=str(u"/tmp/" + filename) ) - with PapiExecutor(node) as papi_exec: + with PapiSocketExecutor(node) as papi_exec: return papi_exec.add(cmd, **args).get_reply(err_msg) @staticmethod @@ -100,22 +103,23 @@ class Memif(object): :returns: sw_if_index :rtype: int """ - cmd = 'memif_create' - err_msg = 'Failed to create memif interface on host {host}'.format( - host=node['host']) + cmd = u"memif_create" + err_msg = f"Failed to create memif interface on host {node[u'host']}" args = dict( role=role, rx_queues=int(rxq), tx_queues=int(txq), socket_id=int(sid), - id=int(mid) + id=int(mid), + secret=u"" ) - with PapiExecutor(node) as papi_exec: + + with PapiSocketExecutor(node) as papi_exec: return papi_exec.add(cmd, **args).get_sw_if_index(err_msg) @staticmethod - def create_memif_interface(node, filename, mid, sid, rxq=1, txq=1, - role="SLAVE"): + def create_memif_interface( + node, filename, mid, sid, rxq=1, txq=1, role=u"SLAVE"): """Create Memif interface on the given node. :param node: Given node to create Memif interface on. @@ -136,7 +140,6 @@ class Memif(object): :rtype: int :raises ValueError: If command 'create memif' fails. """ - role = getattr(MemifRole, role.upper()).value # Create socket @@ -144,10 +147,11 @@ class Memif(object): # Create memif sw_if_index = Memif._memif_create( - node, mid, sid, rxq=rxq, txq=txq, role=role) + node, mid, sid, rxq=rxq, txq=txq, role=role + ) # Update Topology - if_key = Topology.add_new_port(node, 'memif') + if_key = Topology.add_new_port(node, u"memif") Topology.update_interface_sw_if_index(node, if_key, sw_if_index) ifc_name = Memif.vpp_get_memif_interface_name(node, sw_if_index) @@ -156,7 +160,9 @@ class Memif(object): ifc_mac = Memif.vpp_get_memif_interface_mac(node, sw_if_index) Topology.update_interface_mac_address(node, if_key, ifc_mac) - Topology.update_interface_memif_socket(node, if_key, '/tmp/' + filename) + Topology.update_interface_memif_socket( + node, if_key, u"/tmp/" + filename + ) Topology.update_interface_memif_id(node, if_key, mid) Topology.update_interface_memif_role(node, if_key, str(role)) @@ -169,7 +175,6 @@ class Memif(object): :param node: Given node to show Memif data on. :type node: dict """ - Memif._memif_details(node) @staticmethod @@ -180,7 +185,7 @@ class Memif(object): :type nodes: dict """ for node in nodes.values(): - if node['type'] == NodeType.DUT: + if node[u"type"] == NodeType.DUT: Memif.show_memif(node) @staticmethod @@ -194,12 +199,11 @@ class Memif(object): :returns: Memif interface name, or None if not found. :rtype: str """ - details = Memif._memif_details(node) for memif in details: - if memif["sw_if_index"] == sw_if_index: - return memif["if_name"] + if memif[u"sw_if_index"] == sw_if_index: + return memif[u"if_name"] return None @staticmethod @@ -213,10 +217,9 @@ class Memif(object): :returns: Memif interface MAC address, or None if not found. :rtype: str """ - details = Memif._memif_details(node) for memif in details: - if memif["sw_if_index"] == sw_if_index: - return memif["hw_addr"] + if memif[u"sw_if_index"] == sw_if_index: + return memif[u"hw_addr"] return None