unittest模塊使用方法

unittest模塊常用屬性

1、 unittest.TestCase類：所有的測試用例類繼承的基類

定義一個測試用例類，需要繼承TestCase，比如：

class BaiduTest(unittest.TestCase):

2、 unittest.main():使用這個可以方便的將一個測試模塊變為可直接運行的測試腳本，main()方法使用TestLoader類來搜索所有包含在該模塊中以test命名開頭的測試方法，並自動執行它們。執行方法的默認順序是：根據ASCII碼的順序加載測試用例，數字與字母的順序為：0-9，A-Z，a-z。

3、 unittest.TestSuite()：用來創建測試套件

4、 unittest.TextTestRunner():通過該類的run()方法可運行suite組裝的測試用例。

5、 unittest.defaultTestLoader()：通過該類下的discover()方法可以自動根據測試目錄匹配查找到的測試用例文件，並將查找到的測試用例組裝到測試套件，因此可以直接通過run()方法執行discover。例如：

discover = unittest.defaultTestLoader.discover(test_dir, pattern=’test_*.py’)

第一個參數是測試用例目錄，第二個參數是查找以test_開頭的py文件

6、 unittest.skip()：裝飾器，當運行用例時，有些用例可能不想執行等，可以用裝飾器暫時屏蔽該條測試用例。一種常見的用法就是比如說想調試某一個測試用例，想先屏蔽其他用例，就可以用裝飾器屏蔽。

a) @unittest.skip(reason):skip(reason)這個裝飾器是無條件跳過裝飾器所裝飾的case，並說明跳過測試的原因

b) @unittest.skipIf(reason):skipIf(condition, reason):條件為真時，跳過裝飾的case，並說明跳過原因。

c) @unittest.skipUnless(reason):skipUnless(condition, reason):條件為假時，跳過裝飾的測試，並說明跳過測試的原因

d) @unittest.expecteFailure():expecteFailure():測試標記為失敗

TestCase類常用屬性

1. setUp()：setUp()方法是用於測試用例執行前的初始化工作。如測試用例需要訪問數據庫，可以在setUp()中建立數據庫連接並初始化，需要登錄web，可以先實例化瀏覽器等。
2. tearDown()：用於測試用例執行之后的善后工作，比如關閉數據庫連接，關閉瀏覽器等。
3. assertEqual(a, b, [msg=測試失敗時打印的信息]):斷言a,b兩個值是否相等，相等則測試通過。
4. assertNotEqual(a, b, [msg=測試失敗時打印的信息]):斷言a,b兩個值是否相等，不相等則測試通過.
5. assertTrue(x, [msg]):斷言x是否True，true則測試通過
6. assertFalse(x, [msg]):斷言x是否False，False則測試通過
7. assertIs(a, b, [msg])：斷言a是否是b，是則測試通過
8. assertNotIs(a, b, [msg])：斷言a是否是b，不是則測試通過
9. assertIsNone(x, [msg]):斷言x是否是None，是None則測試通過
10. assertIsNotNone(x, [msg]):斷言x是否是None，是None則測試通過
11. assertIn(a, b, [msg])：斷言a是否在b中，是則測試通過
12. assertNotIn(a, b, [msg])：斷言a是否在b中，不在則測試通過
13. assertIsInstance(a, b, [msg])：斷言a是否是b的一個實例，是則測試通過
14. assertNotIsInstance(a, b, [msg])：斷言a是否是b的一個實例，不是則測試通過

 

舉例：

class TestSub(unittest.TestCase):

    def setUp(self):
        """為測試方法初始化，每個case運行前執行一次"""
        self.a = 20
        self.b = 10

    def tearDown(self):
        """清理函數，每個case執行后執行一次"""
        pass

    def test_sub(self):
        """驗證減法"""
        result = self.a - self.b
        self.assertEqual(result, 10)

 

def test_add(self):
        """驗證加法"""
        result = self.a + self.b
        self.assertEqual(result, 30)

 

TestSuite類常用屬性

addTest()：將測試用例添加到測試套件中，比如：

#將test_baidu模塊下的BaiduTest類下的test_baidu_serach和test_baidu_login測試用例添加到套件

suite = unittest.TestSuite()

suite.addTest(TestSub ("test_sub"))

suite.addTest(TestSub ("test_add "))

runner = unittest.TextTestRunner()

runner.run(suite)

TextTestRunner類常用屬性

run()方法是運行測試套件中的測試用例,參數是一個測試套件，例如：

runner = unittest.TextTestRunner()

runner.run(suite)

 

if __name__ == '__main__':
    # 可以直接調用main方法執行測試用例
    # unittest.main()

    # 也可以像下面這樣先創建測試套件，然后將測試用例放入套件，再使用run方法運行套件
    test_dir = './'
    discover = unittest.defaultTestLoader.discover(test_dir, pattern='test_*.py')
    runner = unittest.TextTestRunner()
    runner.run(discover)

 

綜合實例

實例1：

# coding=utf-8

"""
create on 2019.12.26
@author:suyang
Project:使用unittest編寫測試用例
"""
import unittest

# 定義測試類，繼承TestCase的所有方法
class Test(unittest.TestCase):
    # 重寫父類的setUp方法10

    def setUp(self):
        self.number = input("Enter a number:")
        self.number = int(self.number)

    # 定義第一個測試用例
    def test_case1(self):
        print(self.number)
        self.assertEqual(self.number, 10, "Your input is not 10")

     # 定義第二個測試用例
    @unittest.skip("暫時跳過用例2的測試")
    def test_case2(self):
        print(self.number)
        self.assertEqual(self.number, 20, "Your input is not 20")

    def test_case3(self):
        result = self.number + 10
        self.assertEqual(result, 30, "Your input is not 20")

    def tearDown(self):
        print('test over')

# 3種方法都可以執行測試用例
if __name__ == '__main__':
    # 可以直接調用main方法執行測試用例
    # unittest.main()

    # 先創建一個測試套件，然后將用例依次用addTest方法加入測試套件中，再用run方法執行用例
    # suit = unittest.TestSuite()
    # suit.addTest(Test('test_case1'))
    # suit.addTest(Test('test_case2'))
    # suit.addTest(Test('test_case3'))
    # runner = unittest.TextTestRunner()
    # runner.run(suit)

    # 也可以像下面這樣先創建測試套件，然后將測試用例放入套件，再使用run方法運行套件
    test_dir = './'
    discover = unittest.defaultTestLoader.discover(test_dir, pattern='test_*.py')
    runner = unittest.TextTestRunner()
    runner.run(discover)

 

 

實例2：

# coding=utf-8
"""
created on 2019.12.26
@author:suyang
Project:測試百度搜索
"""
from selenium import webdriver
import unittest
import time


class BaiduTest(unittest.TestCase):
    def setUp(self):
        # pass
        self.driver = webdriver.Chrome()
        # 隱性等待時間為30秒
        self.driver.implicitly_wait(30)
        self.base_url = "https://www.baidu.com"

    def test_baidu(self):
        driver = self.driver
        driver.get(self.base_url+'/')
        driver.find_element_by_id("kw").clear()
        driver.find_element_by_id("kw").send_keys("unittest")
        driver.find_element_by_id("su").click()
        time.sleep(3)
        title = driver.title
        self.assertEqual(title, u"unittest_百度搜索")

    def tearDown(self):
        self.driver.quit()


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