Selenium+Python ---- 免登录、等待、unittest单元测试框架、PO模型

url = "http://www.baidu.com"
driver = webdriver.Firefox()
driver.get(url)
time.sleep(3)
#添加cookies的方式 c1 = {'domain': '.baidu.com',
'name':'BDUSS',
'value': 'dyRlBucW9WOUhpNXducElPT1dlVHN1SFdFZUllTkJ-TEg5djNAAVe6gcaGFuc2hvdWthaQAAAAAAAAAAAAAAAAAAAAAAAAAAAAAHa68lh2uvJYU2',
'path': '/',
'httpOnly': True,
'secure':False
}
driver.add_cookie(c1)
time.sleep(3)
# print(driver.get_cookies()) #得到当前cookies信息
# driver.delete_all_cookies() #删除所有cookies信息
driver.refresh()
time.sleep(5) driver.quit()

#利用火狐profile文件的方式（前提：必须先手动登录一下）
profile_ff = "C:/Users/Hanxiaobei/AppData/Roaming/Mozilla/Firefox/Profiles/ytw908g4.default" fp = webdriver.FirefoxProfile(profile_ff)
driver = webdriver.Firefox(fp)
url = "http://www.baidu.com"
driver.get(url)
time.sleep(5)
driver.quit()

driver.find_element_by_link_text("登录").click()
driver.find_element_by_id("TANGRAM__PSP_8__userName").send_keys("hankai")
driver.find_element_by_id("TANGRAM__PSP_8__password").send_keys("hankai")
time.sleep(15) #导入等待时间人工输入验证码
driver.find_element_by_id("TANGRAM__PSP_8__submit").click() time.sleep(5)
driver.quit()

1 import time
2
3 time.sleep(5)

1 driver.implicitly_wait(20)

1 url = "http://www.baidu.com"
2 driver = webdriver.Firefox()
3 driver.get(url)
4 #显性等待
5 kk = WebDriverWait(driver,10).until(lambda driver:driver.find_element_by_id("kw"),message="worry!")
6 kk.send_keys("测试")

import unittest,time
from selenium import webdriver
from selenium.webdriver.common.by import By
class Baidu(unittest.TestCase):
def setUp(self):
self.driver = webdriver.Chrome()
url = "http://www.baidu.com"
self.driver.get(url)
self.driver.implicitly_wait(20)
self.verificationErrors = [] def tearDown(self):
self.driver.quit()
self.assertEqual([],self.verificationErrors,msg="验证失败") def test_search(self):
self.driver.find_element(By.ID,"kw").send_keys("hanxiaobei")
self.driver.find_element(By.ID,"su").click()
time.sleep(5) if __name__ == "__main__":
第一种形式：
#unittest.main() 第二种形式：
suite = unittest.TestSuite()
suite.addTest(Baidu("test_search")) runner = unittest.TextTestRunner()
runner.run(suite)

from selenium import webdriver
import time
import unittest
import HTMLTestRunner class Test_case(unittest.TestCase):
"""测试类"""
def setUp(self):
self.url = "http://www.baidu.com"
self.driver = webdriver.Firefox()
self.driver.implicitly_wait(20)
self.verificationErrors = [] def tearDown(self):
self.driver.quit()
self.assertEqual([],self.verificationErrors) def test_sou1(self):
"""测试搜索演示1"""
self.driver.get(self.url)
self.driver.find_element_by_id("kw").send_keys("测试")
self.driver.find_element_by_id("su").click()
self.driver.close()
time.sleep(5)
def test_sou2(self):
"""测试搜索演示2"""
self.driver.get(self.url)
self.driver.find_element_by_id("k").send_keys("自动化测试")
self.driver.find_element_by_id("su").click()
self.driver.close()
time.sleep(5) #生成一个运行测试用例集合
suite = unittest.TestSuite()
suite.addTest(Test_case('test_sou1'))
suite.addTest(Test_case('test_sou2')) '''
生成基于html的测试报告：
1定义一个文件的路径
2以写的方式进行打开文件
3调用htmltestrunner的方法生成测试报告
4运行测试集合
5关闭文件
'''
report_file=".\\20170423_report.html"
fp = open(report_file,"wb")
runner = HTMLTestRunner.HTMLTestRunner(stream=fp,title="搜索",description="测试搜索结果")
runner.run(suite)
fp.close()

from selenium.webdriver.support.wait import WebDriverWait '''
这个类主要是完成所有页面的一些公共方法的封装
'''
class Action(object):
#初始化
def __init__(self,se_driver):
self.driver = se_driver #定义open方法
def open(self,url):
self.driver.get(url)
self.driver.maximize_window() #重写元素定位的方法
def find_element(self,*loc):
try:
WebDriverWait(self.driver,20).until(lambda driver:driver.find_element(*loc).is_displayed())
return self.driver.find_element(*loc)
except Exception as e:
print("未找到%s"%(self,loc)) #定义script方法，用于执行js脚本
def script(self,src):
self.driver.execute_script(src) #重写send_keys方法
def send_keys(self,loc,value,clear_first=True,clik_first=True):
try:
if clik_first:
self.find_element(*loc).click()
if clear_first:
self.find_element(*loc).clear()
self.find_element(*loc).send_keys(value)
except AttributeError:
print("未找到%s"%(self,loc))

from selenium.webdriver.common.by import By
from seleniumframework.PO import base_page
import time class LoginPage(base_page.Action):
link_loc = (By.LINK_TEXT,"登录")
name_loc = (By.ID,"TANGRAM__PSP_8__userName")
password_loc = (By.ID,"TANGRAM__PSP_8__password")
submit_loc = (By.ID,"TANGRAM__PSP_8__submit") username_top = (By.LINK_TEXT,"hanxiaobei") def click_link(self):
self.find_element(*self.link_loc).click()
time.sleep(3) #等待3秒，等待登录弹窗加载完成 def run_case(self,value1,value2):
self.find_element(*self.name_loc).send_keys(value1)
self.find_element(*self.password_loc).send_keys(value2)
time.sleep(20) #手动输入验证码
self.find_element(*self.submit_loc).click()
time.sleep(5) #等待5秒，登录后的页面加载完成 def get_username(self):
return self.find_element(*self.username_top).text

import unittest
from selenium import webdriver
from seleniumframework.PO.login_page import LoginPage
import time class TestBaiduLogin(unittest.TestCase):
"""UI自动化登录"""
def setUp(self):
self.url = "http://www.baidu.com"
self.driver = webdriver.Firefox()
self.driver.implicitly_wait(20)
# self.verificationErrors = [] def tearDown(self):
time.sleep(5)
self.driver.quit()
# self.assertEqual([],self.verificationErrors) def test_login(self):
"""百度登录"""
sp = LoginPage(self.driver)
sp.open(self.url)
sp.click_link()
sp.run_case("hanxiaobei","xxxxxxx")
self.assertEqual(sp.get_username(),"hanxiaobei",msg="验证失败！")

import unittest
import HTMLTestRunner #相对路径
testcase_path = ".\\testcase"
report_path = ".\\report\\report.html"
def creat_suite():
uit = unittest.TestSuite()
discover = unittest.defaultTestLoader.discover(testcase_path,pattern="test_*.py")
for test_suite in discover:
# print(test_suite)
for test_case in test_suite:
uit.addTest(test_case)
return uit suite = creat_suite()
fp = open(report_path,"wb")
runner = HTMLTestRunner.HTMLTestRunner(stream=fp,title="测试结果",description="测试搜索结果")
runner.run(suite)
fp.close()

1 now = time.strftime("%Y-%m-%d-%H-%M-%S",time.localtime(time.time()))
2 print(now)
3 report_path = ".\\report\\"+now+"report.html"

from selenium.webdriver.common.by import By
from seleniumframework.PO import base_page #继承base后既可以调用base的方法也可自己添加新的方法
class SearchPage(base_page.Action): #通过id进行定位元素
search_loc = (By.ID,"kw") def run_case(self,value):
#第一种利用原生的send_keys方法
self.find_element(*self.search_loc).send_keys(value) #第二种利用二次封装的send_keys方法
# self.send_keys(self.search_loc,value)

import unittest
from selenium import webdriver
from seleniumframework.PO.search_page import SearchPage
import time class TestBaiduSearch(unittest.TestCase):
"""UI自动化搜索"""
def setUp(self):
self.url = "http://www.baidu.com"
self.driver = webdriver.Firefox()
self.driver.implicitly_wait(20)
self.verificationErrors = [] def tearDown(self):
time.sleep(5)
self.driver.quit()
self.assertEqual([],self.verificationErrors) def test_search(self):
"""搜索测试关键字"""
sp = SearchPage(self.driver)
sp.open(self.url)
sp.run_case("测试")