__init__.py

#
# Copyright(c) 2020-2021 Intel Corporation
# SPDX-License-Identifier: BSD-3-Clause
#
from time import sleep

from core.test_run_utils import TestRun
from storage_devices.device import Device
from connection.utils.output import CmdException
from test_tools.os_tools import load_kernel_module, is_kernel_module_loaded, unload_kernel_module


class ScsiDebug:
    def __init__(self, params, config):
        self.params = params
        self.module_name = "scsi_debug"

    def pre_setup(self):
        pass

    def post_setup(self):
        self.reload()

    def reload(self):
        self.teardown()
        sleep(1)
        load_output = load_kernel_module(self.module_name, self.params)
        if load_output.exit_code != 0:
            raise CmdException(f"Failed to load {self.module_name} module", load_output)
        TestRun.LOGGER.info(f"{self.module_name} loaded successfully.")
        sleep(10)
        TestRun.scsi_debug_devices = Device.get_scsi_debug_devices()

    def teardown(self):
        if is_kernel_module_loaded(self.module_name):
            unload_kernel_module(self.module_name)


plugin_class = ScsiDebug