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)