red-mail/redmail/test/email/test_send.py

from email.message import EmailMessage
import pytest

from redmail import EmailSender, send_email

class MockServer:

    def __init__(self, host, port):
        self.host = host
        self.port = port
        self.is_login = False

    def starttls(self):
        return

    def login(self, user=None, password=None):
        self.is_login = True
        self.user = user
        self.password = password
        return

    def send_message(self, msg):
        return

    def quit(self):
        return

def test_send():
    email = EmailSender(host="localhost", port=0, cls_smtp=MockServer)
    assert email.connection is None

    msg = email.send(
        subject="An example",
        receivers=['me@example.com']
    )
    assert isinstance(msg, EmailMessage)
    assert email.connection is None

def test_send_with_user():
    email = EmailSender(host="localhost", port=0, username="myuser", password="1234", cls_smtp=MockServer)
    assert email.connection is None
    assert email.username == "myuser"
    assert email.password == "1234"

    msg = email.send(
        subject="An example",
        receivers=['me@example.com']
    )
    assert isinstance(msg, EmailMessage)
    assert email.connection is None

    # Testing the server
    server = email.get_server()
    assert server.user == "myuser"
    assert server.password == "1234"

def test_send_multi():
    email = EmailSender(host="localhost", port=0, cls_smtp=MockServer)

    assert email.connection is None
    with email:
        assert email.connection is not None
        msg = email.send(
            subject="An example",
            receivers=['me@example.com']
        )
        assert isinstance(msg, EmailMessage)
        assert email.connection is not None
        msg = email.send(
            subject="An example",
            receivers=['me@example.com']
        )
        assert isinstance(msg, EmailMessage)
        assert email.connection is not None
    assert email.connection is None

def test_send_function():
    # This should fail but we test everything else goes through
    with pytest.raises(ConnectionRefusedError):
        send_email(host="localhost", port=0, subject="An example")