Exemple #1
0
 def test_notify_not_login(self, mock_fqdn, mock_connect, mock_ehlo, mock_starttls, mock_login, mock_log, mock_error):
     res = Operator()
     mock_fqdn.return_value = '1.1.1.1'
     mock_connect.return_value = (220, "ok")
     mock_login.side_effect = Exception('could not login')
     res.result = 'Passed'
     mfile = os.path.join(os.path.dirname(__file__), 'configs', 'mail.yml')
     mail_file = open(mfile, 'r')
     mail_file = yaml.load(mail_file)   #smtplib.SMTP#connectsocket.getfqdn, Loader=yaml.FullLoader
     passwd = mail_file['passwd']
     notf = Notification()
     notf.notify(mail_file, self.hostname, passwd, res)
     mock_log.assert_called()
     mock_error.assert_called_with('\x1b[31mERROR occurred: could not login', extra={'hostname': '10.209.12.121', 'hostame': None})
 def test_notify_send_mail(self, mock_fqdn, mock_connect, mock_ehlo,
                           mock_starttls, mock_login, mock_quit, mock_log,
                           mock_error, mock_send):
     res = Operator()
     mock_send.side_effect = Exception('not able to send mail')
     mock_fqdn.return_value = '1.1.1.1'
     mock_connect.return_value = (220, "ok")
     res.result = 'Passed'
     mfile = os.path.join(os.path.dirname(__file__), 'configs', 'mail.yml')
     mail_file = open(mfile, 'r')
     mail_file = yaml.load(mail_file)  #smtplib.SMTP#connectsocket.getfqdn
     passwd = mail_file['passwd']
     notf = Notification()
     notf.notify(mail_file, self.hostname, passwd, res)
     mock_log.assert_called()
     mock_error.assert_called_with(
         '\x1b[31mERROR!!  in sending mail: not able to send mail',
         extra={
             'hostname': '10.209.12.121',
             'hostame': None
         })
     mock_quit.assert_called()