from api.schemas.agent_message import MessageCreate
from api.schemas.doi import DoIReport
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.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_tpsc_schemas() -> None:
    assert TPSCCatalogRead is CoreTPSCCatalogRead
    assert TPSCGDPRReport is CoreTPSCGDPRReport
    assert GDPR_WARNING_LEVELS is CORE_GDPR_WARNING_LEVELS