7

我目前正在尝试模拟 Python 中的 open() 内置方法进行测试。但是，我总是最终会崩溃，并产生以下消息：

   File "/opt/home/venv/lib/python2.7/site-packages/nose-1.3.0-py2.7.egg/nose/result.py", line 187, in _exc_info_to_string
return _TextTestResult._exc_info_to_string(self, err, test)
 File "/opt/python-2.7.3/lib/python2.7/unittest/result.py", line 164, in _exc_info_to_string
msgLines = traceback.format_exception(exctype, value, tb)
 File "/opt/python-2.7.3/lib/python2.7/traceback.py", line 141, in format_exception
list = list + format_tb(tb, limit)
 File "/opt/python-2.7.3/lib/python2.7/traceback.py", line 76, in format_tb
return format_list(extract_tb(tb, limit))
  File "/opt/python-2.7.3/lib/python2.7/traceback.py", line 101, in extract_tb
line = linecache.getline(filename, lineno, f.f_globals)
  File "/opt/home/venv/lib/python2.7/linecache.py", line 14, in getline
lines = getlines(filename, module_globals)
 File "/opt/home/venv/lib/python2.7/linecache.py", line 40, in getlines
return updatecache(filename, module_globals)
 File "/opt/home/venv/lib/python2.7/linecache.py", line 127, in updatecache
with open(fullname, 'rU') as fp:
AttributeError: __exit__

这是我的测试代码：

m = mox.Mox()
m.StubOutWithMock(__builtin__, 'open')
mock_file = m.CreateMock(__builtin__.file)

open(mox.IgnoreArg(), mox.IgnoreArg()).AndReturn(mock_file)
mock_file.write(mox.IgnoreArg()).MultipleTimes()
mock_file.close()

write_file_method()
4

1 回答 1

4

__exit__是当您尝试关闭文件时调用的方法。您的模拟文件不处理mock_file.close()，只是open(). 您还需要模拟该close方法。

编辑：

但是，在第二个问题上，您为什么要嘲笑open？AFAIK你不应该这样做。被测方法应该采用开放流（例如，而不是文件名）。在生产代码中，客户端负责打开文件（例如pickle.dump）。在您的测试中，您传入一个StringIO, 或一个支持写作的模拟对象。

编辑2：我会将您的方法一分为二并分别测试每一位。

  • 创建文件：在调用此方法之前检查该文件是否不存在，之后是否存在。有人可能会争辩说这种单行方法不值得测试。
  • 写入文件：见上文。创建一个StringIO并写入它，这样您的测试就可以验证是否写入了正确的东西。
于 2014-07-01T14:45:30.713 回答