def setUp(self):
# 前置条件
# 实例化对象
self.driver = AutomateDriver(choose='firefox')
self.base_url = self.driver.base_url
self.base_page = BasePage(self.driver, self.base_url)
self.command_management_page = CommandManagementPage(
self.driver, self.base_url)
self.log_in_base = LogInBase(self.driver, self.base_url)
self.command_management_page_read_csv = CommandManagementPageReadCsv()
self.connect_sql = ConnectSql()
self.search_sql = SearchSql()
self.assert_text = AssertText()
# 打开页面，填写用户名、密码、点击登录
self.base_page.open_page()
self.driver.set_window_max()
self.driver.implicitly_wait(5)
self.driver.clear_cookies()
self.log_in_base.log_in_jimitest()
self.log_in_base.click_account_center_button()
self.current_account = self.log_in_base.get_log_in_account()
# 登录之后点击控制台，然后点击指令管理
self.command_management_page.click_control_after_click_command_management(
)
sleep(3)
class TestCase45IssuedWorkTypetTaskManagementSearch(unittest.TestCase):
""" 发工作模式任务管理搜索功能 """
# author:邓肖斌
driver = None
base_url = None
base_page = None
log_in_page = None
command_management_page = None
def setUp(self):
# 前置条件
# 实例化对象
self.driver = AutomateDriver(choose='firefox')
self.base_url = self.driver.base_url
self.base_page = BasePage(self.driver, self.base_url)
self.command_management_page = CommandManagementPage(self.driver, self.base_url)
self.command_management_page_read_csv = CommandManagementPageReadCsv()
self.log_in_base = LogInBase(self.driver, self.base_url)
self.connect_sql = ConnectSql()
self.assert_text = AssertText()
self.search_sql = SearchSql()
# 打开页面，填写用户名、密码、点击登录
self.base_page.open_page()
self.driver.set_window_max()
self.driver.implicitly_wait(5)
self.driver.clear_cookies()
self.log_in_base.log_in()
self.log_in_base.click_account_center_button()
self.current_account = self.log_in_base.get_log_in_account()
# 登录之后点击控制台，然后点击指令管理
self.command_management_page.click_control_after_click_command_management()
sleep(3)
def tearDown(self):
self.driver.quit_browser()
def test_case_issued_work_type_task_management_search(self):
# 断言url
expect_url_after_click_command_management = self.base_url + '/custom/toTemplate'
self.assertEqual(expect_url_after_click_command_management,
self.command_management_page.actual_url_click_command_management())
# 断言左侧列表的title文本
expect_title_text_after_click_command_management = self.assert_text.command_manager_page_command_type()
self.assertEqual(expect_title_text_after_click_command_management,
self.command_management_page.actual_title_text_after_click_command_management())
# 点击工作模式模板管理
self.command_management_page.click_lift_list('issued_work_type_task_management')
# 断言
expect_title_text_after_click_issued_work_type_task_mangagement = \
self.assert_text.command_manager_page_work_type_task_manager_text()
self.assertEqual(expect_title_text_after_click_issued_work_type_task_mangagement,
self.command_management_page.actual_title_text_click_issued_work_type_task_management())
csv_file = self.command_management_page_read_csv.read_csv('search_work_type_task_manage_data.csv')
csv_data = csv.reader(csv_file)
is_header = True
for row in csv_data:
if is_header:
is_header = False
continue
search_data = {
'number': row[0],
'name': row[1]
}
self.command_management_page.issued_work_type_task_management_search(search_data['number'],
search_data['name'])
# 创建连接
connect = self.connect_sql.connect_tuqiang_sql()
# 创建游标
cursor = connect.cursor()
# 获取当前登录账号的usedID
get_current_user_id_sql = \
"select o.account,o.userId from user_info o where o.account = '" + self.current_account + "';"
cursor.execute(get_current_user_id_sql)
user_relation = cursor.fetchall()
for row1 in user_relation:
user_id = {
'user_account': row1[0],
'user_id': row1[1]
}
get_total_sql = self.search_sql.search_work_task_manager_sql(user_id['user_id'], search_data)
print(get_total_sql)
cursor.execute(get_total_sql)
# 读取数据
total_data = cursor.fetchall()
# 从数据tuple中获取最终查询记录统计条数
total_list = []
for range1 in total_data:
for range2 in range1:
total_list.append(range2)
print(total_list)
total = len(total_list)
print('本次查询数据库的条数为：%s' % total)
web_total = self.command_management_page.search_total_number_work_task_manage()
print('本次查询页面的条数是：%s' % web_total)
self.assertEqual(total, web_total)
cursor.close()
connect.close()
csv_file.close()
class TestCase307SetUpIssuedWorkTypeManagementSearch(unittest.TestCase):
# 测试 设置 - 指令管理 - 下发工作模式管理页面搜索
driver = None
base_url = None
base_page = None
log_in_page = None
command_management_page = None
def setUp(self):
# 前置条件
# 实例化对象
self.driver = AutomateDriver(choose='firefox')
self.base_url = self.driver.base_url
self.base_page = BasePage(self.driver, self.base_url)
self.command_management_page = CommandManagementPage(
self.driver, self.base_url)
self.log_in_base = LogInBase(self.driver, self.base_url)
self.command_management_page_read_csv = CommandManagementPageReadCsv()
self.connect_sql = ConnectSql()
self.search_sql = SearchSql()
self.assert_text = AssertText()
# 打开页面，填写用户名、密码、点击登录
self.base_page.open_page()
self.driver.set_window_max()
self.driver.implicitly_wait(5)
self.driver.clear_cookies()
self.log_in_base.log_in_jimitest()
self.log_in_base.click_account_center_button()
self.current_account = self.log_in_base.get_log_in_account()
# 登录之后点击控制台，然后点击指令管理
self.command_management_page.click_control_after_click_command_management(
)
sleep(3)
def tearDown(self):
self.driver.quit_browser()
def test_case_set_up_issued_work_type_management_search(self):
# 断言url
expect_url_after_click_command_management = self.base_url + '/custom/toTemplate'
self.assertEqual(
expect_url_after_click_command_management,
self.command_management_page.actual_url_click_command_management())
# 断言左侧列表的title文本
expect_title_text_after_click_command_management = self.assert_text.command_manager_page_command_type(
)
self.assertEqual(
expect_title_text_after_click_command_management,
self.command_management_page.
actual_title_text_after_click_command_management())
# 点击下发工作模式管理
self.command_management_page.click_lift_list(
'issued_work_type_management')
# 断言
expect_title_text_click_issued_work_type = self.assert_text.command_manager_page_issued_work_type(
)
self.assertEqual(
expect_title_text_click_issued_work_type,
self.command_management_page.actual_text_click_look_equipment())
# 读取数据
csv_file = self.command_management_page_read_csv.read_csv(
'issused_work_template.csv')
csv_data = csv.reader(csv_file)
is_header = True
for row in csv_data:
if is_header:
is_header = False
continue
search_data = {
'batch': row[0],
'execute_state': row[1],
'state': row[2],
'imei': row[3]
}
# 传入数据
self.command_management_page.add_data_to_search(search_data)
# 创建数据库连接
connect = self.connect_sql.connect_tuqiang_sql()
# 创建游标
cursor = connect.cursor()
# 获取登录账号的ID
get_current_user_id_sql = \
"select o.account,o.userId from user_info o where o.account = '" + self.current_account + "';"
cursor.execute(get_current_user_id_sql)
user_relation = cursor.fetchall()
for row1 in user_relation:
user_id = {'user_account': row1[0], 'user_id': row1[1]}
# 查询数据库有多少条记录
get_total_count_sql = self.search_sql.search_issued_work_template_sql(
user_id['user_id'], search_data)
print(get_total_count_sql)
cursor.execute(get_total_count_sql)
current_total = cursor.fetchall()
total_list = []
for range1 in current_total:
for range2 in range1:
total_list.append(range2)
total = len(total_list)
web_total = self.command_management_page.search_total_number_issued_work_type(
)
self.assertEqual(total, web_total)
cursor.close()
connect.close()
csv_file.close()
class TestCase309SetUpIssuedCommandTaskManagementSearch(unittest.TestCase):
# 测试 设置 - 指令管理 - 下发指令任务管理的搜索功能
driver = None
base_url = None
base_page = None
log_in_page = None
command_management_page = None
base_paging_function = None
def setUp(self):
# 前置条件
# 实例化对象
self.driver = AutomateDriver(choose='firefox')
self.base_url = self.driver.base_url
self.base_page = BasePage(self.driver, self.base_url)
self.command_management_page = CommandManagementPage(self.driver, self.base_url)
self.base_paging_function = BasePagingFunction(self.driver, self.base_url)
self.command_management_page_read_csv = CommandManagementPageReadCsv()
self.connect_sql = ConnectSql()
self.log_in_base = LogInBase(self.driver, self.base_url)
self.search_sql = SearchSql()
self.assert_text = AssertText()
# 打开页面，填写用户名、密码、点击登录
self.base_page.open_page()
self.driver.set_window_max()
self.driver.implicitly_wait(5)
self.driver.clear_cookies()
self.log_in_base.log_in()
self.log_in_base.click_account_center_button()
self.current_account = self.log_in_base.get_log_in_account()
# 登录之后点击控制台，然后点击指令管理
self.command_management_page.click_control_after_click_command_management()
sleep(3)
def tearDown(self):
self.driver.quit_browser()
def test_case_set_up_issued_command_task_management_search(self):
# 断言url
expect_url_after_click_command_management = self.base_url + '/custom/toTemplate'
self.assertEqual(expect_url_after_click_command_management,
self.command_management_page.actual_url_click_command_management())
# 断言左侧列表的title文本
expect_title_text_after_click_command_management = self.assert_text.command_manager_page_command_type()
self.assertEqual(expect_title_text_after_click_command_management,
self.command_management_page.actual_title_text_after_click_command_management())
# 点击下发任务指令管理
self.command_management_page.click_lift_list('issued_command_task_management')
# 断言
expect_title_text_after_click_issued_command_task_management = \
self.assert_text.command_manager_page_issued_command_task()
self.assertEqual(expect_title_text_after_click_issued_command_task_management,
self.command_management_page.actual_text_after_click_issued_command_task())
csv_file = self.command_management_page_read_csv.read_csv('issued_command_task_management_serach_data.csv')
csv_data = csv.reader(csv_file)
i = 1
is_header = True
for row in csv_data:
if is_header:
is_header = False
continue
search_data = {
'batch': row[0],
'name': row[1]
}
print(search_data)
self.command_management_page.issued_command_task_add_data_to_search(search_data)
'''# 建立数据库的连接
connect = self.connect_sql.connect_tuqiang_sql()
# 建立游标
cursor = connect.cursor()
# 查询登录用户的ID 和 父ID
get_current_user_id = \
"select o.userId,o.fullParentId from user_info o where o.account = '" + self.current_account + "';"
# 执行
cursor.execute(get_current_user_id)
user = cursor.fetchall()
for row1 in user:
user_info = {
'id': row1[0],
'fullparent': row1[1]
}
# 查询当前登录用户的全部下级
get_next_id_sql = "select userId from user_info where fullParentId like" + \
"'" + user_info["fullparent"] + user_info["id"] + "%'" + ";"
# 执行sql脚本
cursor.execute(get_next_id_sql)
current_account = cursor.fetchall()
current_account_list = [user_info['id']]
for range1 in current_account:
for range2 in range1:
current_account_list.append(range2)
current_user_next = tuple(current_account_list)
print(current_user_next)
# 查询数据库'''
# 执行sql
user_info = self.command_management_page.get_current_user_id(self.current_account)
all_user_id = self.command_management_page.get_all_user_id(user_info)
connect = self.connect_sql.connect_tuqiang_sql()
cursor = connect.cursor()
search_sql = self.search_sql.search_issued_command_task_management_sql(all_user_id, search_data)
print(search_sql)
cursor.execute(search_sql)
current_total = cursor.fetchall()
cursor.close()
connect.close()
total = len(current_total)
web_total = self.command_management_page.search_total_number_with_issued_command_task()
# 断言
print('第%s次sql查询的结果是%s' % (i, total))
print('第%s次web查询的结果是%s' % (i, web_total))
i += 1
self.assertEqual(total, web_total)
csv_file.close()