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

1、免登录
在进行测试的过程中难免会遇到登录的情况，给测试工作添加了工作量，本文仅提供一些思路供参考

解决方式：手动请求中添加cookies、火狐的profile文件记录信息实现、人工介入、万能验证码、去掉验证码

1.1、手动在请求中添加cookies信息

 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()

  1.2、火狐的profile文件记录信息实现

 #利用火狐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()

1.3、人工介入

 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.4、万能验证码、去掉验证码

万能验证码、去掉验证码需要开发的配合

2、等待

2.1、time模块

1 import time
2
3 time.sleep(5)

2.2、隐式等待

1 driver.implicitly_wait(20) 

2.3、显式等待

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("测试")

3、unittest单元测试框架

简单的unittest框架代码如下：

 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)

可生成html报告的unittest框架代码如下：

 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()

5、PO模型

PO就是一个设计思想，将代码以页面为单位进行组织，针对这个页面上的所有信息、相关操作都放到一个类中，从而使具体的测试用例变成了简单的调用和验证操作。

优点：进行了拆分和分层

缺点：对于复杂的业务page层变了，case也需要去改动

PO模型的目录结构：

其中，base_page是login_page、search_page的基础。test_login调用login_page，login_page调用base_page，同理test_search。

PO代码示例：

base_page.py

 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))

login_page.py

 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

test_login.py

 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="验证失败！")

main.py 运行的主入口

 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"

测试报告截图：

search_page.py

 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)

test_search.py

 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("测试")