本篇轉自博客：上海-悠悠

原文地址：http://www.cnblogs.com/yoyoketang/tag/unittest/

前言

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

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

一、裝飾器

1.用setUp與setUpClass區別

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

2.@是修飾符，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）【轉載】