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()