conftest.py

import os
import sys
from unittest import mock
import pytest

STUBS = {}

POSTGRES_URL = None
ENABLE_PTVSD = None


class Stub:
    def __init__(self, inner):
        self.inner = inner

    @property
    def found(self) -> bool:
        return not self.inner

    def start(self):
        self.inner and self.inner.start()

    def stop(self):
        self.inner and self.inner.stop()


def stub_indy() -> Stub:
    # detect indy module
    try:
        from indy.libindy import _cdll

        _cdll()

        return Stub(None)
    except ImportError:
        print("Skipping Indy-specific tests: python3-indy module not installed.")
    except OSError:
        print(
            "Skipping Indy-specific tests: libindy shared library could not be loaded."
        )

    modules = {}
    package_name = "indy"
    modules[package_name] = mock.MagicMock()
    for mod in [
        "anoncreds",
        "blob_storage",
        "crypto",
        "did",
        "error",
        "pool",
        "ledger",
        "non_secrets",
        "pairwise",
        "wallet",
    ]:
        submod = f"{package_name}.{mod}"
        modules[submod] = mock.MagicMock()
    return Stub(mock.patch.dict(sys.modules, modules))


def stub_askar() -> Stub:
    # detect aries-askar library
    try:
        from aries_askar.bindings import get_library

        get_library()
        return Stub(None)
    except ImportError:
        print("Skipping Askar-specific tests: aries_askar module not installed.")
    except OSError:
        print(
            "Skipping Askar-specific tests: aries-askar shared library"
            "could not be loaded."
        )

    modules = {}
    package_name = "aries_askar"
    modules[package_name] = mock.MagicMock()
    for mod in [
        "bindings",
        "error",
        "store",
        "types",
    ]:
        submod = f"{package_name}.{mod}"
        modules[submod] = mock.MagicMock()
    return Stub(mock.patch.dict(sys.modules, modules))


def stub_indy_credx() -> Stub:
    # detect indy-credx library
    try:
        from indy_credx.bindings import get_library

        get_library()
        return Stub(None)
    except ImportError:
        print("Skipping Indy-Credx-specific tests: indy_credx module not installed.")
    except OSError:
        print(
            "Skipping Indy-Credx-specific tests: indy-credx shared library"
            "could not be loaded."
        )

    modules = {}
    package_name = "indy_credx"
    modules[package_name] = mock.MagicMock()
    return Stub(mock.patch.dict(sys.modules, modules))


def stub_indy_vdr() -> Stub:
    # detect indy-vdr library
    try:
        from indy_vdr.bindings import get_library

        get_library()
        return Stub(None)
    except ImportError:
        print("Skipping Indy-VDR-specific tests: indy_vdr module not installed.")
    except OSError:
        print(
            "Skipping Indy-VDR-specific tests: indy-vdr shared library"
            "could not be loaded."
        )

    modules = {}
    package_name = "indy_vdr"
    modules[package_name] = mock.MagicMock()
    return Stub(mock.patch.dict(sys.modules, modules))


def stub_ursa_bbs_signatures() -> Stub:
    # detect ursa_bbs_signatures library
    try:
        from ursa_bbs_signatures._ffi.ffi_util import get_library

        get_library()
        return Stub(None)
    except ImportError:
        print(
            "Skipping Ursa-BBS-Signatures-specific tests:"
            " ursa_bbs_signatures module not installed."
        )
    except (OSError, Exception):
        print(
            "Skipping Ursa-BBS-Signatures-specific tests: bbs shared library "
            "could not be loaded."
        )

    modules = {}
    package_name = "ursa_bbs_signatures"
    modules[package_name] = mock.MagicMock()
    # Temporary until ursa_bbs_signatures is updated to export the FfiException
    # from the main package
    modules[package_name + "._ffi.FfiException"] = mock.MagicMock()
    return Stub(mock.patch.dict(sys.modules, modules))


def pytest_sessionstart(session):
    global STUBS, POSTGRES_URL, ENABLE_PTVSD
    ENABLE_PTVSD = os.getenv("ENABLE_PTVSD", False)
    # --debug-vs to use microsoft's visual studio remote debugger
    if ENABLE_PTVSD or "--debug" in sys.argv:
        try:
            import ptvsd

            ptvsd.enable_attach(address=("0.0.0.0", 5678))
            print("ptvsd is running")
            print("=== Waiting for debugger to attach ===")
            # To pause execution until the debugger is attached:
            ptvsd.wait_for_attach()
        except ImportError:
            print("ptvsd library was not found")

    POSTGRES_URL = os.getenv("POSTGRES_URL")

    STUBS.update(
        {
            "askar": stub_askar(),
            "indy": stub_indy(),
            "indy_credx": stub_indy_credx(),
            "indy_vdr": stub_indy_vdr(),
            "ursa_bbs_signatures": stub_ursa_bbs_signatures(),
        }
    )
    for stub in STUBS.values():
        stub.start()


def pytest_sessionfinish(session):
    global STUBS

    for stub in STUBS.values():
        stub.stop()
    STUBS.clear()


def pytest_runtest_setup(item: pytest.Item):
    global STUBS

    if tuple(item.iter_markers(name="askar")) and not STUBS["askar"].found:
        pytest.skip("test requires Askar support")

    if tuple(item.iter_markers(name="indy")) and not STUBS["indy"].found:
        pytest.skip("test requires Indy support")

    if tuple(item.iter_markers(name="indy_credx")) and not STUBS["indy_credx"].found:
        pytest.skip("test requires Indy-Credx support")

    if tuple(item.iter_markers(name="indy_vdr")) and not STUBS["indy_vdr"].found:
        pytest.skip("test requires Indy-VDR support")

    if (
        tuple(item.iter_markers(name="ursa_bbs_signatures"))
        and not STUBS["ursa_bbs_signatures"].found
    ):
        pytest.skip("test requires Ursa-BBS-Signatures support")

    if tuple(item.iter_markers(name="postgres")) and not POSTGRES_URL:
        pytest.skip("test requires Postgres support")