4. 让Mock的函数抛出exception
def is_error(self):
try:
os.mkdir("11")
return False
except Exception as e:
return True
class TestProducer(unittest.TestCase):
@mock.patch('os.mkdir')
def test_exception(self, mock_mkdir):
mock_mkdir.side_effect = Exception
self.assertEqual(self.calculator.is_error(), True)
  5. Mock多个函数,主要是注意顺序
@mock.patch.object(Calculator, 'add')
@mock.patch('test_unit.multiple')
def test_both(self, mock_multiple, mock_add):
mock_add.return_value = 1
mock_multiple.return_value = 2
self.assertEqual(self.calculator.add(8, 14), 1)
self.assertEqual(multiple(8, 14), 2)

  Coverage
  打命令coverage加测试文件,可以得到覆盖率,可以生成html格式的报告,每次运行一个文件都会生成一个.coverage文件,需要将combine所有结果才能得到一个完整的报告。
  具体的命令参数参看:http://nedbatchelder.com/code/coverage/cmd.html
  更加有用的是配置文件,参看:http://nedbatchelder.com/code/coverage/config.html
  配置文件中有用的功能是可以不测某些行的覆盖率,例如:

 

[report]
exclude_lines =
# 只要在某一行加上注释“# pragma: no cover”这一行会被忽略
pragma: no cover
# 忽略掉main函数
if __name__ == .__main__.:

  Nose
  Nose可以将所有的单元测试文件一次全部执行,并且提供了coverage的插件,能够统计整体的覆盖率。
  Nose会扫描目标目录,如果发现目录名以“test”或者“Test”开头,则递归地进去扫描,并自动运行所有发现的以“test”或者“Test”开头的测试文件。
  另外Nose增加了报级别的setup和teardown,只需将他们放到__init__.py文件中即可。
  Nose命令的执行,简单的是nosetest后面加上你的所有测试文件或者测试文件所在的目录,一些运行参数参看:http://nose.readthedocs.org/en/latest/usage.html
  Nose的参数里面以"--cover"开头的都是coverage相关的,但是我发现并没有办法是用coverage的配置文件,需要手动安装一下nose-cov,然后用“--cov-config”来指定配置文件,其他参数参看:https://pypi.python.org/pypi/nose-cov
  我的项目因为测试文件比分散,并且有些并没有以test开头,所以比较麻烦,只能写了一个脚本,把这些都串起来:

 

import os
import subprocess
######################################################################
# 需要测试覆盖率的文件或者目录
cover_list = [
'src/sample/analyzer/unpacker/src/emulator.py',
'src/sample/analyzer/unpacker/src/emulator_manager.py',
'src/sample/analyzer/unpacker/src/unpacker_analyzer.py',
'src/sample/analyzer/bitvalue/src/confparser.py',
'src/sample/analyzer/bitvalue/src/trunk.py',
]
# 测试用例所在的文件或者目录,如果测试文件没有以test开头,则必须制定文件名
ut_list = [
'src/sample/analyzer/unpacker/ut',
'src/sample/analyzer/bitvalue/ut/ut_main.py'
]
######################################################################
PRODUCTION_HOME = os.environ.get("PRODUCTION_HOME", "../..")
def get_command():
command = [
'nosetests',
'--with-cov',
'--cover-erase',
'--cov-report', 'html',
'--cov-config', 'cover.config',
]
for cover in cover_list:
command.append('--cov')
command.append(os.path.join(PRODUCTION_HOME, cover))
for ut in ut_list:
command.append(os.path.join(PRODUCTION_HOME, ut))
return command
if __name__ == '__main__':
command = get_command()
print command
os.chdir(PRODUCTION_HOME)
proc = subprocess.Popen(command, shell=False, stdout=subprocess.PIPE, stderr=subprocess.PIPE)
output, error = proc.communicate()
return_code = proc.poll()
print output
print error
print return_code