前言

在利用單元測試框架執行測試用例的過程中，有時只需要執行一部分用例，或者跳過某些暫不需要執行的用例，python的unittest框架就內置這樣的功能。

本文章會講述以下幾個內容：

1、Unittest 如何跳過用例

2、如何使用sys.argv

3、自動化測試項目中如何一套代碼多套環境運行

一、Unittest跳過用例

例：

test_case_skip.py

# encoding:utf8import unittestclass SkipExample(unittest.TestCase): @unittest.skip('用例 1 無條件跳過') def test_case_one(self): print('---用例 1 ---') @unittest.skipIf(2 > 1, '條件為True ，用例2 跳過') def test_case_two(self): print('---用例 2 ---') @unittest.skipUnless(2 < 1, '條件為False, 用例3 跳過') def test_case_three(self): print('---用例 3 ---')if __name__ == '__main__': unittest.main(verbosity=2)

運行結果：

test_case_one (__main__.SkipExample) ... skipped '用例 1 無條件跳過'
test_case_two (__main__.SkipExample) ... skipped '條件為True ，用例2 跳過'
test_case_three (__main__.SkipExample) ... skipped '條件為False, 用例3 跳過'

二、如何使用sys.argv

例：

how_to_use_argv.py

#encoding:utf8from sys import argvprint('argv是一個數組：',argv)

使用命令行運行上述腳本，外部傳入參數：1 2 3 4

python how_to_use_argv.py 1 2 3 4

運行結果

argv是一個數組： ['how_to_use_argv.py', '1', '2', '3', '4']

小結：

三、自動化測試項目中如何一套代碼多套環境運行

需求1：一套代碼可以測試多個環境，不希望每次測試不同環境的時候都要去改代碼里面的URL，希望把代碼里面的URL參數化
以UI自動化為例：

test_multiple_env.py

# encoding:utf8from selenium import webdriverfrom sys import argvimport unittestfrom time import sleepclass TestEnv(unittest.TestCase): def setUp(self): self.url = argv[-1] print(self.url) self.driver = webdriver.Chrome() def test_load_page(self): self.driver.get(self.url) sleep(10)if __name__ == '__main__': suit = unittest.TestSuite() suit.addTest(TestEnv('test_load_page')) runner = unittest.TextTestRunner() runner.run(suit)