如何在python中同时调用两个函数调用？

Question

我有一个使用下面代码的函数。

def get_doc_hash(doc):
       return hashlib.md5(doc.encode(‘utf-8’)).hexdigest()

如何模拟md5()和十六进制()的调用来为该函数编写测试用例？

Answer 1

正如评论中所指出的，最好是测试预期的行为，但纯粹是为了展示如何连锁模仿，以下是如何做到的：

import hashlib
import unittest
from unittest.mock import patch


def get_doc_hash(doc):
    return hashlib.md5(doc.encode("utf-8")).hexdigest()


class TestHash(unittest.TestCase):
    @patch('hashlib.md5')
    def test_get_doc_hash_to_demonstrate_mock_chaining(self, mock_hashlib):
        mock_hashlib.return_value.hexdigest.return_value = "digested"
        hash_out = get_doc_hash("test doc")
        mock_hashlib.assert_called_once_with("test doc".encode("utf-8"))
        mock_hashlib.return_value.hexdigest.assert_called_once()
        self.assertEqual(hash_out, "digested")