from unittest import mock import pytest from anydi import Container from dmp.audiences.handlers import ( GetAudienceAdAccountsHandler, GetAudienceAdAccountsRequest, ) from dmp.audiences.models import Audience from dmp.audiences.services import AudienceService from tests.unit.faker import FakerTyped from tests.unit.types import CreateModel class TestGetAudienceAdAccountsHandler: @pytest.mark.db def test_get_audience_ad_accounts( self, container: Container, handler: GetAudienceAdAccountsHandler, create_model: CreateModel, identity_id: str, fake: FakerTyped, ) -> None: audience = create_model(Audience) audience_service_mock = mock.MagicMock(spec=AudienceService) audience_service_mock.get_audience.return_value = audience with container.override(AudienceService, audience_service_mock): result = handler.handle( GetAudienceAdAccountsRequest( identity_id=identity_id, audience_id=fake.uuid4_string() ) ) assert result == []