3

我在测试一个方法是否被调用时遇到了麻烦mock——举个简单的例子，假设该方法是os.getcwd. 我想测试我自己的函数 ,是否按预期pickle_wdir调用。os.getcwd但是，我正在测试的函数腌制由返回的值os.getcwd，这会导致错误。

这是一个重现错误的简单示例。

os_ex.py：

import os
import pickle

def pickle_wdir(filename):
    dir = os.getcwd()
    with open(filename, 'wb') as handle:
        pickle.dump(dir, handle)

test_os_ex.py：

from unittest import TestCase
from unittest.mock import patch
from os_ex import pickle_wdir


class TestPickleWdir(TestCase):
    def test_os_called(self):
        fname = 'dir.pickle'
        with patch('os_ex.os') as mocked_obj:
            pickle_wdir(fname)
            mocked_obj.getcwd.assert_called()

返回的错误信息是

_pickle.PicklingError: Can't pickle <class 'unittest.mock.MagicMock'>: it's not the same object as unittest.mock.MagicMock.

我如何在os.getcwd没有得到这个的情况下测试那个被调用的PicklingError？

4

1 回答 1

1

您需要模拟os.getcwd(),open()和pickle.dump()方法。您可以使用unittest.mock.patch作为上下文管理器来执行此操作。

例如

os_ex.py：

import os
import pickle


def pickle_wdir(filename):
    dir = os.getcwd()
    with open(filename, 'wb') as handle:
        pickle.dump(dir, handle)

test_os_ex.py：

import unittest
from unittest.mock import patch, mock_open
from os_ex import pickle_wdir


class TestOsEx(unittest.TestCase):
    def test_pickle_wdir(self):
        fname = 'dir.pickle'
        m = mock_open(read_data='mocked data')
        with patch('os_ex.os') as mocked_obj, patch('builtins.open', m) as mocked_open, patch('pickle.dump') as mocked_dump:
            mocked_obj.getcwd.return_value = '/root'
            pickle_wdir(fname)
            mocked_obj.getcwd.assert_called()
            m.assert_called_with(fname, 'wb')
            handle = mocked_open()
            mocked_dump.assert_called_with('/root', handle)


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

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

.
----------------------------------------------------------------------
Ran 1 test in 0.012s

OK
Name                                       Stmts   Miss  Cover   Missing
------------------------------------------------------------------------
src/stackoverflow/60627827/os_ex.py            6      0   100%
src/stackoverflow/60627827/test_os_ex.py      17      0   100%
------------------------------------------------------------------------
TOTAL                                         23      0   100%
于 2020-04-27T11:19:27.293 回答