def send_zero_day_email(user_id):
tenant_id = session['tenant_id']
user_info = get_all_user_attributes(tenant_id, user_id)
logger.info('app.py - send_zero_day_email ' + tenant_id + ' ' +
user_info.user_name)
link_to_send = create_magic_link(tenant_id, user_info.user_name,
user_info.domain)
return_value = 'Failure'
if link_to_send[0:4] == 'http':
from_address = '*****@*****.**'
from_name = 'Welcome Admin'
subject = 'Welcome Email'
content_type = 'text/html'
email_body = html_email_builder(link_to_send, user_info.user_name,
user_info.display_name)
email_json = build_email_message(user_info.display_name,
user_info.mail_nickname, from_name,
from_address, subject, content_type,
email_body)
return_value = send_email(email_json)
logger.info('app.py - send_email - email_address - ' +
user_info.mail_nickname)
if return_value == '200' or return_value == '202':
return_value = 'Success'
return render_template('file_operation.html', status=return_value)
def test_get_magic_link(self):
curryware_tenant_id = 'b09245b3-60ec-43cf-8a2a-61425b107ac5'
user_id = '550a92fd-b220-42ca-befe-85d8ba3391bb'
user_info = get_all_user_attributes(curryware_tenant_id, user_id)
email_link = create_magic_link(curryware_tenant_id,
user_info.user_name, user_info.domain)
print(email_link)
is_http = email_link[0:4]
self.assertEqual(is_http, 'http')
def set_zero_day_configuration(group_id):
logger.info(
'app.py.set_zero_day/<group_id> - getting user in zero day group')
tenant_id = session['tenant_id']
all_users = get_users_in_group(tenant_id, group_id)
user_display_info = []
for current_user in all_users:
user_info = get_all_user_attributes(tenant_id, current_user.user_id)
user_display_info.append(user_info)
return render_template('zero_day_users.html', all_users=user_display_info)
def test_build_auth_link_email(self):
curryware_tenant_id = 'b09245b3-60ec-43cf-8a2a-61425b107ac5'
from_address = '*****@*****.**'
from_name = 'Welcome Admin'
subject = 'Welcome Email'
content_type = 'text/plain'
user_id = '550a92fd-b220-42ca-befe-85d8ba3391bb'
delete_magic_link_token(curryware_tenant_id, user_id)
user_info = get_all_user_attributes(curryware_tenant_id, user_id)
email_link = create_magic_link(curryware_tenant_id,
user_info.user_name, user_info.domain)
email_json = build_email_message(user_info.display_name,
user_info.email_address, from_name,
from_address, subject, content_type,
email_link)
print(email_json)
status_code = send_email(email_json)
self.assertEqual(status_code, '200')
def test_get_username_by_userid(self):
curryware_tenant_id = 'b09245b3-60ec-43cf-8a2a-61425b107ac5'
user_id = '550a92fd-b220-42ca-befe-85d8ba3391bb'
user_info = get_all_user_attributes(curryware_tenant_id, user_id)
self.assertEqual('scurry', user_info.user_name)