上一篇介紹瞭如何搭建selenium環境，也給出了一小段程式碼演示如何使用selenium
部落格不會過多的介紹如何使用selenium api 更多地在於如何構建一個自動化測試專案，
程式碼的分層是很關鍵一步,本文介紹如何使用pytest+selenium的框架做到測試程式碼與指令碼程式碼的分離

pytest是一個非常成熟的全功能的Python測試框架，主要特點有以下幾點：

1、簡單靈活，容易上手；
2、支援引數化；
3、能夠支援簡單的單元測試和複雜的功能測試，還可以用來做selenium/appnium等自動化測試、介面自動化測試（pytest+requests）;
4、pytest具有很多第三方外掛，並且可以自定義擴充套件，比較好用的如pytest-selenium（整合selenium）、pytest-html（完美html測試報告生成）、pytest-rerunfailures（失敗case重複執行）、pytest-xdist（多CPU分發）等；
5、測試用例的skip和xfail處理；
6、可以很好的和jenkins整合；

1、pytest安裝

pip install -U pytest   

2、pytest執行程式碼指令碼

if __name__ == '__main__':
    pytest.main("-q test_xxx.py")

3、pytest的fixtures

setup_class/teardown_class       在當前測試類的開始與結束執行。

setup/treadown                   在每個測試方法開始與結束執行。

setup_method/teardown_method     在每個測試方法開始與結束執行，與setup/treadown級別相同。
 

copy一段下午寫的程式碼
test_youdanweb.py：

#coding=utf-8
import pytest
from youdanweb import YoudanWeb
youdan = YoudanWeb()
class Testlogin:
    def setup_class(self):
        youdan.initialize()
    def teardown_class(self):
        youdan.quit()
    def test_loginfail(self):
        assert u'佣金' in youdan.login('15914145610'
 
,'456789')
    def test_loginpass(self):
        assert u'佣金' in youdan.login('15914145610','123456')
if __name__ == '__main__':
    pytest.main("-q test_youdanweb.py")

youdanweb.py:

#usr/bin/python
#coding=utf-8
from selenium import webdriver
from selenium.webdriver.common.keys import Keys
import os
import time
class YoudanWeb():
    def initialize(self):
        chromedriver = "C:\Program Files (x86)\Google\Chrome\Application\chromedriver.exe"
        os.environ["webdriver.chrome.driver"] = chromedriver
        self.driver = webdriver.Chrome(chromedriver)
        self.driver.maximize_window()
        self.driver.implicitly_wait(10)
        self.base_url ="https://m.youdan.me"
    def youdanSearch(self,keywords):
        driver = self.driver
        driver.get(self.base_url +"/")
        time.sleep(2)
        driver.find_element_by_xpath(".//a[@class='sprite search-ico txt-over']").click()
        input = driver.find_element_by_id('filterValue')
        input.send_keys(keywords)
        time.sleep(2)
        input.send_keys(Keys.ENTER)
        return driver.page_source
    def login(self,user,pwd):
        driver = self.driver
        driver.get(self.base_url +"/")
        driver.find_element_by_xpath(".//a[@class='nav-ico']").click()
        time.sleep(1)
        driver.find_element_by_xpath(".//a[@class='blue-btn']").click()
        time.sleep(2)
        driver.find_element_by_id('mobile').send_keys(user)
        driver.find_element_by_id('password').send_keys(pwd)
        time.sleep(1)
        driver.find_element_by_id("next").click()
        time.sleep(1)
        return driver.page_source
    def quit(self):
        self.driver.quit()
    def save_screenshot(self,name):
        self.driver.save_screenshot('%s.png'%name)