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

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这里使用 a ，而是设置模拟的返回值：

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

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