from collections.abc import Iterator
from unittest import mock

import pytest

from email_campaigns.emails import tokens


@pytest.fixture()
def tokens_mock() -> Iterator[mock.MagicMock]:
    m = mock.MagicMock()
    with (
        mock.patch.object(tokens, "encrypt_email_id", m.encrypt_email_id),
        mock.patch.object(tokens, "decrypt_token", m.decrypt_token),
    ):
        yield m