class TestLogin(unittest.TestCase): '''测试登录功能''' def setUp(self): # 登录类 self.l = Login() # excel操作类 self.sheet = DoExcel('Login_cases') def login(self, name): # 根据用例编号获取测试数据 test_data = self.sheet.get_test_data(name) # 执行登录 self.driver = self.l.loginAction(test_data['username'], test_data['password']) def test_login_001(self): '''用户名密码都正确,登录成功''' # 获取当前方法名,与用例编号一致 name = sys._getframe().f_code.co_name # 执行登录 self.login(name) time.sleep(2) # 获取页面源代码 text_mes = self.driver.page_source # 判断是否登录成功 self.assertIn('欢迎', text_mes) def test_login_002(self): '''密码为空,登录失败''' # self.driver = self.l.loginAction('admin','') name = sys._getframe().f_code.co_name self.login(name) time.sleep(2) text_mes = self.driver.page_source self.assertNotIn('欢迎', text_mes) def test_login_003(self): '''密码错误,登录失败''' # self.driver = self.l.loginAction('admin','123456') name = sys._getframe().f_code.co_name self.login(name) time.sleep(2) text_mes = self.driver.page_source self.assertNotIn('欢迎', text_mes) def tearDown(self): self.driver.close()
class TestVipTeacher(unittest.TestCase): '''测试会员管理-添加教师功能''' def setUp(self): # 登录类 self.l = Login() self.driver = self.l.loginAction() # 菜单切换 self.m = Menu_teacher(self.driver) # 添加教师类 self.add = AddTeacher(self.driver) # 实例化添加会员管理表格 self.sheet = DoExcel('Vip_cases') def add_teacher_ation(self,name): # 根据用例编号获取测试数据 test_data = self.sheet.get_test_data(name) # 点击菜单操作 self.m.menu_into_addteacher() # 执行添加学生操作 text=self.add.add_teacher(test_data['username'],test_data['realname'],test_data['email'],test_data['phone']) return text def test_add_teacher_001(self): '''验证添加学生功能,数据合法保存成功''' # 获取当前方法名,与用例编号一致 name = sys._getframe().f_code.co_name # 执行添加学生操作 text=self.add_teacher_ation(name) # 判断是否添加成功 self.assertEqual(text,'保存成功') # 删除新增的数据 sql="delete a from xsmart_users a , (select max(id) as m_id from xsmart_users) b where a.id=b.m_id" connectSql(sql=sql) def test_add_teacher_002(self): '''验证添加教师功能,用户名为空保存失败''' name = sys._getframe().f_code.co_name text = self.add_teacher_ation(name) self.assertIsNone(text) def tearDown(self): self.driver.close()
from selenium.webdriver.common.by import By from pageObject.common.getdriver import Driver from pageObject.common.loginPage import Login class Menu_studet(Driver): vip_center = (By.LINK_TEXT, '会员中心') add_atudent_button = (By.PARTIAL_LINK_TEXT, '添加学生') def click_vip_center(self): # 进入会员管理页面 self.driver.find_element(*self.vip_center).click() def click_addstudent_button(self): # iframe跳转 self.driver.switch_to.frame('mainframe') # 点击添加学生 self.driver.find_element(*self.add_atudent_button).click() def menu_into_addstudent(self): self.click_vip_center() self.click_addstudent_button() if __name__ == '__main__': l = Login() driver = l.loginAction('admin', 'admin') m = Menu_studet(driver) m.menu_into_addstudent()