def test_101_email_admin_override_custom_report(self):
"""
1. Use reportuser
2. Reportuser overrides admin user address.
3. Custom report with test not in default.
"""
if (not can_relay):
raise unittest.SkipTest('Unable to relay through ' +
global_functions.TEST_SERVER_HOST)
if runtests.quick_tests_only:
raise unittest.SkipTest('Skipping a time consuming test')
# Create settings to receive test_email_address
configure_mail_relay()
# add administrator
adminsettings = uvmContext.adminManager().getSettings()
orig_adminsettings = copy.deepcopy(adminsettings)
adminsettings['users']['list'].append(
create_admin_user(useremail=test_email_address))
uvmContext.adminManager().setSettings(adminsettings)
settings = app.getSettings()
# add custom template with a test not in daily reports
settings["emailTemplates"]["list"] = settings["emailTemplates"][
"list"][:1]
settings["emailTemplates"]["list"].append(create_email_template())
# add report user with test_email_address
settings["reportsUsers"]["list"] = settings["reportsUsers"]["list"][:1]
settings["reportsUsers"]["list"].append(
create_reports_user(profile_email=test_email_address,
email_template_id=2))
app.setSettings(settings)
# send email
subprocess.call([
global_functions.get_prefix() +
"/usr/share/untangle/bin/reports-generate-fixed-reports.py"
],
stdout=subprocess.PIPE,
stderr=subprocess.PIPE)
# look for email
email_found = fetch_email(
"/tmp/test_101_email_admin_override_custom_report_file",
test_email_address)
if email_found:
email_context_found1 = remote_control.run_command(
"grep -i 'Custom Report' /tmp/test_101_email_admin_override_custom_report_file 2>&1",
stdout=True)
email_context_found2 = remote_control.run_command(
"grep -i 'Administration-VWuRol5uWw' /tmp/test_101_email_admin_override_custom_report_file 2>&1",
stdout=True)
# restore
uvmContext.adminManager().setSettings(orig_adminsettings)
assert (email_found)
assert ((email_context_found1) and (email_context_found2))
def test_102_admin_login_event(self):
uvmContext.adminManager().logAdminLoginEvent( "admin", True, "127.0.1.1", True, 'X' )
events = global_functions.get_events('Administration','Admin Login Events',None,10)
assert(events != None)
for i in events.get('list'):
print(i)
found = global_functions.check_events( events.get('list'), 10,
'client_addr', "127.0.1.1",
'reason', 'X',
'local', True,
'succeeded', True,
'login', 'admin' )
assert( found )
def test_102_admin_login_event(self):
uvmContext.adminManager().logAdminLoginEvent( "admin", True, "127.0.1.1", True, 'X' )
events = global_functions.get_events('Administration','Admin Login Events',None,10)
assert(events != None)
for i in events.get('list'):
print(i)
found = global_functions.check_events( events.get('list'), 10,
'client_addr', "127.0.1.1",
'reason', 'X',
'local', True,
'succeeded', True,
'login', 'admin' )
assert( found )
def test_060_customized_email_alert(self):
"""Create custom email template and verify alert email is received correctly"""
#get settings, backup original settings
email_settings = uvmContext.eventManager().getSettings()
orig_email_settings = copy.deepcopy(email_settings)
admin_settings = uvmContext.adminManager().getSettings()
orig_admin_settings = copy.deepcopy(admin_settings)
#change admin email to verify sent email
new_admin_email = global_functions.random_email()
admin_settings["users"]["list"][0]["emailAddress"] = new_admin_email
uvmContext.adminManager().setSettings(admin_settings)
#set custom email template subject and body
new_email_subject = "NEW EMAIL SUBJECT TEST"
new_email_body = "NEW EMAIL BODY TEST"
email_settings["emailSubject"] = new_email_subject
email_settings["emailBody"] = new_email_body
#set new alert rule for easy trigger of email
new_rule = create_alert_rule("test alert rule", "class", "=", "*SessionEvent*", "localAddr", "=", "*"+remote_control.client_ip+"*", sendEmail=True)
email_settings['alertRules']['list'].append(new_rule)
#set new settings
uvmContext.eventManager().setSettings(email_settings)
#send a session
remote_control.is_online()
time.sleep(4)
#check email sent is correct
emailFound = False
timeout = 5
alertEmail = ""
while not emailFound and timeout > 0:
timeout -= 1
time.sleep(1)
alertEmail = remote_control.run_command("wget -q --timeout=5 -O - http://test.untangle.com/cgi-bin/getEmail.py?toaddress=" + new_admin_email + " 2>&1 | grep TEST" ,stdout=True)
if (alertEmail != ""):
emailFound = True
#set settings back
uvmContext.eventManager().setSettings(orig_email_settings)
uvmContext.adminManager().setSettings(orig_admin_settings)
assert(emailFound)
def test_101_email_admin_override_custom_report(self):
"""
1. Use reportuser
2. Reportuser overrides admin user address.
3. Custom report with test not in default.
"""
if (not can_relay):
raise unittest.SkipTest('Unable to relay through ' + global_functions.TEST_SERVER_HOST)
if runtests.quick_tests_only:
raise unittest.SkipTest('Skipping a time consuming test')
# Create settings to receive test_email_address
configure_mail_relay()
# add administrator
adminsettings = uvmContext.adminManager().getSettings()
orig_adminsettings = copy.deepcopy(adminsettings)
adminsettings['users']['list'].append(create_admin_user(useremail=test_email_address))
uvmContext.adminManager().setSettings(adminsettings)
settings = app.getSettings()
# add custom template with a test not in daily reports
settings["emailTemplates"]["list"] = settings["emailTemplates"]["list"][:1]
settings["emailTemplates"]["list"].append(create_email_template())
# add report user with test_email_address
settings["reportsUsers"]["list"] = settings["reportsUsers"]["list"][:1]
settings["reportsUsers"]["list"].append(create_reports_user(profile_email=test_email_address, email_template_id=2))
app.setSettings(settings)
# send email
subprocess.call([global_functions.get_prefix()+"/usr/share/untangle/bin/reports-generate-fixed-reports.py"],stdout=subprocess.PIPE,stderr=subprocess.PIPE)
# look for email
email_found = fetch_email( "/tmp/test_101_email_admin_override_custom_report_file", test_email_address )
if email_found:
email_context_found1 = remote_control.run_command("grep -i 'Custom Report' /tmp/test_101_email_admin_override_custom_report_file 2>&1", stdout=True)
email_context_found2 = remote_control.run_command("grep -i 'Administration-VWuRol5uWw' /tmp/test_101_email_admin_override_custom_report_file 2>&1", stdout=True)
# restore
uvmContext.adminManager().setSettings(orig_adminsettings)
assert(email_found)
assert((email_context_found1) and (email_context_found2))
def test_020_about_info(self):
uid = uvmContext.getServerUID()
match = re.search(r'\w{4}-\w{4}-\w{4}.\w{4}', uid)
assert( match )
kernel = uvmContext.adminManager().getKernelVersion()
match = re.search(r'\d.*', kernel)
assert(match)
reboot_count = uvmContext.adminManager().getRebootCount()
match = re.search(r'\d{1,2}', reboot_count)
assert(match)
num_hosts = str(uvmContext.hostTable().getCurrentActiveSize())
match = re.search(r'\d{1,2}', num_hosts)
assert(match)
max_num_hosts = str(uvmContext.hostTable().getMaxActiveSize())
match = re.search(r'\d{1,2}', max_num_hosts)
assert(match)
def test_020_about_info(self):
uid = uvmContext.getServerUID()
match = re.search(r'\w{4}-\w{4}-\w{4}.\w{4}', uid)
assert( match )
kernel = uvmContext.adminManager().getKernelVersion()
match = re.search(r'\d.*', kernel)
assert(match)
reboot_count = uvmContext.adminManager().getRebootCount()
match = re.search(r'\d{1,2}', reboot_count)
assert(match)
num_hosts = str(uvmContext.hostTable().getCurrentActiveSize())
match = re.search(r'\d{1,2}', num_hosts)
assert(match)
max_num_hosts = str(uvmContext.hostTable().getMaxActiveSize())
match = re.search(r'\d{1,2}', max_num_hosts)
assert(match)
def test_103_email_report_verify_apps(self):
"""
1) Install all apps
2) Generate a report
3) Verify that the emailed report contains a section for each app
"""
global app, apps_list, apps_name_list
if (not can_relay):
raise unittest.SkipTest('Unable to relay through ' +
global_functions.TEST_SERVER_HOST)
if runtests.quick_tests_only:
raise unittest.SkipTest('Skipping a time consuming test')
# create settings to receive test_email_address
configure_mail_relay()
# add administrator
adminsettings = uvmContext.adminManager().getSettings()
orig_adminsettings = copy.deepcopy(adminsettings)
adminsettings['users']['list'].append(
create_admin_user(useremail=test_email_address))
uvmContext.adminManager().setSettings(adminsettings)
# clear all report users
settings = self._app.getSettings()
settings["reportsUsers"]["list"] = settings["reportsUsers"]["list"][:1]
self._app.setSettings(settings)
# install all the apps that aren't already installed
system_stats = uvmContext.metricManager().getStats()
# print system_stats
system_memory = system_stats['systemStats']['MemTotal']
if (int(system_memory) < 2200000000
): # don't use high memory apps in devices with 2G or less.
apps_list = apps_list_short
apps_name_list = apps_name_list_short
apps = []
for name in apps_list:
if (uvmContext.appManager().isInstantiated(name)):
print("App %s already installed" % name)
else:
apps.append(uvmContext.appManager().instantiate(
name, default_policy_id))
# create some traffic
result = remote_control.is_online(tries=1)
# flush out events
self._app.flushEvents()
# send emails
subprocess.call([
global_functions.get_prefix() +
"/usr/share/untangle/bin/reports-generate-fixed-reports.py"
],
stdout=subprocess.PIPE,
stderr=subprocess.PIPE)
# look for email
email_found = fetch_email("/tmp/test_103_email_report_admin_file",
test_email_address)
# look for all the appropriate sections in the report email
results = []
if email_found:
for str in apps_name_list:
results.append(
remote_control.run_command(
"grep -q -i '%s' /tmp/test_103_email_report_admin_file 2>&1"
% str))
# restore
uvmContext.adminManager().setSettings(orig_adminsettings)
# remove apps that were installed above
for a in apps:
uvmContext.appManager().destroy(a.getAppSettings()["id"])
assert (email_found)
for result in results:
assert (result == 0)
def test_102_email_admin_override_custom_report_mobile(self):
"""
1. Use reportuser
2. Reportuser overrides admin user address.
3. Custom report with test not in default.
"""
if (not can_relay):
raise unittest.SkipTest('Unable to relay through ' +
global_functions.TEST_SERVER_HOST)
if runtests.quick_tests_only:
raise unittest.SkipTest('Skipping a time consuming test')
# Create settings to receive test_email_address
configure_mail_relay()
subprocess.call(
"rm /tmp/test_102_email_admin_override_custom_report_mobile_file > /dev/null 2>&1",
shell=True)
# add administrator
adminsettings = uvmContext.adminManager().getSettings()
orig_adminsettings = copy.deepcopy(adminsettings)
adminsettings['users']['list'].append(
create_admin_user(useremail=test_email_address))
uvmContext.adminManager().setSettings(adminsettings)
settings = app.getSettings()
# add custom template with a test not in daily reports
settings["emailTemplates"]["list"] = settings["emailTemplates"][
"list"][:1]
settings["emailTemplates"]["list"].append(
create_email_template(mobile=True))
# add report user with test_email_address
settings["reportsUsers"]["list"] = settings["reportsUsers"]["list"][:1]
settings["reportsUsers"]["list"].append(
create_reports_user(profile_email=test_email_address,
email_template_id=2))
app.setSettings(settings)
# send email
subprocess.call([
global_functions.get_prefix() +
"/usr/share/untangle/bin/reports-generate-fixed-reports.py"
],
stdout=subprocess.PIPE,
stderr=subprocess.PIPE)
# look for email
email_found = fetch_email(
"/tmp/test_102_email_admin_override_custom_report_mobile_file",
test_email_address)
if email_found:
email_context_found1 = remote_control.run_command(
"grep -i 'Custom Report' /tmp/test_102_email_admin_override_custom_report_mobile_file 2>&1",
stdout=True)
email_context_found2 = remote_control.run_command(
"grep -i 'Administration-VWuRol5uWw' /tmp/test_102_email_admin_override_custom_report_mobile_file 2>&1",
stdout=True)
# restore
uvmContext.adminManager().setSettings(orig_adminsettings)
assert (email_found)
assert ((email_context_found1) and (email_context_found2))
# Verify that all images are less than 3502350.
# copy mail from remote client
subprocess.call(
"scp -q -i %s testshell@%s:/tmp/test_102_email_admin_override_custom_report_mobile_file /tmp/"
% (remote_control.host_key_file, remote_control.client_ip),
shell=True)
fp = open(
"/tmp/test_102_email_admin_override_custom_report_mobile_file")
email_string = fp.read()
fp.close()
# Delete the first line as it is blank and throws off the parser
email_string = '\n'.join(email_string.split('\n')[1:])
msg = email.message_from_string(email_string)
mime_content_ids = []
for part in msg.walk():
if part.get_content_maintype() == "image":
# print("Image found")
for index, key in enumerate(part.keys()):
if key == "Content-ID" and "untangle.int" in part.values(
)[index]:
email_image = part.get_payload(decode=True)
im = Image.open(BytesIO(email_image))
(image_width, image_height) = im.size
print(
"Image %s width: %d height: %d" %
(part.values()[index], image_width, image_height))
assert (image_width <= 350 and image_height <= 350)
def test_100_email_report_admin(self):
"""
The "default" configuration test:
- Administrator email account gets
"""
if (not can_relay):
raise unittest.SkipTest('Unable to relay through ' +
global_functions.TEST_SERVER_HOST)
if runtests.quick_tests_only:
raise unittest.SkipTest('Skipping a time consuming test')
# create settings to receive test_email_address
configure_mail_relay()
subprocess.call(
"rm /tmp/test_100_email_report_admin_file > /dev/null 2>&1",
shell=True)
# add administrator
adminsettings = uvmContext.adminManager().getSettings()
orig_adminsettings = copy.deepcopy(adminsettings)
adminsettings['users']['list'].append(
create_admin_user(useremail=test_email_address))
uvmContext.adminManager().setSettings(adminsettings)
# clear all report users
settings = app.getSettings()
settings["reportsUsers"]["list"] = settings["reportsUsers"]["list"][:1]
app.setSettings(settings)
# send emails
subprocess.call([
global_functions.get_prefix() +
"/usr/share/untangle/bin/reports-generate-fixed-reports.py"
],
stdout=subprocess.PIPE,
stderr=subprocess.PIPE)
# look for email
email_found = fetch_email("/tmp/test_100_email_report_admin_file",
test_email_address)
email_context_found1 = ""
email_context_found2 = ""
if email_found:
email_context_found1 = remote_control.run_command(
"grep -i -e 'Reports:.*Daily.*' /tmp/test_100_email_report_admin_file 2>&1",
stdout=True)
email_context_found2 = remote_control.run_command(
"grep -i -e 'Content-Type: image/png; name=' /tmp/test_100_email_report_admin_file 2>&1",
stdout=True)
# restore
uvmContext.adminManager().setSettings(orig_adminsettings)
assert (email_found)
assert ((email_context_found1) and (email_context_found2))
## Verify that all images are intact.
# copy mail from remote client
subprocess.call(
"scp -q -i %s testshell@%s:/tmp/test_100_email_report_admin_file /tmp/"
% (remote_control.host_key_file, remote_control.client_ip),
shell=True)
fp = open("/tmp/test_100_email_report_admin_file")
email_string = fp.read()
fp.close()
# Delete the first line as it is blank and throws off the parser
email_string = '\n'.join(email_string.split('\n')[1:])
msg = email.message_from_string(email_string)
mime_content_ids = []
parser = ContentIdParser()
for part in msg.walk():
if part.get_content_maintype() == "image":
for index, key in enumerate(part.keys()):
if key == "Content-ID":
mime_content_ids.append(part.values()[index])
elif part.get_content_maintype() == "text":
parser.feed((part.get_payload(decode=True)).decode("utf-8"))
assert (len(parser.content_ids) == len(mime_content_ids))
def test_103_email_report_verify_apps(self):
"""
1) Install all apps
2) Generate a report
3) Verify that the emailed report contains a section for each app
"""
global app,apps_list,apps_name_list
if (not can_relay):
raise unittest.SkipTest('Unable to relay through ' + global_functions.TEST_SERVER_HOST)
if runtests.quick_tests_only:
raise unittest.SkipTest('Skipping a time consuming test')
# create settings to receive test_email_address
configure_mail_relay()
# add administrator
adminsettings = uvmContext.adminManager().getSettings()
orig_adminsettings = copy.deepcopy(adminsettings)
adminsettings['users']['list'].append(create_admin_user(useremail=test_email_address))
uvmContext.adminManager().setSettings(adminsettings)
# clear all report users
settings = app.getSettings()
settings["reportsUsers"]["list"] = settings["reportsUsers"]["list"][:1]
app.setSettings(settings)
# install all the apps that aren't already installed
system_stats = uvmContext.metricManager().getStats()
# print system_stats
system_memory = system_stats['systemStats']['MemTotal']
if (int(system_memory) < 2200000000): # don't use high memory apps in devices with 2G or less.
apps_list = apps_list_short
apps_name_list = apps_name_list_short
apps = []
for name in apps_list:
if (uvmContext.appManager().isInstantiated(name)):
print("App %s already installed" % name)
else:
apps.append( uvmContext.appManager().instantiate(name, default_policy_id) )
# create some traffic
result = remote_control.is_online(tries=1)
# flush out events
app.flushEvents()
# send emails
subprocess.call([global_functions.get_prefix()+"/usr/share/untangle/bin/reports-generate-fixed-reports.py"],stdout=subprocess.PIPE,stderr=subprocess.PIPE)
# look for email
email_found = fetch_email( "/tmp/test_103_email_report_admin_file", test_email_address )
# look for all the appropriate sections in the report email
results = []
if email_found:
for str in apps_name_list:
results.append(remote_control.run_command("grep -q -i '%s' /tmp/test_103_email_report_admin_file 2>&1"%str))
# restore
uvmContext.adminManager().setSettings(orig_adminsettings)
# remove apps that were installed above
for a in apps: uvmContext.appManager().destroy( a.getAppSettings()["id"] )
assert(email_found)
for result in results:
assert(result == 0)
def test_102_email_admin_override_custom_report_mobile(self):
"""
1. Use reportuser
2. Reportuser overrides admin user address.
3. Custom report with test not in default.
"""
if (not can_relay):
raise unittest.SkipTest('Unable to relay through ' + global_functions.TEST_SERVER_HOST)
if runtests.quick_tests_only:
raise unittest.SkipTest('Skipping a time consuming test')
# Create settings to receive test_email_address
configure_mail_relay()
# add administrator
adminsettings = uvmContext.adminManager().getSettings()
orig_adminsettings = copy.deepcopy(adminsettings)
adminsettings['users']['list'].append(create_admin_user(useremail=test_email_address))
uvmContext.adminManager().setSettings(adminsettings)
settings = app.getSettings()
# add custom template with a test not in daily reports
settings["emailTemplates"]["list"] = settings["emailTemplates"]["list"][:1]
settings["emailTemplates"]["list"].append(create_email_template(mobile=True))
# add report user with test_email_address
settings["reportsUsers"]["list"] = settings["reportsUsers"]["list"][:1]
settings["reportsUsers"]["list"].append(create_reports_user(profile_email=test_email_address, email_template_id=2))
app.setSettings(settings)
# send email
subprocess.call([global_functions.get_prefix()+"/usr/share/untangle/bin/reports-generate-fixed-reports.py"],stdout=subprocess.PIPE,stderr=subprocess.PIPE)
# look for email
email_found = fetch_email( "/tmp/test_102_email_admin_override_custom_report_mobile_file", test_email_address )
if email_found:
email_context_found1 = remote_control.run_command("grep -i 'Custom Report' /tmp/test_102_email_admin_override_custom_report_mobile_file 2>&1", stdout=True)
email_context_found2 = remote_control.run_command("grep -i 'Administration-VWuRol5uWw' /tmp/test_102_email_admin_override_custom_report_mobile_file 2>&1", stdout=True)
# restore
uvmContext.adminManager().setSettings(orig_adminsettings)
assert(email_found)
assert((email_context_found1) and (email_context_found2))
# Verify that all images are less than 3502350.
# copy mail from remote client
subprocess.call("scp -q -i %s testshell@%s:/tmp/test_102_email_admin_override_custom_report_mobile_file /tmp/" % (remote_control.host_key_file, remote_control.client_ip), shell=True)
fp = open("/tmp/test_102_email_admin_override_custom_report_mobile_file")
email_string = fp.read()
fp.close()
subprocess.call("rm /tmp/test_102_email_admin_override_custom_report_mobile_file", shell=True)
# Delete the first line as it is blank and throws off the parser
email_string = '\n'.join(email_string.split('\n')[1:])
msg = email.message_from_string(email_string)
mime_content_ids = []
for part in msg.walk():
if part.get_content_maintype() == "image":
# print("Image found")
for index, key in enumerate(part.keys()):
if key == "Content-ID" and "untangle.int" in part.values()[index]:
email_image = part.get_payload(decode=True)
im = Image.open(StringIO(email_image))
(image_width,image_height) = im.size
print("Image %s width: %d height: %d" % (part.values()[index], image_width, image_height))
assert(image_width <= 350 and image_height <= 350)
def test_100_email_report_admin(self):
"""
The "default" configuration test:
- Administrator email account gets
"""
if (not can_relay):
raise unittest.SkipTest('Unable to relay through ' + global_functions.TEST_SERVER_HOST)
if runtests.quick_tests_only:
raise unittest.SkipTest('Skipping a time consuming test')
# create settings to receive test_email_address
configure_mail_relay()
# add administrator
adminsettings = uvmContext.adminManager().getSettings()
orig_adminsettings = copy.deepcopy(adminsettings)
adminsettings['users']['list'].append(create_admin_user(useremail=test_email_address))
uvmContext.adminManager().setSettings(adminsettings)
# clear all report users
settings = app.getSettings()
settings["reportsUsers"]["list"] = settings["reportsUsers"]["list"][:1]
app.setSettings(settings)
# send emails
subprocess.call([global_functions.get_prefix()+"/usr/share/untangle/bin/reports-generate-fixed-reports.py"],stdout=subprocess.PIPE,stderr=subprocess.PIPE)
# look for email
email_found = fetch_email( "/tmp/test_100_email_report_admin_file", test_email_address )
email_context_found1 = ""
email_context_found2 = ""
if email_found:
email_context_found1 = remote_control.run_command("grep -i -e 'Reports:.*Daily.*' /tmp/test_100_email_report_admin_file 2>&1", stdout=True)
email_context_found2 = remote_control.run_command("grep -i -e 'Content-Type: image/png; name=' /tmp/test_100_email_report_admin_file 2>&1", stdout=True)
# restore
uvmContext.adminManager().setSettings(orig_adminsettings)
assert(email_found)
assert((email_context_found1) and (email_context_found2))
## Verify that all images are intact.
# copy mail from remote client
subprocess.call("scp -q -i %s testshell@%s:/tmp/test_100_email_report_admin_file /tmp/" % (remote_control.host_key_file, remote_control.client_ip), shell=True)
fp = open("/tmp/test_100_email_report_admin_file")
email_string = fp.read()
fp.close()
subprocess.call("rm /tmp/test_100_email_report_admin_file", shell=True)
# Delete the first line as it is blank and throws off the parser
email_string = '\n'.join(email_string.split('\n')[1:])
msg = email.message_from_string(email_string)
mime_content_ids = []
parser = ContentIdParser();
for part in msg.walk():
if part.get_content_maintype() == "image":
for index, key in enumerate(part.keys()):
if key == "Content-ID":
mime_content_ids.append(part.values()[index])
elif part.get_content_maintype() == "text":
parser.feed(part.get_payload(decode=True))
assert(len(parser.content_ids) == len(mime_content_ids))