[PATCH v2 08/16] recipes: switchdev: TestLib: Add wrappers for LLDP operations

[Ido Schimmel]

From: Ido Schimmel <idosch(a)mellanox.com&gt;

There are common LLDP operations needed to be done in switchdev QoS
recipes, so add wrappers for them in TestLib.

I didn't want to add these operations to InterfaceAPI, as they are
really only useful for switchdev recipes. If needed, they can be
migrated in the future.

Signed-off-by: Ido Schimmel <idosch(a)mellanox.com&gt;
---
 recipes/switchdev/TestLib.py | 43 +++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 43 insertions(+)

diff --git a/recipes/switchdev/TestLib.py b/recipes/switchdev/TestLib.py
index 88dc0bf..167f408 100644
--- a/recipes/switchdev/TestLib.py
+++ b/recipes/switchdev/TestLib.py
@@ -312,3 +312,46 @@ class TestLib:
             err_msg = ""
 
         self.custom(iface.get_host(), "fdb test", err_msg)
+
+    def _lldp_set(self, iface, tlv_name, arg_name, arg):
+        iface_name = iface.get_devname()
+        m = iface.get_host()
+
+        cmd = "lldptool -i {} -V {} -T {}={}".format(iface_name, tlv_name,
+                                                     arg_name, arg)
+        m.run(cmd)
+
+    def lldp_ets_default_set(self, iface, willing=True):
+        up2tc = ','.join(["{}:0".format(x) for x in range(8)])
+        self._lldp_set(iface, "ETS-CFG", "up2tc", up2tc)
+
+        tsa = ','.join(["{}:strict".format(x) for x in range(8)])
+        self._lldp_set(iface, "ETS-CFG", "tsa", tsa)
+
+        willing = "yes" if willing else "no"
+        self._lldp_set(iface, "ETS-CFG", "willing", willing)
+
+        self._lldp_set(iface, "ETS-CFG", "enableTx",
"yes")
+
+    def lldp_ets_up2tc_set(self, iface, up2tc):
+        up2tc = ','.join(["{}:{}".format(x[0], x[1]) for x in up2tc])
+        self._lldp_set(iface, "ETS-CFG", "up2tc", up2tc)
+
+    def lldp_ets_tsa_set(self, iface, tsa, tcbw):
+        tsa = ','.join(["{}:{}".format(prio, algo) for prio, algo in
tsa])
+        self._lldp_set(iface, "ETS-CFG", "tsa", tsa)
+
+        tcbw_proper = [0] * 8
+        for prio, bw in tcbw:
+            tcbw_proper[prio] = bw
+        tcbw = ','.join(map(str, tcbw_proper))
+        self._lldp_set(iface, "ETS-CFG", "tcbw", tcbw)
+
+    def lldp_pfc_set(self, iface, prio, willing=True, delay=0):
+        prio = "none" if prio == [] else ','.join(map(str, prio))
+        self._lldp_set(iface, "PFC", "enabled", prio)
+
+        willing = "yes" if willing else "no"
+        self._lldp_set(iface, "PFC", "willing", willing)
+
+        self._lldp_set(iface, "PFC", "delay", delay)
-- 
2.7.4