class Test_psw_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_login(self): '''账号密码正确登录''' # login = Login(self.driver) self.login.psw_login(logindata[0]['phone'], logindata[0]['psw'], url) self.assertEqual(self.login.login_success_username(), logindata[0]['except_result']) def test_login_nullpsw(self): '''账号正确,密码为空登录''' self.login.psw_login(logindata[1]['phone'], logindata[1]['psw'], url) self.assertEqual(self.toast.page_toast(), logindata[1]['except_result']) def test_login_nullphone(self): '''账号为空,密码正确''' self.login.psw_login(logindata[2]['phone'], logindata[2]['psw'], url) self.assertEqual(self.toast.page_toast(), logindata[2]['except_result']) def test_login_allnull(self): '''账号和密码都为空''' self.login.psw_login(logindata[3]['phone'], logindata[3]['psw'], url) self.assertEqual(self.toast.page_toast(), logindata[3]['except_result']) def test_login_mismatch(self): '''账号和密码不匹配''' self.login.psw_login(logindata[4]['phone'], logindata[4]['psw'], url) self.assertEqual(self.toast.sever_toast(), logindata[4]['except_result'])
class Cookie(BasePage): # def __init__(self): # self.driverBase = DriverBase() # self.driver = self.driverBase.open_broswer() # 构造方法中写打开浏览器,会导致没调用一个方法时,都会多打开一次浏览器 # self.login = Login(self.driver) # 封装元素操作方法 def save_cookie(self, phone, psw, url): self.login = Login(self.driver) self.login.psw_login(phone, psw, url) login_cookie = self.get_login_cookie() yamlpath = os.path.abspath( os.path.dirname(__file__)) + '\login_cookie.yaml' cookie_value = login_cookie with open(yamlpath, 'w', encoding='utf-8') as f: yaml.dump(cookie_value, f, Dumper=yaml.RoundTripDumper) sleep(3) # self.quit_driver() # 加上这一句,同时执行sage_cookie、keep_login时,会报 由于目标计算机积极拒绝,无法连接 的错误 # self.driverBase.quit_broswer() # self.driver.delete_cookie(name = 'laravel_session') # self.driver.refresh() def get_cookie(self, yamlName="login_cookie.yaml"): # self.save_cookie(phone,psw,url) f = os.path.abspath(os.path.dirname(__file__)) p = f + '\login_cookie.yaml' f = open(p) value = f.read() cookie_all = yaml.load(value) cookie_name = cookie_all['name'] cookie_value = cookie_all['value'] cookie_data_dict = {'name': cookie_name, 'value': cookie_value} # print(cookie_data_dict) return cookie_data_dict def keep_login(self, url): cookieData = self.get_cookie() self.open_url(url) self.add_cookie(cookieData) self.open_url(url) self.driver.implicitly_wait(10)
class AddCompany(unittest.TestCase): def setUp(self): self.driver = driverbase.open_broswer() driverbase.max_window() self.driver.implicitly_wait(10) self.add = Add_company(self.driver) self.login = Login(self.driver) self.toast = ToastText(self.driver) def tearDown(self): driverbase.quit_broswer() # 公司名为空 def test_add_nullname(self): self.add.save_login_cookie(readconfig.inputPhone_cookie, readconfig.inputPsw_cookie, readconfig.url_login) # 登录一次,后面的不用再次登录 self.add.add_company(companyName[0]['name'], readconfig.url_admin) self.assertEqual(self.toast.page_toast(), companyName[0]['except_result']) # 添加公司,默认行业 def test_default_vocation(self): self.add.add_company(companyName[1]['name'], readconfig.url_admin) self.assertEqual(self.add.companyName(), companyName[1]['except_result']) # 添加公司,选择教育培训行业 def test_select_education(self): self.add.add_company_education(companyName[2]['name'], readconfig.url_admin) self.assertEqual(self.add.companyName(), companyName[2]['except_result']) #添加公司,选择水利水电行业 def test_select_waterboard(self): self.add.add_company_waterboard(companyName[3]['name'], readconfig.url_admin) self.assertEqual(self.add.companyName(), companyName[3]['except_result']) #添加已存在的公司 def test_exist_company(self): self.add.add_company_education(companyName[4]['name'], readconfig.url_admin) self.assertEqual(self.toast.sever_toast(), companyName[4]['except_result']) # 添加已被认证的公司 def test_attestation_company(self): self.add.add_company_education(companyName[5]['name'], readconfig.url_admin) self.assertEqual(self.toast.sever_toast(), companyName[5]['except_result']) # 新账号首次添加公司 def test_first_addcompany(self): self.login.psw_login(logindata[0]['phone'], logindata[0]['psw'], readconfig.url_login) self.add.first_login() self.add.add_companyName(companyName[7]['name']) self.add.add_company_verify() self.assertEqual(self.add.companyName(), companyName[7]['except_result']) # if __name__ == '__main__': # suite = unittest.TestSuite() # suite.addTest(AddCompany('test_add_nullname')) # suite.addTest(AddCompany('test_default_vocation')) # suite.addTest(AddCompany('test_select_education')) # suite.addTest(AddCompany('test_select_waterboard')) # suite.addTest(AddCompany('test_exist_company')) # suite.addTest((AddCompany('test_exist_company'))) # suite.addTest((AddCompany('test_first_addcompany'))) # unittest.TextTestRunner().run(suite)