如何在我的类中模拟使用 requests.get 的方法？

Question

我正在尝试为我的班级创建一些单元测试。我想模拟这些，这样我就不会在运行其中一些测试时耗尽我的 API 配额。我有多个测试用例将调用该fetch方法，并且根据传递的 URL，我将得到不同的结果。

我的示例类如下所示：

import requests
class ExampleAPI(object):
    def fetch(self, url, params=None, key=None, token=None, **kwargs):
        return requests.get(url).json() # Returns a JSON string

我正在看的教程表明我可以做这样的事情：

import unittest
from mock import patch

def fake_fetch_test_one(url):
    ...

class TestExampleAPI(unittest.TestCase):
    @patch('mymodule.ExampleAPI.fetch', fake_fetch_test_one)
    def test_fetch(self):
        e = ExampleAPI()
        self.assertEqual(e.fetch('http://my.api.url.example.com'), """{'result': 'True'}""")

但是，当我这样做时，我收到一条错误消息：

TypeError: fake_fetch_test_one() takes exactly 1 argument (3 given)

requests.get模拟类中方法中的调用的正确方法是什么？我需要能够更改每个测试的模拟响应，因为不同的 URL 可以提供不同的响应类型。

Answer 1

你的假提取需要接受与原始相同的参数：

def fake_fetch(self, url, params=None, key=None, token=None, **kwargs):

请注意，最好只模拟外部接口，这意味着让fetch调用requests.get（或者至少，它认为是requests.get）：

@patch('mymodule.requests.get')
def test_fetch(self, fake_get):
    # It would probably be better to just construct
    # a valid fake response object whose `json` method
    # would return the right thing, but this is a easier
    # for demonstration purposes. I'm assuming nothing else
    # is done with the response.
    expected = {"result": "True"}
    fake_get.return_value.json.return_value = expected
    e = ExampleAPI()
    self.assertEqual(e.fetch('http://my.api.url.example.com'), expected)