Python Unittest根据不同测试环境跳过用例的方法

前言

在利用单元测试框架执行测试用例的过程中，有时只需要执行一部分用例，或者跳过某些暂不需要执行的用例，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)