Merge pull request #1546 from Kamalheib/devs

tests: test_device: Fix the use of get_device_list()
linux-rdma · Jan 20, 2025 · fb71cfe · fb71cfe
2 parents 9d45a4f + 6e450ee
commit fb71cfe
Showing 1 changed file with 26 additions and 26 deletions.
diff --git a/tests/test_device.py b/tests/test_device.py
@@ -101,21 +101,21 @@ def test_query_gid_table(self):
         """
         Test ibv_query_gid_table()
         """
-        devs = self.get_device_list()
-        with d.Context(name=devs[0].name.decode()) as ctx:
-            device_attr = ctx.query_device()
-            max_entries = 0
-            for port_num in range(1, device_attr.phys_port_cnt + 1):
-                port_attr = ctx.query_port(port_num)
-                max_entries += port_attr.gid_tbl_len
-            try:
-                if max_entries > 0:
-                    ctx.query_gid_table(max_entries)
-            except PyverbsRDMAError as ex:
-                if ex.error_code in [-errno.EOPNOTSUPP, -errno.EPROTONOSUPPORT]:
-                    raise unittest.SkipTest('ibv_query_gid_table is not'\
-                                            ' supported on this device')
-                raise ex
+        for dev in self.get_device_list():
+            with d.Context(name=dev.name.decode()) as ctx:
+                device_attr = ctx.query_device()
+                max_entries = 0
+                for port_num in range(1, device_attr.phys_port_cnt + 1):
+                    port_attr = ctx.query_port(port_num)
+                    max_entries += port_attr.gid_tbl_len
+                try:
+                    if max_entries > 0:
+                        ctx.query_gid_table(max_entries)
+                except PyverbsRDMAError as ex:
+                    if ex.error_code in [-errno.EOPNOTSUPP, -errno.EPROTONOSUPPORT]:
+                        raise unittest.SkipTest('ibv_query_gid_table is not'\
+                                                ' supported on this device')
+                    raise ex
 
     def test_query_gid_table_bad_flow(self):
         """
@@ -138,17 +138,17 @@ def test_query_gid_ex(self):
         """
         Test ibv_query_gid_ex()
         """
-        devs = self.get_device_list()
-        with d.Context(name=devs[0].name.decode()) as ctx:
-            try:
-                gid_tbl_len = ctx.query_port(self.ib_port).gid_tbl_len
-                if gid_tbl_len > 0:
-                    ctx.query_gid_ex(port_num=self.ib_port, gid_index=0)
-            except PyverbsRDMAError as ex:
-                if ex.error_code in [errno.EOPNOTSUPP, errno.EPROTONOSUPPORT]:
-                    raise unittest.SkipTest('ibv_query_gid_ex is not'\
-                                            ' supported on this device')
-                raise ex
+        for dev in self.get_device_list():
+            with d.Context(name=dev.name.decode()) as ctx:
+                try:
+                    gid_tbl_len = ctx.query_port(self.ib_port).gid_tbl_len
+                    if gid_tbl_len > 0:
+                        ctx.query_gid_ex(port_num=self.ib_port, gid_index=0)
+                except PyverbsRDMAError as ex:
+                    if ex.error_code in [errno.EOPNOTSUPP, errno.EPROTONOSUPPORT]:
+                        raise unittest.SkipTest('ibv_query_gid_ex is not'\
+                                                ' supported on this device')
+                    raise ex
 
     def test_query_gid_ex_bad_flow(self):
         """