class Test_code_login(unittest.TestCase): def setUp(self): self.driver = driverBase.open_broswer() driverBase.max_window() self.login = Login(self.driver) self.driver.implicitly_wait(10) self.toast = ToastText(self.driver) def tearDown(self): driverBase.quit_broswer() def test_code_login_success(self): '''账号验证码正确登录''' self.login.send_code_login(url, logindata[0]['phone']) codes = self.login.get_code(sqldata[4]['set or search'], sqldata[4]['table'], sqldata[4]['where']) self.login.code_login(codes[0]) self.assertEqual(self.login.login_success_username(), logindata[0]['except_result']) def test_code_login_nullcode(self): '''账号正确,验证码为空登录''' self.login.send_code_login(url, logindata[1]['phone']) self.login.code_login(code='') self.assertEqual(self.toast.page_toast(), logindata[1]['except_result']) def test_code_login_allnull(self): '''账号、验证码为空''' self.login.send_code_login(url, logindata[2]['phone']) self.assertEqual(self.toast.page_toast(), logindata[2]['except_result']) def test_code_login_mismatch(self): '''账号和验证码不匹配''' self.login.send_code_login(url, logindata[3]['phone']) codes = self.login.get_code(sqldata[4]['set or search'], sqldata[4]['table'], sqldata[4]['where']) self.login.code_login(codes[1]) self.assertEqual(self.toast.sever_toast(), logindata[3]['except_result']) def test_code_login_newphone(self): '''系统中不存在的账号,验证码登录成功''' self.login.send_code_login(url, logindata[4]['phone']) codes = self.login.get_code(sqldata[4]['set or search'], sqldata[4]['table'], sqldata[4]['where']) self.login.code_login(codes[0]) self.assertEqual(self.login.login_success_username(), logindata[4]['except_result'])
class Test_register(unittest.TestCase): def setUp(self): self.driver = driverBase.open_broswer() driverBase.max_window() self.login = Login(self.driver) self.addcompany = Add_company(self.driver) self.driver.implicitly_wait(10) self.toast = ToastText(self.driver) def tearDown(self): driverBase.quit_broswer() # 注册成功 def test_register_success(self): '''注册成功,进入添加公司页面''' self.login.send_code_register_heard(url, register_data[0]['phone'], register_data[0]['psw'], register_data[0]['name']) codes = self.login.get_code(sqldata[6]['set or search'], sqldata[6]['table'], sqldata[6]['where']) self.login.wirte_code_register(codes[0]) self.login.select_deal() self.login.register() self.assertEqual(self.login.register_success(), register_data[0]['except_result']) # 已存在手机号 def test_exist_phone(self): '''手机号码已经注册''' self.login.send_code_register_heard(url, register_data[1]['phone'], register_data[1]['psw'], register_data[1]['name']) self.assertEqual(self.toast.page_toast(), register_data[1]['except_result']) # 密码为空 def test_empty_psw(self): '''请输入密码''' self.login.send_code_register_heard(url, register_data[2]['phone'], register_data[2]['psw'], register_data[2]['name']) codes = self.login.get_code(sqldata[7]['set or search'], sqldata[7]['table'], sqldata[7]['where']) self.login.wirte_code_register(codes[0]) self.login.select_deal() self.login.register() self.assertEqual(self.toast.page_toast(), register_data[2]['except_result']) # 姓名为空 def test_empty_name(self): '''请输入姓名''' self.login.send_code_register_loginpage(url, register_data[3]['phone'], register_data[3]['psw'], register_data[3]['name']) codes = self.login.get_code(sqldata[7]['set or search'], sqldata[7]['table'], sqldata[7]['where']) self.login.wirte_code_register(codes[0]) self.login.select_deal() self.login.register() self.assertEqual(self.toast.page_toast(), register_data[3]['except_result']) # 手机号为空 def test_empty_phone(self): '''请输入手机号''' self.login.send_code_register_loginpage(url, register_data[4]['phone'], register_data[4]['psw'], register_data[4]['name']) codes = self.login.get_code(sqldata[7]['set or search'], sqldata[7]['table'], sqldata[7]['where']) self.login.wirte_code_register(codes[0]) self.login.select_deal() self.login.register() self.assertEqual(self.toast.page_toast(), register_data[4]['except_result']) # 不符合规则的手机号 def test_incongruent_phone(self): '''请输入正确的手机号''' self.login.send_code_register_loginpage(url, register_data[5]['phone'], register_data[5]['psw'], register_data[5]['name']) self.login.select_deal() self.login.register() self.assertEqual(self.toast.page_toast(), register_data[5]['except_result']) # 6位全英文密码 def test_incongruent_psw(self): '''请输入6-16位字母数字组合的密码''' self.login.send_code_register_loginpage(url, register_data[6]['phone'], register_data[6]['psw'], register_data[6]['name']) codes = self.login.get_code(sqldata[8]['set or search'], sqldata[8]['table'], sqldata[8]['where']) self.login.wirte_code_register(codes[0]) self.login.select_deal() self.login.register() self.assertEqual(self.toast.page_toast(), register_data[6]['except_result']) # 6位全数字的密码 def test_allnumber_psw(self): '''请输入6-16位字母数字组合的密码''' self.login.send_code_register_loginpage(url, register_data[7]['phone'], register_data[7]['psw'], register_data[7]['name']) codes = self.login.get_code(sqldata[8]['set or search'], sqldata[8]['table'], sqldata[8]['where']) self.login.wirte_code_register(codes[0]) self.login.select_deal() self.login.register() self.assertEqual(self.toast.page_toast(), register_data[7]['except_result']) # 小于6位的密码 def test_less_psw(self): '''请输入6-16位字母数字组合的密码''' self.login.send_code_register_loginpage(url, register_data[8]['phone'], register_data[8]['psw'], register_data[8]['name']) codes = self.login.get_code(sqldata[8]['set or search'], sqldata[8]['table'], sqldata[8]['where']) self.login.wirte_code_register(codes[0]) self.login.select_deal() self.login.register() self.assertEqual(self.toast.page_toast(), register_data[8]['except_result']) # 验证码与手机号不匹配 def test_mismatch(self): '''验证码错误''' self.login.send_code_register_loginpage(url, register_data[9]['phone'], register_data[9]['psw'], register_data[9]['name']) codes = self.login.get_code(sqldata[9]['set or search'], sqldata[9]['table'], sqldata[9]['where']) self.login.wirte_code_register(codes[1]) self.login.select_deal() self.login.register() self.assertEqual(self.toast.page_toast(), register_data[9]['except_result']) # 验证码为空 def test_empty_code(self): '''请输入验证码''' self.login.send_code_register_loginpage(url, register_data[10]['phone'], register_data[10]['psw'], register_data[10]['name']) self.login.wirte_code_register(code='') self.login.select_deal() self.login.register() self.assertEqual(self.toast.page_toast(), register_data[10]['except_result']) # 所有数据都为空 def test_all_empty(self): '''请输入手机号''' self.login.send_code_register_loginpage(url, register_data[11]['phone'], register_data[11]['psw'], register_data[11]['name']) self.login.register() self.assertEqual(self.toast.page_toast(), register_data[11]['except_result']) # 不勾选服务协议 def test_ignore_deal(self): '''请阅读服务协议并勾选''' self.login.send_code_register_loginpage(url, register_data[12]['phone'], register_data[12]['psw'], register_data[12]['name']) codes = self.login.get_code(sqldata[9]['set or search'], sqldata[9]['table'], sqldata[9]['where']) self.login.wirte_code_register(codes[0]) self.login.register() self.assertEqual(self.toast.page_toast(), register_data[12]['except_result']) # 注册成功后添加公司 def test_register_addcompany(self): '''注册成功,添加公司''' self.login.send_code_register_loginpage(url, register_data[13]['phone'], register_data[13]['psw'], register_data[13]['name']) codes = self.login.get_code(sqldata[10]['set or search'], sqldata[10]['table'], sqldata[10]['where']) self.login.wirte_code_register(codes[0]) self.login.select_deal() self.login.register() self.addcompany.add_companyName(companydata[6]['name']) self.addcompany.add_company_verify() self.assertEqual(self.addcompany.companyName(), companydata[6]['except_result'])
class Reset_password(unittest.TestCase): def setUp(self): self.driver = driverBase.open_broswer() driverBase.max_window() self.login = Login(self.driver) self.driver.implicitly_wait(30) self.toast = ToastText(self.driver) def tearDown(self): driverBase.quit_broswer() # 修改密码成功 def reset_success(self): '''密码修改成功''' self.login.send_code_reset(url,resetdata[0]['phone']) codes = self.login.get_code(sqldata[3]['set or search'],sqldata[3]['table'],sqldata[3]['where']) self.login.psw_reset(codes[0],resetdata[0]['psw']) self.assertEqual(self.toast.sever_toast(),resetdata[0]['except_result']) # 手机号为空 def reset_empty_phone(self): '''请输入手机号''' self.login.send_code_reset(url,resetdata[1]['phone']) self.assertEqual(self.toast.page_toast(),resetdata[1]['except_result']) # 错误的验证码 def error_code(self): '''验证码错误''' self.login.send_code_reset(url,resetdata[2]['phone']) codes = self.login.get_code(sqldata[3]['set or search'], sqldata[3]['table'], sqldata[3]['where']) self.login.psw_reset(codes[1], resetdata[2]['psw']) self.assertEqual(self.toast.sever_toast(), resetdata[2]['except_result']) # 新密码为全数字 def error_new_psw(self): '''密码为全数字''' self.login.send_code_reset(url,resetdata[3]['phone']) codes = self.login.get_code(sqldata[3]['set or search'], sqldata[3]['table'], sqldata[3]['where']) self.login.psw_reset(codes[0], resetdata[3]['psw']) self.assertEqual(self.toast.sever_toast(), resetdata[3]['except_result']) # 新密码为全英文 def new_psw_error(self): '''密码为全英文''' self.login.send_code_reset(url,resetdata[4]['phone']) codes = self.login.get_code(sqldata[3]['set or search'], sqldata[3]['table'], sqldata[3]['where']) self.login.psw_reset(codes[0], resetdata[4]['psw']) self.assertEqual(self.toast.sever_toast(), resetdata[4]['except_result']) # 新密码小于6位 def less_new_psw(self): '''密码少于6位数''' self.login.send_code_reset(url,resetdata[5]['phone']) codes = self.login.get_code(sqldata[3]['set or search'], sqldata[3]['table'], sqldata[3]['where']) self.login.psw_reset(codes[0], resetdata[5]['psw']) self.assertEqual(self.toast.sever_toast(), resetdata[5]['except_result']) # 密码为空 def empty_psw(self): '''请输入密码''' self.login.send_code_reset(url,resetdata[6]['phone']) codes = self.login.get_code(sqldata[3]['set or search'], sqldata[3]['table'], sqldata[3]['where']) self.login.psw_reset(codes[0],resetdata[6]['psw']) self.assertEqual(self.toast.page_toast(), resetdata[6]['except_result']) # 修改密码成功后,登录 def reset_login(self): '''修改密码成功后,登录''' self.login.send_code_reset(url, resetdata[7]['phone']) codes = self.login.get_code(sqldata[3]['set or search'], sqldata[3]['table'], sqldata[3]['where']) self.login.psw_reset(codes[0], resetdata[7]['psw']) self.login.login_psw(logindata[5]['psw']) self.login.login_button() self.assertEqual(self.login.login_success_username(),logindata[5]['except_result']) # if __name__ == '__main__': # suite = unittest.TestSuite() # suite.addTest(Reset_password('reset_success')) # suite.addTest(Reset_password('reset_empty_phone')) # suite.addTest(Reset_password('error_code')) # suite.addTest(Reset_password('error_new_psw')) # suite.addTest(Reset_password('new_psw_error')) # suite.addTest(Reset_password('less_new_psw')) # suite.addTest(Reset_password('empty_psw')) # suite.addTest(Reset_password('reset_login')) # unittest.TextTestRunner().run(suite)