test_data_container.py

import pytest

from topgg.data import DataContainerMixin, data
from topgg.errors import TopGGException


class _Int(int):
    ...


@pytest.fixture
def data_container() -> DataContainerMixin:
    dc = DataContainerMixin()
    dc.set_data("TEXT")
    dc.set_data(_Int(200))
    dc.set_data({"a": "b"})
    return dc


async def _async_callback(
    text: str = data(str), number: int = data(int), mapping: dict = data(dict)
):
    ...


def _sync_callback(
    text: str = data(str), number: int = data(int), mapping: dict = data(dict)
):
    ...


def _invalid_callback(number: float = data(float)):
    ...


@pytest.mark.asyncio
async def test_data_container_invoke_async_callback(data_container: DataContainerMixin):
    await data_container._invoke_callback(_async_callback)


@pytest.mark.asyncio
async def test_data_container_invoke_sync_callback(data_container: DataContainerMixin):
    await data_container._invoke_callback(_sync_callback)


def test_data_container_raises_data_already_exists(data_container: DataContainerMixin):
    with pytest.raises(
        TopGGException,
        match="<class 'str'> already exists. If you wish to override it, "
        "pass True into the override parameter.",
    ):
        data_container.set_data("TEST")


@pytest.mark.asyncio
async def test_data_container_raises_lookup_error(data_container: DataContainerMixin):
    with pytest.raises(LookupError):
        await data_container._invoke_callback(_invalid_callback)


def test_data_container_get_data(data_container: DataContainerMixin):
    assert data_container.get_data(str) == "TEXT"
    assert data_container.get_data(float) is None
    assert isinstance(data_container.get_data(set, set()), set)
    assert data_container.get_data(_Int) == 200


def test_data_container_get_data_by_subclass(data_container: DataContainerMixin):
    data = data_container.get_data(int)
    assert isinstance(data, _Int)
    assert data == 200
    assert int in data_container._lookup_cache


def test_data_container_override_lookup_cache(data_container: DataContainerMixin):
    assert data_container.get_data(int) is not None
    data_container.set_data(_Int(300), override=True)
    assert data_container.get_data(int) == 300