import pytest

from dmp.audiences.models import AudienceExport
from dmp.audiences.repositories import AudienceExportRepository

from tests.unit.types import CreateModel


class TestAudienceExportRepository:
    @pytest.mark.db
    def test_get_by_id_and_audience_id(
        self, repository: AudienceExportRepository, create_model: CreateModel
    ) -> None:
        export = create_model(AudienceExport)

        result = repository.get_by_id_and_audience_id(
            export_id=export.id, audience_id=export.audience_id
        )

        assert result