state-hub/tests/test_hub_core_imports.py

from api.schemas.agent_message import MessageCreate
from api.schemas.doi import DoIReport
from api.schemas.domain import DomainCreate, DomainRead, DomainRename, DomainUpdate
from api.schemas.tpsc import GDPR_WARNING_LEVELS, TPSCCatalogRead, TPSCGDPRReport
from hub_core.schemas.agent_message import MessageCreate as CoreMessageCreate
from hub_core.schemas.doi import DoIReport as CoreDoIReport
from hub_core.schemas.domain import (
    DomainCreate as CoreDomainCreate,
    DomainRead as CoreDomainRead,
    DomainRename as CoreDomainRename,
    DomainUpdate as CoreDomainUpdate,
)
from hub_core.schemas.tpsc import (
    GDPR_WARNING_LEVELS as CORE_GDPR_WARNING_LEVELS,
    TPSCCatalogRead as CoreTPSCCatalogRead,
    TPSCGDPRReport as CoreTPSCGDPRReport,
)


def test_state_hub_reexports_core_message_schema() -> None:
    assert MessageCreate is CoreMessageCreate


def test_state_hub_reexports_core_doi_schema() -> None:
    assert DoIReport is CoreDoIReport


def test_state_hub_reexports_core_domain_base_schemas() -> None:
    assert DomainCreate is CoreDomainCreate
    assert DomainRead is CoreDomainRead
    assert DomainRename is CoreDomainRename
    assert DomainUpdate is CoreDomainUpdate


def test_state_hub_reexports_core_tpsc_schemas() -> None:
    assert TPSCCatalogRead is CoreTPSCCatalogRead
    assert TPSCGDPRReport is CoreTPSCGDPRReport
    assert GDPR_WARNING_LEVELS is CORE_GDPR_WARNING_LEVELS