python-dlinfo/conftest.py

import builtins
import os
import re
import unittest.mock


def dyld_find_mock(lib_filename):
    # https://github.com/python/cpython/blob/master/Lib/ctypes/macholib/dyld.py#L116
    lib_name = re.match(r'^(lib\w+)\.\w+(\.\d+)?$', lib_filename).group(1)
    return os.path.join(os.sep, 'usr', 'lib', lib_name + '.dylib')

_BUILTIN_IMPORT = builtins.__import__

def import_mock(name, *args, **kwargs):
    try:
        return _BUILTIN_IMPORT(name, *args, **kwargs)
    except ModuleNotFoundError as exc:
        if name == 'ctypes.macholib.dyld': # only available on mac
            dyld_module = unittest.mock.MagicMock()
            dyld_module.dyld_find = dyld_find_mock
            return dyld_module
        raise exc

# required for tests/dlinfo_macosx_mock_test.py and doctests in dlinfo/_macosx.py.
# @pytest.fixture(autouse=True) does not run before import statements.
builtins.__import__ = import_mock