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
|
#!/usr/bin/env python3
# Copyright (C) 2024 The Qt Company Ltd.
# SPDX-License-Identifier: LicenseRef-Qt-Commercial OR LGPL-3.0-only OR GPL-2.0-only OR GPL-3.0-only
import unittest
import mock
import email_alert
from email.message import EmailMessage
class TestEmailAlert(unittest.TestCase):
@mock.patch('email_alert.GerritRestAPI')
def test_get_authors_ok_case(self, gerrit_rest_mock):
gerrit_rest_mock().get().__getitem__().__getitem__.return_value = "test@email.com"
authors = email_alert.get_authors(
"codereview.qt-project.org",
"qt/qtdeclarative",
["9a66e7981202f9b465934e3dde43bbd44c54b4f8"])
self.assertEqual(len(authors), 1)
self.assertEqual(authors[0], "test@email.com")
@mock.patch('email_alert.smtplib')
def test_send_email(self, smtp_mock):
email_alert.send_email(
'smtp.qt.io', "sender@qt.io", ["test@qt.io"], "cc@qt.io", "test subject", "test message")
msg = smtp_mock.SMTP().send_message.call_args_list[0].args[0]
self.assertEqual(msg.get_content(), "test message\n")
self.assertEqual(msg['Subject'], "test subject")
self.assertEqual(msg['From'], "sender@qt.io")
self.assertEqual(msg['Cc'], "cc@qt.io")
self.assertEqual(msg['To'], "test@qt.io")
smtp_mock.SMTP.assert_called()
smtp_mock.SMTP().quit.assert_called()
@mock.patch('email_alert.smtplib')
def test_send_email_multiple_receivers(self, smtp_mock):
email_alert.send_email(
'smtp.qt.io', "sender@qt.io", ["test@qt.io", "test2@qt.io"], "cc@qt.io", "test subject", "test message")
msg = smtp_mock.SMTP().send_message.call_args_list[0].args[0]
self.assertEqual(msg.get_content(), "test message\n")
self.assertEqual(msg['Subject'], "test subject")
self.assertEqual(msg['From'], "sender@qt.io")
self.assertEqual(msg['Cc'], "cc@qt.io")
self.assertEqual(msg['To'], "test@qt.io, test2@qt.io")
smtp_mock.SMTP.assert_called()
smtp_mock.SMTP().quit.assert_called()
|
