1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
|
# Copyright 2014-2021 The aiosmtpd Developers
# SPDX-License-Identifier: Apache-2.0
"""Test SMTP over SSL/TLS."""
from email.mime.text import MIMEText
from smtplib import SMTP, SMTP_SSL
from typing import Generator, Union
import pytest
from aiosmtpd.controller import Controller
from aiosmtpd.testing.helpers import ReceivingHandler
from aiosmtpd.testing.statuscodes import SMTP_STATUS_CODES as S
from .conftest import Global
@pytest.fixture
def ssl_controller(
get_controller, ssl_context_server
) -> Generator[Controller, None, None]:
handler = ReceivingHandler()
controller = get_controller(handler, ssl_context=ssl_context_server)
controller.start()
Global.set_addr_from(controller)
#
yield controller
#
controller.stop()
@pytest.fixture
def smtps_client(ssl_context_client) -> Generator[Union[SMTP_SSL, SMTP], None, None]:
context = ssl_context_client
with SMTP_SSL(*Global.SrvAddr, context=context) as client:
yield client
class TestSMTPS:
def test_smtps(self, ssl_controller, smtps_client):
sender = "sender@example.com"
recipients = ["rcpt1@example.com"]
resp = smtps_client.helo("example.com")
assert resp == S.S250_FQDN
results = smtps_client.send_message(MIMEText("hi"), sender, recipients)
assert results == {}
handler: ReceivingHandler = ssl_controller.handler
assert len(handler.box) == 1
envelope = handler.box[0]
assert envelope.mail_from == sender
assert envelope.rcpt_tos == recipients
|
