如何導入自己的模塊進行模擬？ （導入錯誤：沒有名爲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()方法在上面的方法調用。

但模擬模塊抱怨：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方法？

（這似乎很奇怪，因爲我可以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。你列出這是你嘗試過的一種方法不起作用的，但我只是試過了，而且對我來說效果很好。當你嘗試時，你有什麼問題？