如何在 flask-restful 资源方法中模拟函数调用

Question

我使用 Flask-restful 开发了一个 API。我有一个 API 资源名为 'Server'。此资源有一个方法 get 来处理对“/server”url 的请求。在这个方法中，我有另一个 class 'Connector' 的调用方法，它从另一个服务获取数据：

 class Server(Resource):
    def get(self):
     ...
        status, body = connector.get_servers(page, size) # call method of another class
     ...
    return body, status

我要测试开发API。我写了一些测试：

from application import create_app
from unittest import TestCase

class TestServerResource(TestCase):

def setUp(self):
    self.app = create_app()
    self.client = self.app.test_client

def test_bad_url(self):
    res = self.client().get('/server')
    self.assertEqual(res.status_code, 400)

# Test of get method Server resources described above
def test_pagination(self):
    res = self.client().get('/server?page=1&size=1') # request to my API
    self.assertEqual(res.status_code, 200)

在方法 'test_pagination' 中，我正在测试我的资源的方法 'get'，但是另一个 class 方法的调用是在这个方法中。因此我有一个问题：如何在测试中模拟调用 'connector.get_servers()'？

谢谢。

Answer 1

我找到了解决办法。要在其他方法中模拟方法调用，我们可以使用来自 unittest.mock

的 'patch' 装饰器

例如下面描述的例子，它看起来像下面这样：

from unittest.mock import patch

# Test of get method Server resources described above
@patch('path_to_method_we_want_to_mock.method')
def test_pagination(self, mock):
   mock.return_value = <new value> # set value which mocked method return
   res = self.client().get('/server?page=1&size=1') # request to my API
   self.assertEqual(res.status_code, 200)

现在在方法 get() 中调用 get_servers 方法将 return mock.return_value.

也可以使用一些补丁装饰器：

@patch('application.servers_connector.ServersConnector.get_server_by_id')
@patch('application.rent_connector.RentConnector.get_rents_for_user')
def test_rent_for_user(self, rent_mock, server_mock):
     ...

如何在 flask-restful 资源方法中模拟函数调用

How to mock function call in flask-restul resource method

testing

mocking

flask

flask-restful