如何导入自己的模块进行模拟？ （导入错误：没有名为my_module的模块！）

Question

我想对我的类进行单元测试，该类位于另一个名为client_blogger.py的文件中。

我的单元测试文件位于同一目录中。我的所有其他单元测试都有效，除非我尝试模拟我自己的一种方法。

## unit_test_client_blogger.py
import mock
import json

from client_blogger import BloggerClient, requests

Class TestProperties():

    @pytest.fixture

    def blog(self):
    return BloggerClient(api_key='123', url='http://example.com')

    @mock.patch('client_blogger._jload')
    @mock.patch('client_blogger._send_request')

    def test_gets_blog_info(self, mock_send, mock_jload):
    """ Get valid blog info from API response. """

    valid_blog_info = 'some valid json api response here'
    parsed_response = json.loads(valid_blog_info)
    correct_blog_id = '7488272653173849119'
    mock_jload.return_value = valid_blog_info

    id = self.blog().get_blog_info(parsed_response)
    assert id == correct_blog_id

这是client_blogger.py文件内容：

# client_blogger.py
import requests, json

class BloggerClient(object):
    """ Client interface for Blogger API. """
    def __init__(self, key, url):
         # removed some code here for brevity

    def _send_request(self, api_request):
        """ Sends an HTTP get request to Blogger API.
            Returns HTTP response in text format. """
        # snip

    def _jload(self, api_response):
        """ Accepts text API response. Returns JSON encoded response. """
        # snip

    def get_blog_info(self):
        """ Makes an API request. Returns Blog item information. """
        request = '{b}/blogs/byurl?url={u}&key={k}'.format(b=self.base, u=self.url, k=self.key)
        txt_response = self.send_request(request)
        response = self._jload(txt_response)
        return response['id']

我想在上面的方法中模拟self.send_request()和self._jload()方法调用。

但Mock模块抱怨：ImportError: No module named client_blogger。

错误必须在这里：

@mock.patch('client_blogger._jload')
@mock.patch('client_blogger._send_request')

为了让mock.patch找到我的模块或类，我尝试了很多变化。但他们都没有工作。

我尝试过以下方法：

@mock.patch('client_blogger.BloggerClient._jload')
@mock.patch('BloggerClient._jload')
@mock.patch('._jload')

这些都不起作用。知道如何从我自己的模块中模拟一个方法吗？

（看起来很奇怪，因为我可以mock.patch其他模块，而不是我自己的：-s）

Answer 1

你想要这个：

@mock.patch('client_blogger.BloggerClient._jload')
@mock.patch('client_blogger.BloggerClient._send_request')
def test_gets_blog_info(self, mock_send, mock_jload):
    """ Get valid blog info from API response. """

    valid_blog_info = 'some valid json api response here'
    parsed_response = json.loads(valid_blog_info)
    correct_blog_id = '7488272653173849119'
    mock_jload.return_value = valid_blog_info

    id = self.blog().get_blog_info(parsed_response)
    assert id == correct_blog_id

BloggerClient实施来自client_blogger模块，因此您需要修补client_blogger.BloggerClient。你列举的那个是你试过的不起作用的东西，但我只是尝试过，它对我来说很好。你尝试的时候有什么问题？

Answer 2

对于python3，格式如下：

from unittest.mock import patch
@patch('client_blogger.BloggerClient._jload')
.
.
.

文档：https://docs.python.org/3/library/unittest.mock.html#patch

这非常非常重要：

  

patch（）易于使用。关键是在正确的名称空间中进行修补。请参阅修补程序部分。