Python3: resources and libraries
[csit.git] / resources / libraries / python / QemuManager.py
index ad205c8..cfc4fbd 100644 (file)
@@ -20,14 +20,14 @@ from resources.libraries.python.CpuUtils import CpuUtils
 from resources.libraries.python.QemuUtils import QemuUtils
 from resources.libraries.python.topology import NodeType, Topology
 
-__all__ = ["QemuManager"]
+__all__ = [u"QemuManager"]
 
 
-class QemuManager(object):
+class QemuManager:
     """QEMU lifecycle management class"""
 
     # Use one instance of class per tests.
-    ROBOT_LIBRARY_SCOPE = 'TEST CASE'
+    ROBOT_LIBRARY_SCOPE = u"TEST CASE"
 
     def __init__(self, nodes):
         """Init QemuManager object."""
@@ -46,57 +46,62 @@ class QemuManager(object):
         :param kwargs: Named parameters.
         :type kwargs: dict
         """
-        node = kwargs['node']
-        nf_chains = int(kwargs['nf_chains'])
-        nf_nodes = int(kwargs['nf_nodes'])
-        queues = kwargs['rxq_count_int'] if kwargs['auto_scale'] else 1
-        vs_dtc = kwargs['vs_dtc']
-        nf_dtc = kwargs['vs_dtc'] if kwargs['auto_scale'] else kwargs['nf_dtc']
-        nf_dtcr = kwargs['nf_dtcr'] if isinstance(kwargs['nf_dtcr'], int) else 2
+        node = kwargs[u"node"]
+        nf_chains = int(kwargs[u"nf_chains"])
+        nf_nodes = int(kwargs[u"nf_nodes"])
+        queues = kwargs[u"rxq_count_int"] if kwargs[u"auto_scale"] else 1
+        vs_dtc = kwargs[u"vs_dtc"]
+        nf_dtc = kwargs[u"vs_dtc"] if kwargs[u"auto_scale"] \
+            else kwargs[u"nf_dtc"]
+        nf_dtcr = kwargs[u"nf_dtcr"] \
+            if isinstance(kwargs[u"nf_dtcr"], int) else 2
 
         img = Constants.QEMU_VM_KERNEL
 
         for nf_chain in range(1, nf_chains + 1):
             for nf_node in range(1, nf_nodes + 1):
                 qemu_id = (nf_chain - 1) * nf_nodes + nf_node
-                name = '{node}_{qemu_id}'.format(node=node, qemu_id=qemu_id)
-                sock1 = '/var/run/vpp/sock-{qemu_id}-1'.format(qemu_id=qemu_id)
-                sock2 = '/var/run/vpp/sock-{qemu_id}-2'.format(qemu_id=qemu_id)
+                name = f"{node}_{qemu_id}"
+                sock1 = f"/var/run/vpp/sock-{qemu_id}-1"
+                sock2 = f"/var/run/vpp/sock-{qemu_id}-2"
+                idx1 = (nf_chain - 1) * nf_nodes * 2 + nf_node * 2 - 1
                 vif1_mac = Topology.get_interface_mac(
-                    self.nodes[node], 'vhost{idx}'.format(
-                        idx=(nf_chain - 1) * nf_nodes * 2 + nf_node * 2 - 1)) \
-                    if kwargs['vnf'] == 'testpmd_mac' \
-                    else kwargs['tg_if1_mac'] if nf_node == 1 \
-                    else '52:54:00:00:{id:02x}:02'.format(id=qemu_id - 1)
+                    self.nodes[node], f"vhost{idx1}"
+                ) if kwargs[u"vnf"] == u"testpmd_mac" \
+                    else kwargs[u"tg_if1_mac"] if nf_node == 1 \
+                    else f"52:54:00:00:{(qemu_id - 1):02x}:02"
+                idx2 = (nf_chain - 1) * nf_nodes * 2 + nf_node * 2
                 vif2_mac = Topology.get_interface_mac(
-                    self.nodes[node], 'vhost{idx}'.format(
-                        idx=(nf_chain - 1) * nf_nodes * 2 + nf_node * 2)) \
-                    if kwargs['vnf'] == 'testpmd_mac' \
-                    else kwargs['tg_if2_mac'] if nf_node == nf_nodes \
-                    else '52:54:00:00:{id:02x}:01'.format(id=qemu_id + 1)
+                    self.nodes[node], f"vhost{idx2}"
+                ) if kwargs[u"vnf"] == u"testpmd_mac" \
+                    else kwargs[u"tg_if2_mac"] if nf_node == nf_nodes \
+                    else f"52:54:00:00:{(qemu_id + 1):02x}:01"
 
                 self.machines_affinity[name] = CpuUtils.get_affinity_nf(
                     nodes=self.nodes, node=node, nf_chains=nf_chains,
                     nf_nodes=nf_nodes, nf_chain=nf_chain, nf_node=nf_node,
-                    vs_dtc=vs_dtc, nf_dtc=nf_dtc, nf_dtcr=nf_dtcr)
+                    vs_dtc=vs_dtc, nf_dtc=nf_dtc, nf_dtcr=nf_dtcr
+                )
 
                 self.machines[name] = QemuUtils(
                     node=self.nodes[node], qemu_id=qemu_id,
                     smp=len(self.machines_affinity[name]), mem=4096,
-                    vnf=kwargs['vnf'], img=img)
+                    vnf=kwargs[u"vnf"], img=img
+                )
                 self.machines[name].configure_kernelvm_vnf(
-                    mac1='52:54:00:00:{id:02x}:01'.format(id=qemu_id),
-                    mac2='52:54:00:00:{id:02x}:02'.format(id=qemu_id),
-                    vif1_mac=vif1_mac,
-                    vif2_mac=vif2_mac,
-                    queues=queues,
-                    jumbo_frames=kwargs['jumbo'])
+                    mac1=f"52:54:00:00:{qemu_id:02x}:01",
+                    mac2=f"52:54:00:00:{qemu_id:02x}:02",
+                    vif1_mac=vif1_mac, vif2_mac=vif2_mac, queues=queues,
+                    jumbo_frames=kwargs[u"jumbo"]
+                )
                 self.machines[name].qemu_add_vhost_user_if(
-                    sock1, jumbo_frames=kwargs['jumbo'], queues=queues,
-                    queue_size=kwargs['perf_qemu_qsz'])
+                    sock1, jumbo_frames=kwargs[u"jumbo"], queues=queues,
+                    queue_size=kwargs[u"perf_qemu_qsz"]
+                )
                 self.machines[name].qemu_add_vhost_user_if(
-                    sock2, jumbo_frames=kwargs['jumbo'], queues=queues,
-                    queue_size=kwargs['perf_qemu_qsz'])
+                    sock2, jumbo_frames=kwargs[u"jumbo"], queues=queues,
+                    queue_size=kwargs[u"perf_qemu_qsz"]
+                )
 
     def construct_vms_on_all_nodes(self, **kwargs):
         """Construct 1..Mx1..N VMs(s) with specified name on all nodes.
@@ -106,7 +111,7 @@ class QemuManager(object):
         """
         self.initialize()
         for node in self.nodes:
-            if self.nodes[node]['type'] == NodeType.DUT:
+            if self.nodes[node][u"type"] == NodeType.DUT:
                 self.construct_vms_on_node(node=node, **kwargs)
 
     def start_all_vms(self, pinning=False):
@@ -115,8 +120,8 @@ class QemuManager(object):
         :param pinning: If True, then do also QEMU process pinning.
         :type pinning: bool
         """
-        for machine, machine_affinity in zip(self.machines.values(),
-                                             self.machines_affinity.values()):
+        for machine, machine_affinity in \
+                zip(self.machines.values(), self.machines_affinity.values()):
             machine.qemu_start()
             if pinning:
                 machine.qemu_set_affinity(*machine_affinity)