import pytest

from ows_text_campaigns.assets.models import Asset
from ows_text_campaigns.assets.repositories import AssetRepository

from tests.unit.types import CreateModel


class TestAssetRepository:
    @pytest.mark.db
    def test_get_by_id_for_object(
        self, repository: AssetRepository, create_model: CreateModel
    ) -> None:
        object_type = "artist"
        object_id = "12345"

        asset = create_model(
            Asset,
            object_type=object_type,
            object_id=object_id,
        )

        result = repository.get_by_id_for_object(
            asset_id=asset.id,
            object_type=object_type,
            object_id=object_id,
        )

        assert result == asset

    @pytest.mark.db
    def test_get_by_id_for_object_none(
        self, repository: AssetRepository, create_model: CreateModel
    ) -> None:
        object_type = "artist"
        object_id = "12345"

        asset = create_model(
            Asset,
            object_type=object_type,
            object_id=object_id,
        )

        result = repository.get_by_id_for_object(
            asset_id=asset.id,
            object_type="album",
            object_id=object_id,
        )

        assert result is None