Python 中的单元测试：如何让 if 语句返回 TRUE 以测试 if 语句中的行？

Question

Python 3.7。我正在尝试对以下功能进行单元测试：

    def is_json_clean():
        # function for is_json_clean here, returns bool

    def json_function(blob, raw_bucket, lake_bucket):
        data_string = blob.download_as_bytes()
        data = json.loads(data_string)
        if is_json_clean(data) is True:
            raw_bucket.copy_blob(blob, lake_bucket)

我对模拟补丁还很陌生，可以在这里举个例子！我如何使用 mock.patch 或类似的东西来获取带有 is_json_clean() 函数的 if 语句来返回 True 以便我可以断言调用了 copy_blob 函数？我想我可以制作一个像这样的补丁：

@mock.patch.object(imported_module, "is_json_clean", return_value=True)
def test_json_function(mock):
    # rest of the test function 
    # including assert statement for the copy_blob function 

但是我如何将它插入到我的测试中以断言我的 copy_blob 函数？我知道有很多与单元测试相关的问题，但我还没有找到我的问题的确切答案（就像我说的，我很新）。任何帮助表示赞赏。谢谢！

Answer 1

您的模拟看起来是正确的，但我没有看到在您的测试用例中传递了 blob、raw_bucket 或 lake_bucket 参数。以下是我将如何编写它，嘲笑所有 3 个参数。

import unittest

from unittest import mock

import imported_module

class MyTest(unittest.TestCase):
    @mock.patch.object(imported_module, "is_json_clean", return_value=True)
    def test_json_function(self, mock_is_json_clean):
        json_blob = '{"mykey": "myvalue"}'
        mock_blob = mock.Mock()
        mock_blob.download_as_bytes.return_value = json_blob
        mock_raw_bucket = mock.Mock()
        mock_lake_bucket = mock.Mock()

        imported_module.json_function(mock_blob, mock_raw_bucket, mock_lake_bucket)

        mock_raw_bucket.copy_blob.assert_called_once_with(mock_blob, mock_lake_bucket)

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

可以看到我传入了一个mock_blob，mock了download_as_bytes的返回值，同时传入了两个bucket。从那里，我可以对mock_raw_bucket.copy_blob 调用进行断言。我能够让这段代码在这里运行和传递：https://replit.com/@WesH1/TautLiveTelephone#main.py