From 1b5234573207c6e7d6d444e8518ff2e6cb638063 Mon Sep 17 00:00:00 2001 From: Kamil Gierszewski Date: Thu, 2 Jan 2025 02:31:55 +0100 Subject: [PATCH] test-api: fix core pool init Signed-off-by: Kamil Gierszewski --- test/functional/api/cas/core.py | 25 +++++++++++++++---------- 1 file changed, 15 insertions(+), 10 deletions(-) diff --git a/test/functional/api/cas/core.py b/test/functional/api/cas/core.py index eb34782..ad54219 100644 --- a/test/functional/api/cas/core.py +++ b/test/functional/api/cas/core.py @@ -41,16 +41,21 @@ class Core(Device): def __get_core_info(self) -> dict | None: core_dicts = get_cas_devices_dict()["cores"].values() - return next( - iter( - [ - core - for core in core_dicts - if core["cache_id"] == self.cache_id - and core["device_path"] == self.core_device.path - ] - ) - ) + # for core + core_device = [ + core + for core in core_dicts + if core["cache_id"] == self.cache_id and core["device_path"] == self.core_device.path + ] + if core_device: + return core_device[0] + + # for core pool + core_pool_dicts = get_cas_devices_dict()["core_pool"].values() + core_pool_device = [ + core for core in core_pool_dicts if core["device_path"] == self.core_device.path + ] + return core_pool_device[0] def create_filesystem(self, fs_type: Filesystem, force=True, blocksize=None): super().create_filesystem(fs_type, force, blocksize)