From 96c5ef1d016ed3a33b091aa7cb9cc77f9eea07cd Mon Sep 17 00:00:00 2001 From: Adam Rutkowski Date: Thu, 9 Jun 2022 11:01:41 +0200 Subject: [PATCH] pyocf: register composite volume type id in ctx Signed-off-by: Adam Rutkowski --- tests/functional/tests/conftest.py | 6 ++++++ 1 file changed, 6 insertions(+) diff --git a/tests/functional/tests/conftest.py b/tests/functional/tests/conftest.py index 2570769..f61e187 100644 --- a/tests/functional/tests/conftest.py +++ b/tests/functional/tests/conftest.py @@ -14,7 +14,9 @@ from pyocf.types.volume import RamVolume, ErrorDevice from pyocf.types.volume_cache import CacheVolume from pyocf.types.volume_core import CoreVolume from pyocf.types.volume_replicated import ReplicatedVolume +from pyocf.types.cvolume import CVolume from pyocf.types.ctx import OcfCtx +from pyocf.helpers import get_composite_volume_type_id default_registered_volumes = [RamVolume, ErrorDevice, CacheVolume, CoreVolume, ReplicatedVolume] @@ -28,6 +30,7 @@ def pyocf_ctx(): c = OcfCtx.with_defaults(DefaultLogger(LogLevel.WARN)) for vol_type in default_registered_volumes: c.register_volume_type(vol_type) + c.register_internal_volume_type_id(CVolume, get_composite_volume_type_id()) yield c c.exit() gc.collect() @@ -39,6 +42,7 @@ def pyocf_ctx_log_buffer(): c = OcfCtx.with_defaults(logger) for vol_type in default_registered_volumes: c.register_volume_type(vol_type) + c.register_internal_volume_type_id(CVolume, get_composite_volume_type_id()) yield logger c.exit() gc.collect() @@ -51,6 +55,8 @@ def pyocf_2_ctx(): for vol_type in default_registered_volumes: c1.register_volume_type(vol_type) c2.register_volume_type(vol_type) + c1.register_internal_volume_type_id(CVolume, get_composite_volume_type_id()) + c2.register_internal_volume_type_id(CVolume, get_composite_volume_type_id()) yield [c1, c2] c1.exit() c2.exit()