def test_sends_smtp_email_exceptions(self): """ Call notifications.send_email_smtp when it cannot connect to smtp server (socket.error) starttls. """ smtp_kws = { "host": "my.smtp.local", "port": 999, "local_hostname": "ptms", "timeout": 1200 } with mock.patch('smtplib.SMTP') as SMTP: with mock.patch( 'luigi.notifications.generate_email') as generate_email: SMTP.side_effect = socket.error() generate_email.return_value \ .as_string.return_value = self.mocked_email_msg try: notifications.send_email_smtp(*self.notification_args) except socket.error: self.fail( "send_email_smtp() raised expection unexpectedly") SMTP.assert_called_once_with(**smtp_kws) self.assertEqual(notifications.generate_email.called, False) self.assertEqual(SMTP.sendemail.called, False)
def test_sends_smtp_email_without_tls(self): """ Call notifications.send_email_smtp with fixture parameters with no_tls set to True and check that sendmail is properly called without also calling starttls. """ smtp_kws = { "host": "my.smtp.local", "port": 999, "local_hostname": "ptms", "timeout": 1200 } with mock.patch('smtplib.SMTP') as SMTP: with mock.patch( 'luigi.notifications.generate_email') as generate_email: generate_email.return_value \ .as_string.return_value = self.mocked_email_msg notifications.send_email_smtp(*self.notification_args) SMTP.assert_called_once_with(**smtp_kws) self.assertEqual(SMTP.return_value.starttls.called, False) SMTP.return_value.login.assert_called_once_with( "Robin", "dooH") SMTP.return_value.sendmail \ .assert_called_once_with(self.sender, self.recipients, self.mocked_email_msg)
def test_sends_smtp_email_exceptions(self): """ Call notificaions.send_email_smtp when it cannot connect to smtp server (socket.error) starttls. """ smtp_kws = {"host": "my.smtp.local", "port": 999, "local_hostname": "ptms", "timeout": 1200} with mock.patch('smtplib.SMTP') as SMTP: with mock.patch('luigi.notifications.generate_email') as generate_email: SMTP.side_effect = socket.error() generate_email.return_value \ .as_string.return_value = self.mocked_email_msg try: notifications.send_email_smtp(configuration.get_config(), *self.notification_args) except socket.error: self.fail("send_email_smtp() raised expection unexpectedly") SMTP.assert_called_once_with(**smtp_kws) self.assertEqual(notifications.generate_email.called, False) self.assertEqual(SMTP.sendemail.called, False)
def test_sends_smtp_email(self): """ Call notificaions.send_email_smtp with fixture parameters and check that sendmail is properly called. """ smtp_kws = { "host": "my.smtp.local", "port": 999, "local_hostname": "ptms", "timeout": 1200 } with mock.patch('smtplib.SMTP') as SMTP: with mock.patch( 'luigi.notifications.generate_email') as generate_email: generate_email.return_value\ .as_string.return_value = self.mocked_email_msg notifications.send_email_smtp(configuration.get_config(), *self.notification_args) SMTP.assert_called_once_with(**smtp_kws) SMTP.return_value.login.assert_called_once_with( "Robin", "dooH") SMTP.return_value.sendmail\ .assert_called_once_with(self.sender, self.recipients, self.mocked_email_msg)
def test_sends_smtp_email_without_tls(self): """ Call notificaions.send_email_smtp with fixture parameters with smtp_without_tls set to True and check that sendmail is properly called without also calling starttls. """ smtp_kws = {"host": "my.smtp.local", "port": 999, "local_hostname": "ptms", "timeout": 1200} with mock.patch('smtplib.SMTP') as SMTP: with mock.patch('luigi.notifications.generate_email') as generate_email: generate_email.return_value \ .as_string.return_value = self.mocked_email_msg notifications.send_email_smtp(configuration.get_config(), *self.notification_args) SMTP.assert_called_once_with(**smtp_kws) self.assertEqual(SMTP.return_value.starttls.called, False) SMTP.return_value.login.assert_called_once_with("Robin", "dooH") SMTP.return_value.sendmail \ .assert_called_once_with(self.sender, self.recipients, self.mocked_email_msg)