如何在单元测试中模拟ParameterNotFound boto3异常？

Question

我想测试一些错误处理逻辑，所以我想在我的单元测试中模拟特定的异常类型。我正在嘲笑对boto3的调用，但是我想让这个mock引发一个ParameterNotLocd异常。我正在测试的代码遵循以下模式：

boto3_client = boto3.client('ssm')
try:
    temp_var = boto3_client.get_parameter(Name="Something not found")['Parameter']['Value']
except boto3_client.exceptions.ParameterNotFound:
    ... [logic I want to test]

我已经创建了一个unittest模拟，但我不知道如何使它引发异常，因为这个参数NotFound exception。我尝试了以下方法，但不起作用，因为在计算except子句时，它得到了“exceptions必须派生自基类”：

@patch('patching_config.boto3.client')
def test_sample(self, mock_boto3_client):
        mock_boto3_client.return_value = mock_boto3_client

        def get_parameter_side_effect(**kwargs):
            raise boto3.client.exceptions.ParameterNotFound()

        mock_boto3_client.get_parameter.side_effect = get_parameter_side_effect

如何在单元测试中模拟ParameterNotFound boto3异常？

Answer 1

我认为问题在于我对boto3如何引发例外的误解。我在这里找到了一个解释：https://github.com/boto/boto3/issues/1262在“客户端错误的结构”下

ClientError的结构

在ClientError（但不是BotoCreError）中，将有一个operation_name属性（应该是str）和一个response属性（应该是dict）。response属性应具有以下形式（来自格式错误的ec2.DescribeImages调用的示例）：

还有这里：https://codeday.me/en/qa/20190306/12210.html

{
    "Error": {
        "Code": "InvalidParameterValue",
        "Message": "The filter 'asdfasdf' is invalid"
    },
    "ResponseMetadata": {
        "RequestId": "aaaabbbb-cccc-dddd-eeee-ffff00001111",
        "HTTPStatusCode": 400,
        "HTTPHeaders": {
            "transfer-encoding": "chunked",
            "date": "Fri, 01 Jan 2100 00:00:00 GMT",
            "connection": "close",
            "server": "AmazonEC2"
        },
        "RetryAttempts": 0
    }
}

听起来异常是作为ClientError抛出的，它有一个ParameterNotFound错误代码，所以我需要将事情更改为

from botocore.exceptions import ClientError

然后呢

except ClientError as e:

在嘲笑中，我需要提出一个ClientError来代替它，它将ParameterNot找到作为代码：

raise botocore.exceptions.ClientError({"Error": {"Code": "ParameterNotFound",
                                                 "Message": "Parameter was not found"}}, 
                                      'get_parameter')