无法将 JSON 上传到 S3 的单元测试方法

从现在开始，我们要努力学习啦！今天我给大家带来《无法将 JSON 上传到 S3 的单元测试方法》，感兴趣的朋友请继续看下去吧！下文中的内容我们主要会涉及到等等知识点，如果在阅读本文过程中有遇到不清楚的地方，欢迎留言呀！我们一起讨论，一起学习！

我有一个如下所示的函数：

func savejsontos3(data map[string]interface{}) error {
    name := fmt.sprintf("audit_event_-%v.json", time.now().format(time.rfc822))
    // create an s3 session
    sess, err := session.newsession(&aws.config{
        region: aws.string(utils.aws_region), // aws region
    })
    if err != nil {
        log.printf(utils.log_failed_s3_session, err)
    }

    // create an s3 client
    svc := s3.new(sess)

    // convert the json data to a byte array
    jsondata, err := json.marshal(data)
    if err != nil {
        log.printf(utils.log_failed_marshal_json, err)
        return fmt.errorf("failed to marshal json data: %v", err)
    }

    // set the s3 bucket and object key
    bucket := utils.aws_s3_bucket_name
    key := name

    // upload the json data to s3
    _, err = svc.putobject(&s3.putobjectinput{
        body:   aws.readseekcloser(bytes.newreader(jsondata)),
        bucket: aws.string(bucket),
        key:    aws.string(key),
    })
    if err != nil {
        return fmt.errorf("failed to upload json to s3: %v", err)
    }

    return nil
}

我当前的单元测试如下：

func TestSaveJSONToS3_MarshalError(t *testing.T) {
    // Input
    data := map[string]interface{}{
        "key": make(chan int), // Invalid data that cannot be marshaled
    }

    // Execute the function
    err := SaveJSONToS3(data)

    // Assert the result
    expectedError := fmt.Errorf("failed to marshal JSON data: json: unsupported type: chan int")
    if err == nil || err.Error() != expectedError.Error() {
        t.Errorf("expected error: %v, got: %v", expectedError, err)
    }
}

我的覆盖范围缺少两行，第一行表示无效会话，另一行表示上传到 s3 失败行，我想将其包含在我的测试中，有人可以建议吗我该怎么做。当我在函数内创建会话时，我很难在测试类中模拟会话并将其传递给原始函数。提前致谢。

正确答案

您需要能够确定性地模拟 S3 的行为，这意味着您需要从问题中删除 S3，并将其替换为提供与 S3 相同的合约但允许您决定行为的东西，以便模拟您想要的测试场景。

即模拟/假货/虚拟 S3

有很多方法可以实现这一目标，具体取决于您想要/需要达到的复杂程度。

1. 提供初始化为所需 aws/s3 函数的函数变量，这些函数可以在测试中替换为模拟函数

2. 提供一个封装各个 aws/s3 函数合约的接口。然后将 save 函数实现为结构体上的方法，您可以根据需要注入函数契约的具体实现或测试的模拟实现

3. 类似地，将 save 函数实现为接受 s3 客户端抽象（接口）的方法结构，抽象出所有 s3 问题（需要单独测试）；然后，您可以再次注入此 s3 客户端抽象的模拟实现，以测试保存函数本身

这些方法的实现留作练习。

好了，本文到此结束，带大家了解了《无法将 JSON 上传到 S3 的单元测试方法》，希望本文对你有所帮助！关注golang学习网公众号，给大家分享更多Golang知识！