当两个函数调用在python中一起调用时，如何修补它们

Question

我有一个函数，它使用以下代码。

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

如何模拟md5((和hexsdigest((的调用来为该函数编写测试用例？

Answer 1

正如评论中所指出的，测试预期行为会更好，但纯粹是为了展示如何链接mock，下面是如何做到的：

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")