import pytest
from atst.utils.mailer import Mailer, RedisMailer


class MockHost:
    def __init__(self):
        self.messages = []

    def __enter__(self):
        return self

    def __exit__(self, *args):
        pass

    def send_message(self, message):
        self.messages.append(message)


@pytest.fixture
def mail_host():
    return MockHost()


def test_mailer_can_send_mail(monkeypatch, mail_host):
    monkeypatch.setattr("atst.utils.mailer.Mailer.connection", lambda *args: mail_host)
    mailer = Mailer("localhost", 456, "leia@rebellion.net", "droidsyourelookingfor")
    message_data = {
        "recipients": ["ben@tattoine.org"],
        "subject": "help",
        "body": "you're my only hope",
    }
    mailer.send(**message_data)
    assert len(mail_host.messages) == 1
    message = mail_host.messages[0]
    assert message["To"] == message_data["recipients"][0]
    assert message["Subject"] == message_data["subject"]
    assert message.get_content().strip() == message_data["body"]


def test_redis_mailer_can_save_messages(app):
    mailer = RedisMailer(app.redis, server=None, port=None, sender=None, password=None)
    message_data = {
        "recipients": ["ben@tattoine.org"],
        "subject": "help",
        "body": "you're my only hope",
    }
    mailer.send(**message_data)
    assert len(mailer.messages) == 1
    message = mailer.messages[0]
    assert message_data["recipients"][0] in message
    assert message_data["subject"] in message
    assert message_data["body"] in message