import os
import subprocess
from lbkit.tools import Tools

tool = Tools("component_test")
log = tool.log

class LiteBmcComponentTest(object):
    """
    开发者测试

    lbk test命令执行时会查找组件源码目录下的test.py脚本的LiteBmcComponentTest任务类
    随后创建任务对象并执行test方法
    """
    # 测试文件所在目录，覆盖率统计时会忽略该目录下的源码文件
    test_src_folder = ["test_package"]

    def __init__(self, rootfs_dir):
        self.rootfs_dir = rootfs_dir
        os.environ["ROOTFS_DIR"] = rootfs_dir
        log.info("export LD_LIBRARY_PATH={}".format(os.environ.get("LD_LIBRARY_PATH", "")))
        log.info("export ROOTFS_DIR={}".format(rootfs_dir))
        pass

    def test(self, **kwargs):
        """
        开发者测试

        启动测试，为便于扩展，必要时会传入字典类型的配置，具体传入什么样的值请参考lbkit开发者测试说明
        """
        test_ok = True
        result = subprocess.run(f"gtester -k {self.rootfs_dir}/opt/litebmc/apps/odf_validate_test/odf_validate_test", shell=True)
        if result is None or result.returncode != 0:
            test_ok = False

        if not test_ok:
            raise Exception("Test failed")