fix(linstor): move next call in try block and use vdi object

Signed-off-by: Mathieu Labourier <[email protected]>

Author: Millefeuille42

tests/storage/linstor/conftest.py

@@ -181,10 +181,6 @@ def host_and_corrupted_vdi_on_linstor_sr(host: Host, linstor_sr: SR, vm_ref: str
     vm: VM = host.import_vm(vm_ref, sr_uuid=linstor_sr.uuid)
     pool: Pool = host.pool
     master: Host = pool.master
-    vdi_uuid: str = next((
-        uuid for uuid in vm.vdi_uuids()
-        if pool.get_vdi_sr_uuid(uuid) == linstor_sr.uuid
-    ))
 
     def get_vdi_volume_name_from_linstor() -> str:
         result = master.ssh([
@@ -212,6 +208,10 @@ def get_vdi_host(path: str) -> Host:
         raise FileNotFoundError(f"Could not find matching host for `{vdi_uuid}`")
 
     try:
+        vdi_uuid: str = next((
+            vdi.uuid for vdi in vm.vdis if vdi.sr.uuid == linstor_sr.uuid
+        ))
+
         volume_name = get_vdi_volume_name_from_linstor()
         lv_path = f"/dev/{GROUP_NAME}/{volume_name}_00000"
         vdi_host = get_vdi_host(lv_path)