仅使用装饰器配置返回值模拟

Question

有没有办法在patch装饰器中捕获以下逻辑，而不必将模拟传递到函数中：

@patch('boto3.client')
def test_playing_with_saml(self, boto3_client):
boto3_client.return_value.assume_role_with_saml = lambda *args, **kwargs: ('foo', 'bar')
self.assertEqual(playing_with_saml(), 'expected')

Answer 1

不，不是真的，不是没有指定boto3_client其余部分，这不会更清晰或更具可读性。

我不会在这里使用lambda，而是设置模拟的返回值：

boto3_client.return_value.assume_role_with_saml.return_value = ('foo', 'bar')

现在，您可以对boto3_client.return_value.assume_role_with_saml方法进行断言(例如断言它已被调用(。