ceph/src/spdk/ocf/tests/functional/tests/conftest.py

   1 #
   2 # Copyright(c) 2019 Intel Corporation
   3 # SPDX-License-Identifier: BSD-3-Clause-Clear
   4 #
   5
   6 import os
   7 import sys
   8 import pytest
   9 import gc
  10
  11 sys.path.append(os.path.join(os.path.dirname(__file__), os.path.pardir))
  12 from pyocf.types.logger import LogLevel, DefaultLogger, BufferLogger
  13 from pyocf.types.volume import Volume, ErrorDevice
  14 from pyocf.types.ctx import get_default_ctx
  15
  16
  17 def pytest_configure(config):
  18     sys.path.append(os.path.join(os.path.dirname(__file__), os.path.pardir))
  19
  20
  21 @pytest.fixture()
  22 def pyocf_ctx():
  23     c = get_default_ctx(DefaultLogger(LogLevel.WARN))
  24     c.register_volume_type(Volume)
  25     c.register_volume_type(ErrorDevice)
  26     yield c
  27     c.exit()
  28     gc.collect()
  29
  30
  31 @pytest.fixture()
  32 def pyocf_ctx_log_buffer():
  33     logger = BufferLogger(LogLevel.DEBUG)
  34     c = get_default_ctx(logger)
  35     c.register_volume_type(Volume)
  36     c.register_volume_type(ErrorDevice)
  37     yield logger
  38     c.exit()
  39     gc.collect()