9

我正在用 python 测试一个自定义 API，它发出 http 请求，但我不想每次运行单元测试时都向真正的外部系统发出请求。我正在使用带有 side_effect 函数的 python 模拟库来动态伪造 API 响应。如何让 side_effect 方法表现得像类方法？

import requests

class MyApiClass():
    def make_request(self, params):
        return requests.get('http://someurl.com', params=params)

    def create_an_object(self, params):
        return self.make_request(params)

import unittest, mock

def side_effect_func(self, params):
    if params['name'] == 'Specific Name':
        return {'text': 'Specific Action'}
    else:
        return {'text': 'General Action'}

class MyApiTest(unittest.TestCase):
    def setUp(self):
        super(MyApiTest, self).setUp()
        mocked_method = mock.Mock(side_effect=side_effect_func)
        MyApiClass.make_request = mocked_method

    def test_create_object(self):
        api = MyApiClass()
        params = {'name': 'Specific Name'}
        r = api.create_an_object(params) # Complains that two arguments are needed!
        self.assertEqual(r['text'], 'Specific Action')

我收到这个错误

TypeError: side_effect_func() takes exactly 2 arguments (1 given)

但我想作为第一个参数side_effect_func传递。api感谢任何帮助！

4

1 回答 1

4

最简单的方法可能是让你的模拟方法接受一个参数，然后MyApiClass在模拟方法本身内静态引用。否则，您可以尝试模拟类对象本身（基本上是制作模拟元类），或者使用工厂partial来动态构建模拟类方法。但是，如果单个参数/静态引用方法对您有用，那对我来说似乎是最好的。

此外，在 Mock 文档中，使用 patch 模拟了一个未绑定的方法，看起来它可能更符合您的需要。

于 2013-10-30T14:16:21.257 回答