class EmailNotification:
def __init__(self, arguments):
self.args = arguments
self.data_manager = DataManager(arguments)
self.report_builder = ReportBuilder()
self.smtp_config = {
'host': arguments['smpt_host'],
'port': arguments['smpt_port'],
'user': arguments['smpt_user'],
'password': arguments['smpt_password']
}
def email_notification(self):
tests_data, last_test_data, baseline, violation, compare_with_thresholds = self.data_manager.get_api_test_info(
)
email_body, charts, date = self.report_builder.create_api_email_body(
tests_data, last_test_data, baseline,
self.args['comparison_metric'], violation, compare_with_thresholds)
self.send_email(self.smtp_config, self.args['user_list'], email_body,
charts, date)
def ui_email_notification(self):
tests_data, last_test_data = self.data_manager.get_ui_test_info()
email_body, charts, date = self.report_builder.create_ui_email_body(
tests_data, last_test_data)
self.send_email(self.smtp_config, self.args['user_list'], email_body,
charts, date)
def send_email(self, smtp_config, user_list, email_body, charts, date):
s = smtplib.SMTP_SSL(host=smtp_config['host'],
port=int(smtp_config['port']))
s.ehlo()
s.login(smtp_config['user'], smtp_config['password'])
subject = "[" + str(self.args['notification_type']) + "] "
subject += "Test results for \"" + str(self.args['test'])
subject += "\". Users count: " + str(
self.args['users']) + ". From " + str(date) + "."
for user in user_list:
if all(i in user for i in ["<mailto:", "|"]):
user = user.split("|")[1].replace(">", "").replace("<", "")
msg_root = MIMEMultipart('related')
msg_root['Subject'] = subject
msg_root['From'] = smtp_config['user']
msg_root['To'] = user
msg_alternative = MIMEMultipart('alternative')
msg_alternative.attach(MIMEText(email_body, 'html'))
msg_root.attach(msg_alternative)
for chart in charts:
msg_root.attach(chart)
s.sendmail(smtp_config['user'], user, msg_root.as_string())
s.quit()
class ApiEmailNotification:
def __init__(self, arguments):
self.args = arguments
self.data_manager = DataManager(arguments, arguments["galloper_url"],
arguments["token"],
arguments["project_id"])
self.report_builder = ReportBuilder()
def email_notification(self):
tests_data, last_test_data, baseline, violation, compare_with_thresholds = self.data_manager.get_api_test_info(
)
email_body, charts, date = self.report_builder.create_api_email_body(
self.args, tests_data, last_test_data, baseline,
self.args['comparison_metric'], violation, compare_with_thresholds)
subject = "[" + str(self.args['notification_type']) + "] "
subject += "Test results for \"" + str(self.args['test'])
subject += "\". Users count: " + str(
self.args['users']) + ". From " + str(date) + "."
return Email(self.args['test'], subject, self.args['user_list'],
email_body, charts, date)
