fix(infra): Octeon binding

Signed-off-by: Peter Mikus <peter.mikus@icloud.com>
Change-Id: I0e8ee3de3fbac53aad6a530804c67b0114b89b50

1 files changed, 15 insertions, 0 deletions

diff --git a/resources/libraries/python/DUTSetup.py b/resources/libraries/python/DUTSetup.py
index bc8af372df..8c240d9735 100644
--- a/resources/libraries/python/DUTSetup.py
+++ b/resources/libraries/python/DUTSetup.py
@@ -381,6 +381,21 @@ class DUTSetup:
                 DUTSetup.pci_driver_unbind(node, pci_addr)
 
     @staticmethod
+    def bind_pci_devices_to_specific_driver(node, driver, *pci_addrs):
+        """Bind PCI devices to input driver on node. Unbind from previous.
+
+        :param node: DUT node.
+        :param driver: Driver to bind to.
+        :param pci_addrs: PCI device addresses.
+        :type node: dict
+        :type driver: str
+        :type pci_addrs: list
+        """
+        for pci_addr in pci_addrs:
+            DUTSetup.pci_driver_unbind(node, pci_addr)
+            DUTSetup.pci_driver_bind(node, pci_addr, driver)
+
+    @staticmethod
     def pci_driver_bind(node, pci_addr, driver):
         """Bind PCI device to driver on node.