如何对EventBridge触发的Go语言Lambda函数进行单元测试？

你当前的问题是直接把JSON字符串传给handleRequest，但函数参数要求的是events.CloudWatchEvent类型，所以需要把JSON反序列化成这个结构体实例，具体解决步骤如下：

1. 补充导入必要的包 #

在测试文件里加上encoding/json、context包以及AWS的events包：

import (
    "context"
    "encoding/json"
    "testing"

    "github.com/aws/aws-lambda-go/events"
)

2. 反序列化JSON字符串到CloudWatchEvent实例 #

修改测试函数，将JSON字符串转换为events.CloudWatchEvent类型，同时处理反序列化可能出现的错误：

func TestHandleRequest(t *testing.T) {
    dummyRequest := `{
        "version": "0",
        "id": "fe8d3c65-xmpl-c5c3-2c87-81584709a377",
        "detail-type": "RDS DB Instance Event",
        "source": "aws.rds",
        "account": "123456789012",
        "time": "2020-04-28T07:20:20Z",
        "region": "us-east-2",
        "resources": [
            "arn:aws:rds:us-east-2:123456789012:db:rdz6xmpliljlb1"
        ],
        "detail": {
            "EventCategories": [
                "backup"
            ],
            "SourceType": "DB_INSTANCE",
            "SourceArn": "arn:aws:rds:us-east-2:123456789012:db:rdz6xmpliljlb1",
            "Date": "2020-04-28T07:20:20.112Z",
            "Message": "Finished DB Instance backup",
            "SourceIdentifier": "rdz6xmpliljlb1"
        }
    }`

    // 声明CloudWatchEvent变量
    var event events.CloudWatchEvent
    // 把JSON字符串反序列化到变量中
    err := json.Unmarshal([]byte(dummyRequest), &event)
    if err != nil {
        t.Fatalf("反序列化模拟事件失败: %v", err)
    }

    // 使用context.Background()代替nil，更贴合Lambda运行时的实际场景
    ctx := context.Background()
    result, err := handleRequest(ctx, event)
    if err != nil {
        t.Fatalf("handleRequest执行出错: %v", err)
    }

    // 这里可以添加断言逻辑，验证返回结果是否符合预期
    expected := "你的预期返回值"
    if result != expected {
        t.Errorf("返回结果不符，期望: %s，实际: %s", expected, result)
    }
}

关键说明 #

• json.Unmarshal是Go标准库的反序列化方法，events.CloudWatchEvent的字段结构和EventBridge事件的JSON格式完全匹配，因此可以直接完成转换。
• 不要传nil作为context参数，context.Background()是更标准的用法，能避免后续代码操作ctx时出现panic。
• 测试中必须处理反序列化和函数执行的错误，t.Fatalf会在出错时立即终止测试，便于快速定位问题。

内容的提问来源于stack exchange，提问作者ServerMonkey