dify/api/tests/integration_tests/utils/test_module_import_helper.py

import os

from core.helper.module_import_helper import import_module_from_source, load_single_subclass_from_source
from tests.integration_tests.utils.parent_class import ParentClass


def test_loading_subclass_from_source():
    current_path = os.getcwd()
    module = load_single_subclass_from_source(
        module_name='ChildClass',
        script_path=os.path.join(current_path, 'child_class.py'),
        parent_type=ParentClass)
    assert module and module.__name__ == 'ChildClass'


def test_load_import_module_from_source():
    current_path = os.getcwd()
    module = import_module_from_source(
        module_name='ChildClass',
        py_file_path=os.path.join(current_path, 'child_class.py'))
    assert module and module.__name__ == 'ChildClass'


def test_lazy_loading_subclass_from_source():
    current_path = os.getcwd()
    clz = load_single_subclass_from_source(
        module_name='LazyLoadChildClass',
        script_path=os.path.join(current_path, 'lazy_load_class.py'),
        parent_type=ParentClass,
        use_lazy_loader=True)
    instance = clz('dify')
    assert instance.get_name() == 'dify'