模拟跨库的调用链

Question

我正在使用hammock库将请求抽象为内部API。在我的代码中，self.api是一个吊床对象，我进行了以下调用：

帖子= self.api.posts().GET(params=params).json().get('objects')

其中.GET返回响应对象，.json将其转换为字典，.get返回该字典中“objects”键的值。

我想要的是整个调用只返回一个我已经设置好的fixture。然而，当我尝试像self.worker.api.posts.get = Mock(return_value=fixture)这样的东西时，它不起作用。它调用实际的API。

这是可能的吗，或者我应该放弃尝试模拟这样一个特定的调用，而只是满足于模拟包含它的方法？

Answer 1

您需要在模拟上设置所有return_values。当你调用self.api.posts()时，会返回一个模拟，你想要得到这个模拟并设置它的返回值，这将是一个有一个GET()方法的对象，你也想模拟它的返回值，直到你想模拟的最后一个调用。

示例如下：

self.api.posts.return_value.GET.return_value.json.return_value.get = Mock(return_value=fixture)

我的工作示例：

json_obj = mock.MagicMock(return_value=self.hardcoced_labels) con.return_value.repos.return_value.labels.GET.return_value.json = json_obj