Gecko [ 167.86.127.34 ]

Name	Size	Permission
__pycache__	[ DIR ]	drwxr-xr-x
display	[ DIR ]	drwxr-xr-x
testdata	[ DIR ]	drwxr-xr-x
__init__.py	20 B	-rw-r--r--
account_test.py	14.45 KB	-rw-r--r--
acme_util.py	3.18 KB	-rw-r--r--
auth_handler_test.py	24 KB	-rw-r--r--
cert_manager_test.py	28.07 KB	-rw-r--r--
cli_test.py	19.94 KB	-rw-r--r--
client_test.py	28.76 KB	-rw-r--r--
compat_test.py	736 B	-rw-r--r--
configuration_test.py	6.82 KB	-rw-r--r--
crypto_util_test.py	13.56 KB	-rw-r--r--
eff_test.py	5.94 KB	-rw-r--r--
error_handler_test.py	5.31 KB	-rw-r--r--
errors_test.py	1.8 KB	-rw-r--r--
hook_test.py	16.67 KB	-rw-r--r--
lock_test.py	3.84 KB	-rw-r--r--
log_test.py	14.95 KB	-rw-r--r--
main_test.py	82.52 KB	-rw-r--r--
notify_test.py	2.07 KB	-rw-r--r--
ocsp_test.py	6.27 KB	-rw-r--r--
renewal_test.py	4.18 KB	-rw-r--r--
renewupdater_test.py	5.32 KB	-rw-r--r--
reporter_test.py	2.73 KB	-rw-r--r--
reverter_test.py	18.7 KB	-rw-r--r--
storage_test.py	42.89 KB	-rw-r--r--
util.py	14.12 KB	-rw-r--r--
util_test.py	21.58 KB	-rw-r--r--

Code Editor : notify_test.py

"""Tests for certbot.notify."""
import socket
import unittest

import mock

class NotifyTests(unittest.TestCase):
    """Tests for the notifier."""

@mock.patch("certbot.notify.smtplib.LMTP")
    def test_smtp_success(self, mock_lmtp):
        from certbot.notify import notify
        lmtp_obj = mock.MagicMock()
        mock_lmtp.return_value = lmtp_obj
        self.assertTrue(notify("Goose", "auntrhody@example.com",
                               "The old grey goose is dead."))
        self.assertEqual(lmtp_obj.connect.call_count, 1)
        self.assertEqual(lmtp_obj.sendmail.call_count, 1)

@mock.patch("certbot.notify.smtplib.LMTP")
    @mock.patch("certbot.notify.subprocess.Popen")
    def test_smtp_failure(self, mock_popen, mock_lmtp):
        from certbot.notify import notify
        lmtp_obj = mock.MagicMock()
        mock_lmtp.return_value = lmtp_obj
        lmtp_obj.sendmail.side_effect = socket.error(17)
        proc = mock.MagicMock()
        mock_popen.return_value = proc
        self.assertTrue(notify("Goose", "auntrhody@example.com",
                               "The old grey goose is dead."))
        self.assertEqual(lmtp_obj.sendmail.call_count, 1)
        self.assertEqual(proc.communicate.call_count, 1)

@mock.patch("certbot.notify.smtplib.LMTP")
    @mock.patch("certbot.notify.subprocess.Popen")
    def test_everything_fails(self, mock_popen, mock_lmtp):
        from certbot.notify import notify
        lmtp_obj = mock.MagicMock()
        mock_lmtp.return_value = lmtp_obj
        lmtp_obj.sendmail.side_effect = socket.error(17)
        proc = mock.MagicMock()
        mock_popen.return_value = proc
        proc.communicate.side_effect = OSError("What we have here is a "
                                               "failure to communicate.")
        self.assertFalse(notify("Goose", "auntrhody@example.com",
                                "The old grey goose is dead."))
        self.assertEqual(lmtp_obj.sendmail.call_count, 1)
        self.assertEqual(proc.communicate.call_count, 1)

if __name__ == "__main__":
    unittest.main()  # pragma: no cover

${this.title}

Code Editor : notify_test.py