def setUp(self):
self.driver = AutomateDriver()
self.base_url = self.driver.base_url
self.base_page = BasePage(self.driver, self.base_url)
self.login_page = LoginPage(self.driver, self.base_url)
self.user_center = UserCenter(self.driver, self.base_url)
self.role_management = RoleManagement(self.driver, self.base_url)
self.log_in_page_read_csv = LogInPageReadCsv()
self.user_center_read_csv = UserCenterReadCsv()
self.role_management_read_csv = RoleManagementReadCsv()
self.driver.set_window_max()
self.connect_sql = ConnectSql()
self.driver.wait(1)
self.driver.clear_cookies()
self.driver.wait(1)
class TestCase06RoleManageDeleteRole(unittest.TestCase):
# 测试角色管理删除角色
def setUp(self):
self.driver = AutomateDriver()
self.base_url = self.driver.base_url
self.base_page = BasePage(self.driver, self.base_url)
self.login_page = LoginPage(self.driver, self.base_url)
self.user_center = UserCenter(self.driver, self.base_url)
self.role_management = RoleManagement(self.driver, self.base_url)
self.log_in_page_read_csv = LogInPageReadCsv()
self.user_center_read_csv = UserCenterReadCsv()
self.role_management_read_csv = RoleManagementReadCsv()
self.driver.set_window_max()
self.connect_sql = ConnectSql()
self.driver.wait(1)
self.driver.clear_cookies()
self.driver.wait(1)
def tearDown(self):
self.driver.quit_browser()
def test_delete_role(self):
# 通过csv测试删除角色功能
csv_file = self.role_management_read_csv.read_csv('delete_role.csv')
csv_data = csv.reader(csv_file)
for row in csv_data:
delete_role = {
"account": row[0],
"passwd": row[1],
"delete_role_name": row[2],
}
# 打开风控首页-登录页
self.base_page.open_page()
sleep(1)
# 登录账号
self.login_page.user_login(delete_role['account'],
delete_role['passwd'])
# 点击进入角色管理
self.role_management.click_role_manage()
# 按名称查找角色
self.role_management.search_role(delete_role['delete_role_name'])
# 点击删除
self.role_management.click_delete_role()
# 跳出frame
self.role_management.switch_to_default_content()
# 点击取消
self.role_management.delete_role_dismiss()
# 切入frame
self.role_management.switch_to_1_frame()
# 点击删除
self.role_management.click_delete_role()
# 跳出frame
self.role_management.switch_to_default_content()
# 点击关闭
self.role_management.delete_role_close()
# 切入frame
self.role_management.switch_to_1_frame()
# 点击删除
self.role_management.click_delete_role()
# 跳出frame
self.role_management.switch_to_default_content()
# 点击确认
self.role_management.delete_role_accept()
# 数据库查找验证部门是否删除成功
# 数据库查找角色
role_name_after_delete = self.role_management.get_search_result_rolename_by_sql(
delete_role['account'], '')
self.assertNotIn(delete_role['delete_role_name'],
role_name_after_delete)
# 退出登录
self.user_center.logout()
csv_file.close()
class TestCase01OrgManageAddRole(unittest.TestCase):
# 测试角色管理新增角色
def setUp(self):
self.driver = AutomateDriver()
self.base_url = self.driver.base_url
self.base_page = BasePage(self.driver, self.base_url)
self.login_page = LoginPage(self.driver, self.base_url)
self.user_center = UserCenter(self.driver, self.base_url)
self.role_management = RoleManagement(self.driver, self.base_url)
self.log_in_page_read_csv = LogInPageReadCsv()
self.user_center_read_csv = UserCenterReadCsv()
self.role_management_read_csv = RoleManagementReadCsv()
self.driver.set_window_max()
self.connect_sql = ConnectSql()
self.driver.wait(1)
self.driver.clear_cookies()
self.driver.wait(1)
def tearDown(self):
self.driver.quit_browser()
def test_add_role(self):
# 通过csv测试新增角色功能
csv_file = self.role_management_read_csv.read_csv('add_role.csv')
csv_data = csv.reader(csv_file)
for row in csv_data:
add_role = {
"account": row[0],
"password": row[1],
"add_role_name": row[2],
"add_role_description": row[3],
"add_role_limit": row[4],
}
# 打开风控首页-登录页
self.base_page.open_page()
sleep(1)
# 登录账号
self.login_page.user_login(add_role['account'],
add_role['password'])
# 判断登录成功后招呼栏的用户名是否正确
username = self.user_center.get_username()
# 从数据库获取登录账号的用户名
account_info = self.user_center.get_account_info_by_sql(
add_role['account'])
print(account_info)
account_name = account_info[1]
self.assertEqual(account_name, username, '登录成功后招呼栏的用户名错误')
# 点击进入角色管理
self.role_management.click_role_manage()
# 点击创建角色
self.role_management.click_add_role()
# 切入内层frame
self.role_management.switch_to_2_frame()
# 输入角色名称
self.role_management.input_add_role_name(add_role['add_role_name'])
# 输入角色描述
self.role_management.input_add_role_description(
add_role['add_role_description'])
# 选择角色权限
self.role_management.choose_add_role_limit(
add_role['add_role_limit'])
# 点击取消
self.role_management.click_add_role_dismiss()
# 从数据库查询验证角色未创建成功
new_role_name_01 = self.role_management.get_add_role_by_sql(
add_role['account'])
self.assertNotEqual(new_role_name_01, add_role['add_role_name'])
self.driver.wait(1)
# 点击创建角色
self.role_management.click_add_role()
# 切入内层frame
self.role_management.switch_to_2_frame()
# 输入角色名称
self.role_management.input_add_role_name(add_role['add_role_name'])
# 输入角色描述
self.role_management.input_add_role_description(
add_role['add_role_description'])
# 选择角色权限
self.role_management.choose_add_role_limit(
add_role['add_role_limit'])
# 点击关闭按钮
self.role_management.click_add_role_close()
# 从数据库查询验证角色未创建成功
new_role_name_02 = self.role_management.get_add_role_by_sql(
add_role['account'])
self.assertNotEqual(new_role_name_02, add_role['add_role_name'])
self.driver.wait(1)
# 点击创建角色
self.role_management.click_add_role()
# 切入内层frame
self.role_management.switch_to_2_frame()
# 输入角色名称
self.role_management.input_add_role_name(add_role['add_role_name'])
# 输入角色描述
self.role_management.input_add_role_description(
add_role['add_role_description'])
# 选择角色权限
self.role_management.choose_add_role_limit(
add_role['add_role_limit'])
# 点击保存按钮
self.role_management.click_add_role_save()
# 跳出最外层frame
self.role_management.switch_to_default_content()
# 从数据库查询验证角色创建成功
new_role_name_03 = self.role_management.get_add_role_by_sql(
add_role['account'])
self.assertEqual(new_role_name_03, add_role['add_role_name'])
self.driver.wait(1)
# 退出登录
self.user_center.logout()
csv_file.close()
class TestCase05RoleManageSearchRole(unittest.TestCase):
# 测试角色管理搜索角色
def setUp(self):
self.driver = AutomateDriver()
self.base_url = self.driver.base_url
self.base_page = BasePage(self.driver, self.base_url)
self.login_page = LoginPage(self.driver, self.base_url)
self.user_center = UserCenter(self.driver, self.base_url)
self.role_management = RoleManagement(self.driver, self.base_url)
self.log_in_page_read_csv = LogInPageReadCsv()
self.user_center_read_csv = UserCenterReadCsv()
self.role_management_read_csv = RoleManagementReadCsv()
self.driver.set_window_max()
self.connect_sql = ConnectSql()
self.driver.wait(1)
self.driver.clear_cookies()
self.driver.wait(1)
def tearDown(self):
self.driver.quit_browser()
def test_search_role(self):
# 通过csv测试搜索角色功能
csv_file = self.role_management_read_csv.read_csv('search_role.csv')
csv_data = csv.reader(csv_file)
for row in csv_data:
search_role = {
"account": row[0],
"password": row[1],
"search_name": row[2],
}
# 打开风控首页-登录页
self.base_page.open_page()
sleep(1)
# 登录账号
self.login_page.user_login(search_role['account'],
search_role['password'])
# 判断登录成功后招呼栏的用户名是否正确
username = self.user_center.get_username()
# 从数据库获取登录账号的用户名
account_info = self.user_center.get_account_info_by_sql(
search_role['account'])
print(account_info)
account_name = account_info[1]
self.assertEqual(account_name, username, '登录成功后招呼栏的用户名错误')
# 点击进入角色管理
self.role_management.click_role_manage()
# 输入搜索关键词进行搜索
self.role_management.search_role(search_role['search_name'])
# 获取搜索结果
num = int(self.role_management.get_search_result_num())
role_name = self.role_management.get_search_result_all()
# 数据库查询搜索结果
role_name_by_sql = self.role_management.get_search_result_rolename_by_sql(
search_role['account'], search_role['search_name'])
num_by_sql = self.role_management.get_search_result_num_by_sql(
search_role['account'], search_role['search_name'])
# 验证搜索结果是否一致
self.assertEqual(num, num_by_sql)
self.assertEqual(set(role_name), set(role_name_by_sql))
# 跳出外层frame
self.role_management.switch_to_default_content()
# 退出登录
self.user_center.logout()
csv_file.close()
class TestCase04RoleManageEditRole(unittest.TestCase):
# 测试角色管理修改角色
def setUp(self):
self.driver = AutomateDriver()
self.base_url = self.driver.base_url
self.base_page = BasePage(self.driver, self.base_url)
self.login_page = LoginPage(self.driver, self.base_url)
self.user_center = UserCenter(self.driver, self.base_url)
self.role_management = RoleManagement(self.driver, self.base_url)
self.log_in_page_read_csv = LogInPageReadCsv()
self.user_center_read_csv = UserCenterReadCsv()
self.role_management_read_csv = RoleManagementReadCsv()
self.driver.set_window_max()
self.connect_sql = ConnectSql()
self.driver.wait(1)
self.driver.clear_cookies()
self.driver.wait(1)
def tearDown(self):
self.driver.quit_browser()
def test_edit_role(self):
# 通过csv测试修改角色功能
csv_file = self.role_management_read_csv.read_csv('edit_role_info.csv')
csv_data = csv.reader(csv_file)
for row in csv_data:
edit_role_info = {
"account": row[0],
"password": row[1],
"search_key": row[2],
"edit_role_name": row[3],
"edit_role_description": row[4],
"edit_role_limit": row[5],
}
# 打开风控首页-登录页
self.base_page.open_page()
sleep(1)
# 登录账号
self.login_page.user_login(edit_role_info['account'],
edit_role_info['password'])
# 判断登录成功后招呼栏的用户名是否正确
username = self.user_center.get_username()
# 从数据库获取登录账号的用户名
account_info = self.user_center.get_account_info_by_sql(
edit_role_info['account'])
print(account_info)
account_name = account_info[1]
self.assertEqual(account_name, username, '登录成功后招呼栏的用户名错误')
# 点击进入角色管理
self.role_management.click_role_manage()
# 查找11111角色
self.role_management.search_role(edit_role_info['search_key'])
# 获取页面查找结果
search_role_name_01 = self.role_management.get_search_result_one()
# 查询数据库获取搜索结果
role_info_00 = self.role_management.get_search_result_by_sql(
edit_role_info['account'], edit_role_info['search_key'])
role_name_00 = role_info_00[0]
role_desc_00 = role_info_00[1]
self.assertEqual(search_role_name_01, role_name_00)
# 点击修改
self.role_management.click_edit_role()
# 切入内层frame
self.role_management.switch_to_2_frame()
# 获取当前显示的角色名称是否与未修改前一致
current_role_name_01 = self.role_management.get_current_role_name()
self.assertEqual(search_role_name_01, current_role_name_01)
# 获取显示的角色描述是否与未修改前一致
current_role_desc = self.role_management.get_current_role_desc()
self.assertEqual(role_desc_00, current_role_desc)
# 输入角色名称
self.role_management.input_add_role_name(
edit_role_info['edit_role_name'])
# 输入角色描述
self.role_management.input_add_role_description(
edit_role_info['edit_role_description'])
# 选择角色权限
self.role_management.choose_add_role_limit(
edit_role_info['edit_role_limit'])
# 点击取消
self.role_management.click_add_role_dismiss()
# 数据库查询是否修改失败
role_info_01 = self.role_management.get_search_result_by_sql(
edit_role_info['account'], edit_role_info['search_key'])
role_name_01 = role_info_01[0]
role_desc_01 = role_info_01[1]
self.assertEqual(role_name_00, role_name_01)
self.assertEqual(role_desc_00, role_desc_01)
# 点击修改
self.role_management.click_edit_role()
# 切入内层frame
self.role_management.switch_to_2_frame()
# 获取当前显示的角色名称是否与未修改前一致
current_role_name_01 = self.role_management.get_current_role_name()
self.assertEqual(search_role_name_01, current_role_name_01)
# 获取显示的角色描述是否与未修改前一致
current_role_desc = self.role_management.get_current_role_desc()
self.assertEqual(role_desc_00, current_role_desc)
# 输入角色名称
self.role_management.input_add_role_name(
edit_role_info['edit_role_name'])
# 输入角色描述
self.role_management.input_add_role_description(
edit_role_info['edit_role_description'])
# 选择角色权限
self.role_management.choose_add_role_limit(
edit_role_info['edit_role_limit'])
# 点击关闭按钮
self.role_management.click_add_role_close()
# 数据库查询是否修改失败
role_info_02 = self.role_management.get_search_result_by_sql(
edit_role_info['account'], edit_role_info['search_key'])
role_name_02 = role_info_02[0]
role_desc_02 = role_info_02[1]
self.assertEqual(role_name_00, role_name_02)
self.assertEqual(role_desc_00, role_desc_02)
# 点击修改
self.role_management.click_edit_role()
# 切入内层frame
self.role_management.switch_to_2_frame()
# 获取当前显示的角色名称是否与未修改前一致
current_role_name_01 = self.role_management.get_current_role_name()
self.assertEqual(search_role_name_01, current_role_name_01)
# 获取显示的角色描述是否与未修改前一致
current_role_desc = self.role_management.get_current_role_desc()
self.assertEqual(role_desc_00, current_role_desc)
# 输入角色名称
self.role_management.input_add_role_name(
edit_role_info['edit_role_name'])
# 输入角色描述
self.role_management.input_add_role_description(
edit_role_info['edit_role_description'])
# 选择角色权限
self.role_management.choose_add_role_limit(
edit_role_info['edit_role_limit'])
# 点击保存按钮
self.role_management.click_add_role_save()
# 数据库查询是否修改成功
role_info_03 = self.role_management.get_search_result_by_sql(
edit_role_info['account'], edit_role_info['edit_role_name'])
role_name_03 = role_info_03[0]
role_desc_03 = role_info_03[1]
self.assertNotEqual(role_name_00, role_name_03)
self.assertNotEqual(role_desc_00, role_desc_03)
# 跳出外层frame
self.role_management.switch_to_default_content()
# 退出登录
self.user_center.logout()
csv_file.close()
class TestCase02RoleManageAddRoleException1(unittest.TestCase):
# 测试角色管理新增角色功能异常情况
def setUp(self):
self.driver = AutomateDriver()
self.base_url = self.driver.base_url
self.base_page = BasePage(self.driver, self.base_url)
self.login_page = LoginPage(self.driver, self.base_url)
self.user_center = UserCenter(self.driver, self.base_url)
self.role_management = RoleManagement(self.driver, self.base_url)
self.log_in_page_read_csv = LogInPageReadCsv()
self.user_center_read_csv = UserCenterReadCsv()
self.role_management_read_csv = RoleManagementReadCsv()
self.driver.set_window_max()
self.connect_sql = ConnectSql()
self.driver.wait(1)
self.driver.clear_cookies()
self.driver.wait(1)
# 打开风控首页-登录页
self.base_page.open_page()
sleep(1)
# 登录账号
self.login_page.test_user_login()
def tearDown(self):
self.driver.quit_browser()
def test_add_role_exception1(self):
# 通过csv测试新增角色功能异常情况
# 获取当前登录账户
log_in_account = self.user_center.get_login_account()
print(log_in_account)
csv_file = self.role_management_read_csv.read_csv(
'add_role_exception1.csv')
csv_data = csv.reader(csv_file)
for row in csv_data:
add_role_exception = {
"add_role_name": row[0],
"add_role_description": row[1],
"add_role_limit": row[2],
"exception_text": row[3],
}
# 点击进入角色管理
self.role_management.click_role_manage()
# 点击创建角色
self.role_management.click_add_role()
# 切入内层frame
self.role_management.switch_to_2_frame()
# 输入角色名称
self.role_management.input_add_role_name(
add_role_exception['add_role_name'])
# 输入角色描述
self.role_management.input_add_role_description(
add_role_exception['add_role_description'])
# 选择角色权限
self.role_management.choose_add_role_limit(
add_role_exception['add_role_limit'])
# 点击保存按钮
self.role_management.click_add_role_save()
# 获取异常提示
text = self.role_management.get_add_role_exception1()
self.assertEqual(add_role_exception['exception_text'], text)
# 关闭创建角色框
self.role_management.click_add_role_close()
# 跳出最外层frame
self.role_management.switch_to_default_content()
# 从数据库查询验证新增失败
new_role_name = self.role_management.get_add_role_by_sql(
log_in_account)
self.assertNotEqual(new_role_name,
add_role_exception['add_role_name'])
self.driver.wait(1)
csv_file.close()
# 退出登录
self.user_center.logout()