from unittest import mock

import pytest
from fansifter_common.adapters.ows_account import Vendor
from fansifter_common.auth.account import Account

from ows_text_campaigns.campaigns.models import UrlDomain
from ows_text_campaigns.campaigns.services import UrlDomainService

from tests.unit.types import BuildModel, CreateModel


class TestUrlDomainService:
    @pytest.mark.db
    def test_get_allowed_domains(
        self,
        service: UrlDomainService,
        build_model: BuildModel,
        create_model: CreateModel,
        identity_id: str,
        ows_account_client_mock: mock.MagicMock,
    ) -> None:
        vendor = build_model(Vendor, company_brand="sme")
        account = Account(vendor_id=vendor.vendor_id, subaccount_id=0)
        url_domain = create_model(
            UrlDomain, brand=vendor.brand, vendor_id=None, subaccount_id=None
        )

        ows_account_client_mock.get_vendor.return_value = vendor

        allowed_domains = service.get_allowed_domains(
            identity_id=identity_id, account=account
        )

        assert allowed_domains == [url_domain]