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
53
54
55
56
57
58
59
60
61
62
63
64
|
import smtplib
from unittest import mock
import pytest
from health_check.contrib.mail.backends import MailHealthCheck
class TestMailHealthCheck:
"""Test mail health check."""
@mock.patch("health_check.contrib.mail.backends.get_connection")
def test_mail_conn_ok(self, mocked_backend):
# instantiates the class
mail_health_checker = MailHealthCheck()
# invokes the method check_status()
mail_health_checker.check_status()
assert len(mail_health_checker.errors) == 0, mail_health_checker.errors
# mock assertions
assert mocked_backend.return_value.timeout == 15
mocked_backend.return_value.open.assert_called_once()
mocked_backend.return_value.close.assert_called_once()
@mock.patch("health_check.contrib.mail.backends.get_connection")
@pytest.mark.parametrize(
argnames=[
"error",
"expected_msg",
],
argvalues=[
[
smtplib.SMTPException(),
"Failed to open connection with SMTP server",
],
[
ConnectionRefusedError(),
"Connection refused error",
],
[
Exception(),
f"Unknown error {Exception}"
],
],
)
def test_mail_conn_refused(
self,
mocked_backend,
error: Exception,
expected_msg: str,
):
"""Test case then connection refused."""
mocked_backend.return_value.open.side_effect = error
# instantiates the class
mail_health_checker = MailHealthCheck()
# invokes the method check_status()
mail_health_checker.check_status()
assert len(mail_health_checker.errors) == 1, mail_health_checker.errors
assert mail_health_checker.errors[0].message == expected_msg
# mock assertions
mocked_backend.return_value.open.assert_called_once()
|
