From d9c0c0e83a855d5b1afc0d70c7d7ba24361d38ea Mon Sep 17 00:00:00 2001 From: Michal Rakowski Date: Fri, 3 Jan 2020 12:17:54 +0100 Subject: [PATCH] cas_module.py: fixup load/unload methods Signed-off-by: Michal Rakowski --- test/functional/api/cas/cas_module.py | 13 +++++++++++-- 1 file changed, 11 insertions(+), 2 deletions(-) diff --git a/test/functional/api/cas/cas_module.py b/test/functional/api/cas/cas_module.py index 09faa7e..58f0dd6 100644 --- a/test/functional/api/cas/cas_module.py +++ b/test/functional/api/cas/cas_module.py @@ -2,7 +2,9 @@ # Copyright(c) 2019 Intel Corporation # SPDX-License-Identifier: BSD-3-Clause-Clear # + from aenum import Enum + from test_utils import os_utils from test_utils.os_utils import ModuleRemoveMethod @@ -13,5 +15,12 @@ class CasModule(Enum): def reload_all_cas_modules(): - os_utils.unload_kernel_module(CasModule.cache, ModuleRemoveMethod.modprobe) - os_utils.load_kernel_module(CasModule.cache) + os_utils.unload_kernel_module(CasModule.cache.value, ModuleRemoveMethod.modprobe) + os_utils.load_kernel_module(CasModule.cache.value) + + +def unload_all_cas_modules(): + os_utils.unload_kernel_module(CasModule.cache.value, + os_utils.ModuleRemoveMethod.rmmod) + os_utils.unload_kernel_module(CasModule.disk.value, + os_utils.ModuleRemoveMethod.rmmod)