from unittest import mock

import faker
import pytest
from fansifter_common.adapters.ows_account import Vendor
from fansifter_common.auth.account import Account, AccountAccess

from ows_text_campaigns.campaigns.handlers import (
    GetUrlDomainsHandler,
    GetUrlDomainsRequest,
)
from ows_text_campaigns.campaigns.models import UrlDomain

from tests.unit.types import BuildModel, CreateModel


class TestGetUrlDomainsHandler:
    @pytest.mark.db
    def test_get_url_domains_with_brand_only(
        self,
        handler: GetUrlDomainsHandler,
        create_model: CreateModel,
        identity_id: str,
        faker: faker.Faker,
        ows_account_client_mock: mock.Mock,
    ) -> None:
        vendor_id = faker.pyint()

        url_domain = create_model(
            UrlDomain, brand="sme", vendor_id=None, subaccount_id=None
        )

        ows_account_client_mock.get_vendor.return_value = Vendor(
            vendor_id=vendor_id,
            account_name="Test",
            company_brand="sme",
        )

        url_domains = handler.handle(
            GetUrlDomainsRequest(
                identity_id=identity_id,
                vendor_id=vendor_id,
                subaccount_id=faker.pyint(),
            )
        )

        assert url_domains == [url_domain]

    @pytest.mark.db
    def test_get_url_domains_with_brand_not_found(
        self,
        handler: GetUrlDomainsHandler,
        create_model: CreateModel,
        identity_id: str,
        faker: faker.Faker,
        ows_account_client_mock: mock.Mock,
    ) -> None:
        vendor_id = faker.pyint()

        create_model(UrlDomain, brand="theorchard", vendor_id=None, subaccount_id=None)

        ows_account_client_mock.get_vendor.return_value = Vendor(
            vendor_id=vendor_id,
            account_name="Test",
            company_brand="sme",
        )

        url_domains = handler.handle(
            GetUrlDomainsRequest(
                identity_id=identity_id,
                vendor_id=vendor_id,
                subaccount_id=faker.pyint(),
            )
        )

        assert url_domains == []

    @pytest.mark.db
    def test_get_url_domains_allowed_account(
        self,
        handler: GetUrlDomainsHandler,
        build_model: BuildModel,
        create_model: CreateModel,
        identity_id: str,
        ows_account_client_mock: mock.Mock,
        auth_service_mock: mock.MagicMock,
    ) -> None:
        account = build_model(Account)

        auth_service_mock.authorize_account = mock.Mock(
            return_value=AccountAccess(accounts=[account])
        )

        url_domain_1 = create_model(
            UrlDomain,
            brand="sme",
            vendor_id=account.vendor_id,
            subaccount_id=account.subaccount_id,
        )
        create_model(
            UrlDomain,
            brand="sme",
            vendor_id=account.vendor_id,
            subaccount_id=account.subaccount_id + 1,
        )

        ows_account_client_mock.get_vendor.return_value = Vendor(
            vendor_id=account.vendor_id,
            account_name="Test",
            company_brand="sme",
        )

        url_domains = handler.handle(
            GetUrlDomainsRequest(
                identity_id=identity_id,
                vendor_id=account.vendor_id,
                subaccount_id=account.subaccount_id,
            )
        )

        assert url_domains == [url_domain_1]