def remove_authority_record(self, email_list, password=None, status=None, company: list = None, site: dict = None, role: dict = None): print("At remove_authority_record \n " + str(locals())) if email_list is None: return email_list = list(set(email_list)) authority = Authority() for email in email_list: authority.remove_authority(email, company=company, site=site, role=role) print("At remove_authority_record, to save to file: ") print(str(authority.df)) authority.commit()
def add_authority_record(self, email_list, password=None, status=None, company: list = None, site: dict = None, role: dict = None): #print("At add_authority_record \n " + str(locals())) if email_list is None: return email_list = list(set(email_list)) authority = Authority() for email in email_list: authority.add_authority(email, password=password, status=status, company=company, site=site, role=role) #print(str(authority.df)) authority.commit()
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)