def setUp(self):
self.driver = AutomateDriverServer(choose='chrome')
self.base_url = self.driver.base_url
self.base_page = BasePageServer(self.driver, self.base_url)
self.login_page = LoginPage(self.driver, self.base_url)
self.cust_manage_basic_info_and_add_cust_page = CustManageBasicInfoAndAddCustPage(
self.driver, self.base_url)
self.account_center_page_details = AccountCenterDetailsPage(
self.driver, self.base_url)
self.cust_manage_cust_list_page = CustManageCustListPage(
self.driver, self.base_url)
self.cust_manage_my_dev_page = CustManageMyDevPage(
self.driver, self.base_url)
self.cust_manage_lower_account_page = CustManageLowerAccountPage(
self.driver, self.base_url)
self.account_center_page_navi_bar = AccountCenterNaviBarPage(
self.driver, self.base_url)
self.driver.set_window_max()
self.log_in_base = LogInBaseServer(self.driver, self.base_url)
self.cust_manage_page_read_csv = CustManagePageReadCsv()
self.search_sql = SearchSql()
self.connect_sql = ConnectSql()
self.driver.wait(1)
self.driver.clear_cookies()
self.driver.wait(1)
class TestCase402CustomerManagementLowerUserSearch(unittest.TestCase):
# 测试客户管理 搜索 客户
def setUp(self):
self.driver = AutomateDriverServer(choose='chrome')
self.base_url = self.driver.base_url
self.base_page = BasePageServer(self.driver, self.base_url)
self.login_page = LoginPage(self.driver, self.base_url)
self.cust_manage_basic_info_and_add_cust_page = CustManageBasicInfoAndAddCustPage(
self.driver, self.base_url)
self.account_center_page_details = AccountCenterDetailsPage(
self.driver, self.base_url)
self.cust_manage_cust_list_page = CustManageCustListPage(
self.driver, self.base_url)
self.cust_manage_my_dev_page = CustManageMyDevPage(
self.driver, self.base_url)
self.cust_manage_lower_account_page = CustManageLowerAccountPage(
self.driver, self.base_url)
self.account_center_page_navi_bar = AccountCenterNaviBarPage(
self.driver, self.base_url)
self.driver.set_window_max()
self.log_in_base = LogInBaseServer(self.driver, self.base_url)
self.cust_manage_page_read_csv = CustManagePageReadCsv()
self.search_sql = SearchSql()
self.connect_sql = ConnectSql()
self.driver.wait(1)
self.driver.clear_cookies()
self.driver.wait(1)
def tearDown(self):
self.driver.close_window()
self.driver.quit_browser()
def test_customer_management_lower_user_search(self):
# 打开途强在线首页-登录页
self.base_page.open_page()
# 登录
self.log_in_base.log_in()
# 点击账户中心
# self.account_center_page_navi_bar.click_account_center_button()
# current_account = self.log_in_base.get_log_in_account()
# 进入客户管理页面
current_handle = self.driver.get_current_window_handle()
self.cust_manage_basic_info_and_add_cust_page.enter_cust_manage()
self.base_page.change_windows_handle(current_handle)
csv_file = self.cust_manage_page_read_csv.read_csv('acc_search.csv')
csv_data = csv.reader(csv_file)
is_header = True
for row in csv_data:
if is_header:
is_header = False
continue
search_data = {
'account': row[0],
"account_type": row[1],
"info": row[2]
}
print(search_data)
sleep(1)
self.cust_manage_lower_account_page.add_data_to_search_account(
search_data)
connect = self.connect_sql.connect_tuqiang_sql()
# 创建数据库游标
cur = connect.cursor()
# 执行sql脚本查询当前登录账号的userId,fullParent
get_id_sql = "select o.account,o.userId,o.fullParentId from user_info o where o.account = '" + search_data[
'account'] + "' ;"
cur.execute(get_id_sql)
# 读取数据
user_relation = cur.fetchall()
# 遍历数据
for row in user_relation:
user_relation_id = {
"account": row[0],
"userId": row[1],
"fullParent": row[2]
}
# 执行sql脚本，根据当前登录账号的userId,fullParent查询出当前账户的所有下级账户
get_lower_account_sql = "select userId from user_info where fullParentId = " + \
"'" + user_relation_id["fullParent"] + user_relation_id[
"userId"] + ",'" + ";"
cur.execute(get_lower_account_sql)
# 读取数据
lower_account = cur.fetchall()
print(lower_account)
lower_account_list = []
for range1 in lower_account:
for range2 in range1:
lower_account_list.append(range2)
lower_account_tuple = tuple(lower_account_list)
print(lower_account_tuple)
get_total_sql = self.search_sql.search_account_sql(
lower_account_tuple, search_data)
print(get_total_sql)
cur.execute(get_total_sql)
# 读取数据
total_data = cur.fetchall()
# 从数据tuple中获取最终查询记录统计条数
total_list = []
for range1 in total_data:
for range2 in range1:
total_list.append(range2)
total = len(total_list)
print('本次查询数据库的条数为：%s' % total)
web_total = self.cust_manage_lower_account_page.get_account_number(
)
print('本次查询页面的条数是：%s' % web_total)
self.assertEqual(total, web_total)
cur.close()
connect.close()
csv_file.close()
class TestCase58CustManageAddAcc(unittest.TestCase):
def setUp(self):
self.driver = AutomateDriverServer()
self.base_url = self.driver.base_url
self.base_page = BasePageServer(self.driver, self.base_url)
self.login_page = LoginPage(self.driver, self.base_url)
self.cust_manage_basic_info_and_add_cust_page = CustManageBasicInfoAndAddCustPage(
self.driver, self.base_url)
self.cust_manage_cust_list_page = CustManageCustListPage(
self.driver, self.base_url)
self.cust_manage_my_dev_page = CustManageMyDevPage(
self.driver, self.base_url)
self.cust_manage_lower_account_page = CustManageLowerAccountPage(
self.driver, self.base_url)
self.account_center_page_navi_bar = AccountCenterNaviBarPage(
self.driver, self.base_url)
self.assert_text2 = AssertText2()
self.search_sql = SearchSql()
self.driver.set_window_max()
self.log_in_base = LogInBaseServer(self.driver, self.base_url)
self.cust_manage_page_read_csv = CustManagePageReadCsv()
self.connect_sql = ConnectSql()
self.assert_text = AssertText()
self.driver.wait(1)
self.driver.clear_cookies()
self.driver.wait(1)
def tearDown(self):
self.driver.quit_browser()
def test_cust_manage_add_acc(self):
'''测试客户管理-新增用户'''
# 打开途强在线首页-登录页
self.base_page.open_page()
# 登录
self.log_in_base.log_in()
# 进入客户管理页面
self.cust_manage_basic_info_and_add_cust_page.enter_cust_manage()
self.cust_manage_basic_info_and_add_cust_page.add_acc()
self.cust_manage_basic_info_and_add_cust_page.close_add_account()
csv_file = self.cust_manage_page_read_csv.read_csv('acc_add.csv')
csv_data = csv.reader(csv_file)
for row in csv_data:
add_info = {
"keyword": row[0],
"acc_type": row[1],
"acc_name": row[2],
"account": row[3],
"passwd": row[4],
"phone": row[5],
"email": row[6],
"conn": row[7],
"com": row[8],
"search_user": row[9]
}
# 左侧客户列表搜索并选中唯一客户
self.cust_manage_cust_list_page.acc_exact_search(
add_info["keyword"])
# 点击新增用户
self.cust_manage_basic_info_and_add_cust_page.add_acc()
sleep(2)
# 验证所选中的上级用户类型来显示可创建的下级类型
user_list = add_info["search_user"].split("/")
for type in user_list:
self.cust_manage_basic_info_and_add_cust_page.acc_search(type)
type_list = self.cust_manage_basic_info_and_add_cust_page.get_acc_user_type_list(
)
sql_data = self.search_sql.search_current_account_data(type)
user_type = self.assert_text.log_in_page_account_type(
sql_data[2])
print(user_type)
if user_type == "销售":
self.assertEqual(3, type_list["length"])
self.assertIn(
self.assert_text.log_in_page_account_type(11),
type_list["sale"])
self.assertIn(self.assert_text.log_in_page_account_type(8),
type_list["distributor"])
self.assertIn(self.assert_text.log_in_page_account_type(9),
type_list["user"])
elif user_type == "代理商":
self.assertEqual(2, type_list["length"])
self.assertIn(self.assert_text.log_in_page_account_type(8),
type_list["distributor"])
self.assertIn(self.assert_text.log_in_page_account_type(9),
type_list["user"])
elif user_type == "用户":
self.assertNotEqual(1, type_list["length"])
#self.assertIn(self.assert_text.log_in_page_account_type(9),type_list["user"])
sleep(1)
#添加用户类型获取提示
self.cust_manage_basic_info_and_add_cust_page.acc_search(
"yonghu222")
sleep(2)
self.cust_manage_basic_info_and_add_cust_page.locate_to_iframe()
# status = self.cust_manage_lower_account_page.edit_info_save_status()
self.driver.default_frame()
# self.assertEqual(self.assert_text2.cust_manage_add_user_type_prompt(), status, "提示显示不一致")
# 右侧搜索栏中搜索并选中作为上级用户
self.cust_manage_basic_info_and_add_cust_page.acc_search(
add_info["keyword"])
# 选择客户类型
self.cust_manage_basic_info_and_add_cust_page.locate_to_iframe()
self.cust_manage_basic_info_and_add_cust_page.acc_type_choose(
add_info["acc_type"])
# 编辑用户输入框信息
self.cust_manage_basic_info_and_add_cust_page.add_acc_input_info_edit(
add_info["acc_name"], add_info["account"], add_info["passwd"],
add_info["phone"], add_info["email"], add_info["conn"],
add_info["com"])
# 修改用户登录权限
self.cust_manage_basic_info_and_add_cust_page.acc_login_limit_modi(
)
self.driver.default_frame()
self.cust_manage_basic_info_and_add_cust_page.acc_add_save()
# 获取保存操作状态
status = self.cust_manage_basic_info_and_add_cust_page.acc_info_save_status(
)
# 验证是否操作成功
self.assertIn(
self.assert_text.account_center_page_operation_done(), status,
"操作失败")
# 搜索新增客户
sleep(4)
self.cust_manage_lower_account_page.input_search_info(
add_info["account"])
# 搜索
self.cust_manage_lower_account_page.click_search_btn()
# 删除该新增客户
self.cust_manage_lower_account_page.delete_acc()
# 确定删除
self.cust_manage_lower_account_page.delete_acc_ensure()
# 获取删除操作状态
del_status = self.cust_manage_lower_account_page.get_del_status()
# 验证是否操作成功
self.assertIn(
self.assert_text.account_center_page_operation_done(),
del_status, "操作失败")
csv_file.close()