在 python 中，如何为使用打开文件的函数编写单元测试？答案

【问题标题】：In python, how to write unit test for a function is using with open file?在 python 中，如何为使用打开文件的函数编写单元测试？
【发布时间】：2020-02-12 08:06:32
【问题描述】：

这是获取版本并使用 open 从文件路径位置读取文件的功能。

def get_version(self):
    try:
        with open("file_path") as openfile:
            for line in openfile:
                sline = line.split()
                for row, column in enumerate(sline):
                    if column == "version=":
                        version = sline[row+1].strip('"')
    return version

【问题讨论】：

请使用正确的缩进正确格式化您的代码。
这只是你功能的一部分，因为我没有看到except 块。
嗨@sid，欢迎！看起来有一些类似的问题，关于单元测试内置函数，例如open()。以here 为起点。
除了块在UT部分处理

标签： python python-unittest python-unittest.mock

【解决方案1】：

您可以使用unittest.mock.mock_open(mock=None, read_data=None) 模拟open 函数。

例如

main.py:

class MyClass:
    def get_version(self):
        version = ''
        with open("file_path") as openfile:
            for line in openfile:
                sline = line.split()
                for row, column in enumerate(sline):
                    if column == "version=":
                        version = sline[row+1].strip('"')

        return version

test_main.py:

from main import MyClass
import unittest
from unittest.mock import mock_open, patch


class TestMain(unittest.TestCase):
    def test_get_version(self):
        m = mock_open(read_data='version= 1.0.0')
        with patch('builtins.open', m) as mocked_open:
            myclass_instace = MyClass()
            version = myclass_instace.get_version()
            self.assertEqual(version, '1.0.0')
            m.assert_called_with('file_path')


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

100% 覆盖率的单元测试结果：

.
----------------------------------------------------------------------
Ran 1 test in 0.011s

OK
Name                                      Stmts   Miss  Cover   Missing
-----------------------------------------------------------------------
src/stackoverflow/60183706/main.py           10      0   100%
src/stackoverflow/60183706/test_main.py      14      0   100%
-----------------------------------------------------------------------
TOTAL                                        24      0   100%

Python 版本：Python 3.7.5

【讨论】：