#
# Copyright(c) 2019 Intel Corporation
# SPDX-License-Identifier: BSD-3-Clause-Clear
#
from aenum import Enum
from config.configuration import cas_kernel_module, disk_kernel_module
from test_utils import os_utils
from test_utils.os_utils import ModuleRemoveMethod


class CasModule(Enum):
    cache = cas_kernel_module
    disk = disk_kernel_module


def reload_all_cas_modules():
    os_utils.unload_kernel_module(CasModule.cache, ModuleRemoveMethod.modprobe)
    os_utils.load_kernel_module(CasModule.cache)