Selenium2+python自動化55-unittest之裝飾器（@classmethod）

標籤：需要   class   執行個體   元素   1.0   ddt   fir   odi   開啟   

前言

前面講到unittest裡面setUp可以在每次執行用例前執行，這樣有效減少了代碼量，但是有個弊端，比如開啟瀏覽器操作，每次執行用例時候都會重新開啟，這樣就會浪費很多時間。

於是就想是不是可以只開啟一次瀏覽器，執行完用例再關閉呢？這就需要用到裝飾器（@classmethod）來解決了。

 

一、裝飾器

1.用setUp與setUpClass區別

setup():每個測試case運行前運行
teardown():每個測試case運行完後執行
setUpClass():必須使用@classmethod 裝飾器,所有case運行前只運行一次
tearDownClass():必須使用@classmethod裝飾器,所有case運行完後只運行一次

[email protected]是修飾符，classmethod是python裡的類方法

 

二、執行順序

1.用類方法寫幾個簡單case，可以對比這篇：Selenium2+python自動化52-unittest執行順序

# coding:utf-8
import unittest
import time
class Test(unittest.TestCase):
    @classmethod
    def setUpClass(cls):
        print "start!"
   

    @classmethod
    def tearDownClass(cls):
        time.sleep(1)
        print "end!"

    def test01(self):
        print "執行測試案例01"

    def test03(self):
        print "執行測試案例03"

    def test02(self):
        print "執行測試案例02"

    def addtest(self):
        print "add方法"

if __name__ == "__main__":
    unittest.main()

2.從執行結果可以看出，前置和後置在執行用例前只執行了一次。

start!
執行測試案例01
執行測試案例02
執行測試案例03
...end!

----------------------------------------------------------------------
Ran 3 tests in 1.001s

 

三、selenium執行個體

1.可以把開啟瀏覽器操作放到前置setUpClass(cls)裡，這樣就可以實現開啟一次瀏覽器，執行多個case了

# coding:utf-8
from selenium import webdriver
from selenium.webdriver.support import expected_conditions as EC
import unittest
class BolgHome(unittest.TestCase):
    u‘‘‘部落格首頁‘‘‘
    @classmethod
    def setUpClass(cls):
        cls.driver = webdriver.Firefox()
        url = "http://www.cnblogs.com/yoyoketang/"
        cls.driver.get(url)
        cls.driver.implicitly_wait(30)

    @classmethod
    def tearDownClass(cls):
        cls.driver.quit()

    def test_01(self):
        u‘‘‘驗證元素存在：部落格園‘‘‘
        locator = ("id", "blog_nav_sitehome")
        text = u"部落格園"
        result = EC.text_to_be_present_in_element(locator, text)(self.driver)
        self.assertTrue(result)

    def test_02(self):
        u‘‘‘驗證元素存在：首頁‘‘‘
        locator = ("id", "blog_nav_myhome")
        text = u"首頁"
        result = EC.text_to_be_present_in_element(locator, text)(self.driver)
        self.assertTrue(result)

if __name__ == "__main__":
    unittest.main()

Selenium2+python自動化55-unittest之裝飾器（@classmethod）