import pytest

from dmp.ad_reporting.models import AdReportingCampaignBenchmarkByAccountDbt
from dmp.ad_reporting.repositories import (
    AdReportingCampaignBenchmarkByAccountRepository,
)

from tests.unit.types import CreateReportingModel


class TestAdReportingCampaignBenchmarkByAccountRepository:
    @pytest.mark.db
    def test_find_by_account_id(
        self,
        repository: AdReportingCampaignBenchmarkByAccountRepository,
        create_reporting_model: CreateReportingModel,
    ) -> None:
        benchmark = create_reporting_model(AdReportingCampaignBenchmarkByAccountDbt)

        result = repository.find_by_account_id(
            vendor_id=benchmark.vendor_id,
            subaccount_id=benchmark.subaccount_id,
        )

        assert result == [benchmark]

    @pytest.mark.db
    def test_find_by_account_id_empty(
        self,
        repository: AdReportingCampaignBenchmarkByAccountRepository,
    ) -> None:
        result = repository.find_by_account_id(
            vendor_id=0,
            subaccount_id=0,
        )

        assert result == []