def role_delete(self, companyname, rolename, expectation):
self.start_test(locals())
self.login_company_admin()
self.choose_company(companyname=companyname)
self.role_manage_enter()
found_role = self.role_search(rolename)
if expectation != "pass":
self.assert_true(found_role == {}, "Should not find role: " + rolename)
else:
self.assertTrue(found_role != {}, "Can not find role: " + rolename)
all_deleted = False
authority = Authority()
while not all_deleted:
delete_selector = found_role["delete_selector"]
self.role_delete_action(delete_selector)
self.remove_authority_record(
email_list=authority.get_email_list(bycolumn="role", value=rolename, company=companyname),
role={companyname: [rolename]})
# self.remove_role_record(company=companyname, custome_role=[rolename])
time.sleep(6) # wait for refresh
found_role = self.role_search(rolename)
if found_role == {}:
all_deleted = True
self.logout(COMPANYADMIN_ACCOUNT)
self.end_test(self._testMethodName)
def site_delete(self, companyname, sitename, expectation):
self.start_test(locals())
self.login_company_admin()
self.choose_company(companyname=companyname)
self.site_manage_enter()
time.sleep(10) # too slow
found_site = self.site_search(sitename)
if expectation != "pass":
self.assert_true(found_site == {},
"Should not find site: " + sitename)
else:
self.assertTrue(found_site != {}, "Can not find site: " + sitename)
all_deleted = False
authority = Authority()
while not all_deleted:
delete_selector = found_site["delete_selector"]
self.site_delete_action(delete_selector)
self.remove_authority_record(
email_list=authority.get_email_list(bycolumn="site",
value=sitename,
company=companyname),
site={companyname: [sitename]})
# after deleteļ¼ user list does not show
self.refresh_page()
self.site_manage_enter()
found_site = self.site_search(sitename)
if found_site == {}:
all_deleted = True
self.logout(COMPANYADMIN_ACCOUNT)
self.end_test(self._testMethodName)
def verify_companyadmin_password_email(self):
self.start_test(locals())
# get email_list
authority = Authority()
email_list = authority.get_email_list(bycolumn="role",
value=ROLE_COMPANY_ADMIN)
print("Start to check email and password for: " + str(email_list))
valuedict = self.verify_password_email(email_list)
self.assert_true(
valuedict != [],
"Did not send email or password to: " + str(email_list))
for item in valuedict:
authority.add_authority(email=item["email"],
password=item["password"])
print(str(authority.df))
authority.commit()
self.end_test(self._testMethodName)
def company_update(self, companyname, valuedict, expectation):
self.start_test(locals())
companyname, key, value, expectation = self.process_update_input(
companyname, valuedict, expectation)
self.login_sysadmin()
found_company = self.company_search(companyname)
if found_company == {}:
self.assertTrue(False, "Cannot find company: " + companyname)
found_correct = found_company["companyname"] == companyname \
and self.get_text(found_company["name_selector"]) == found_company["companyname"] \
and self.get_text(found_company["id_selector"]) == found_company["id"]
print("Found info: \n" + str(found_company))
self.assert_true(
found_correct,
"Found info doesn't match what show on user interface \n" +
"Try to find: " + companyname + "\n But find " +
str(found_company))
edit_selector = str(found_company["edit_selector"])
if True:
self.company_update_start(edit_selector)
authority = Authority()
company = companyname
if "companyname" in key:
company = str(value["companyname"])
self.company_create_update_set_companyname(company)
email_list = authority.get_email_list(bycolumn="company",
value=companyname)
for email in email_list:
authority.update_authority_list_value_column(
email=email,
column="company",
value=[companyname],
action="remove")
authority.update_authority_list_value_column(
email=email,
column="company",
value=[company],
action="add")
if "status" in key:
status = str(value["status"])
self.company_create_update_set_status(status)
else:
status = None
if "email" in key:
email_list = str(value["email"])
email_list = utilities.str_to_list(email_list)
if expectation == "illegal_email_msg":
fail_msg = "The email address is invalid"
else:
fail_msg = None
self.company_create_update_set_email(email_list=email_list,
fail_msg=fail_msg,
expectation=expectation)
# deal with user profile
# email list connected with target company
existing_connected_email_list = authority.get_email_list(
bycolumn="company", value=company)
to_add_connection_email_list = [
i for i in email_list
if i not in existing_connected_email_list
]
print("To_add_connection_email_list: " +
str(to_add_connection_email_list))
for email in to_add_connection_email_list:
authority.add_authority(email=email, company=[company])
to_delete_email_list = [
i for i in existing_connected_email_list
if i not in email_list
]
print("To_delete_email_list: " +
str(to_delete_email_list)) #remove connection
for email in to_delete_email_list:
authority.update_authority_list_value_column(
email=email,
column="company",
value=[company],
action="remove")
else:
email_list = None
self.company_create_update_submit(expectation)
if expectation == "pass":
self.verify_company_info(company, status, email_list)
print("start to update user profile with: \n" +
str(authority.df))
authority.commit()
self.logout()
self.end_test(self._testMethodName)