unittest mock.patch仅一次

Question

我正在测试一个函数，每次执行两次“requests.post”。Si I存根此函数的调用。但是，我希望第一个存根返回一个假数据，第二个存根返回另一个假数据。怎么做？ 目前，我有：

@mock.patch('requests.post', side_effect=mocked_object)
def test_function_ok(self, mock_post):
    ...

我想要这样的东西：

@mock.patch_once('requests.post', side_effect=mocked_1)
@mock.patch_once('requests.post', side_effect=mocked_2)
def test_function_ok(self, mock_post):
    ...

Answer 1

您可以向请求发送两次要模拟的不同数据，它将模拟该方法并返回在return_value中指定的预期数据

@patch('the_function_path_you_want_to_mock')
def test_data(self, mock_obj):
    mock_obj.return_value.function_to_be_mocked.return_value = [data_to_mocked]
    response = self.client.post('/request', data=json.dumps(self.input_data), content_type='application/vnd.api+json', )

    mock_obj.return_value.function_to_be_mocked.return_value = [data_to_mocked]
    response = self.client.post('/request', data=json.dumps(self.input_data), content_type='application/vnd.api+json', )