如何导入自己的模块进行模拟?(导入错误:没有名为my_module!的模块!)

2024-04-26 17:21:35 发布

您现在位置:Python中文网/ 问答频道 /正文

我想对我的类进行单元测试,它位于另一个名为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()方法调用。

但模拟模块抱怨道: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)


Tags: 方法selfinfoclientsendresponserequestdef
2条回答

对于python3,格式如下:

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

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

这非常非常重要:

patch() is straightforward to use. The key is to do the patching in the right namespace. See the section where to patch.

你想要这个:

@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。你把它列为你尝试过的一件不起作用的事情,但我只是尝试过,对我来说很好。你试过的时候有什么问题?

相关问题 更多 >