登录
首页 >  Golang >  Go问答

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

来源:stackoverflow

时间:2024-02-07 20:18:22 331浏览 收藏

从现在开始,我们要努力学习啦!今天我给大家带来《无法将 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知识!

声明:本文转载于:stackoverflow 如有侵犯,请联系study_golang@163.com删除
相关阅读
更多>
最新阅读
更多>
课程推荐
更多>