如何为aws-sdk-go-v2的DynamoDB实现编写单元测试？Mock PutItemRequest受阻

嘿，我明白你卡在这儿了！问题的核心在于PutItemRequest返回的结构体自带一个Send()方法，你之前的Mock只返回了空结构体，没有覆盖这个Send方法的行为。咱们一步步来搞定它：

问题分析 #

当你的业务代码调用req.Send()时，实际执行的是dynamodb.PutItemRequest结构体里的Send方法。要Mock这个流程，我们需要自定义一个带预设行为的PutItemRequest，让它的Send方法返回我们想要的结果或错误。

修改后的测试代码实现 #

1. 定义Mock的PutItemRequest结构体

首先创建一个自定义的Mock请求结构体，实现Send方法来控制返回结果：

import (
    "context"
    "errors"
    "testing"

    "github.com/aws/aws-sdk-go-v2/aws/middleware"
    "github.com/aws/aws-sdk-go-v2/service/dynamodb"
    "github.com/aws/aws-sdk-go-v2/service/dynamodb/dynamodbiface"
)

// 自定义Mock的PutItemRequest，用来预设Send方法的返回值
type mockPutItemRequest struct {
    result *dynamodb.PutItemOutput
    err    error
}

// 实现Send方法，返回我们预设的结果或错误
func (m *mockPutItemRequest) Send(ctx context.Context, opts ...func(*middleware.Stack)) (*dynamodb.PutItemOutput, error) {
    return m.result, m.err
}

2. 更新Mock客户端

让你的mockDynamoDBClient支持预设结果和错误，这样可以灵活测试不同场景：

// 更新mock客户端，加入预设的结果和错误字段
type mockDynamoDBClient struct {
    dynamodbiface.DynamoDBAPI
    putItemResult *dynamodb.PutItemOutput
    putItemErr    error
}

// 修改PutItemRequest方法，返回我们自定义的Mock请求
func (m *mockDynamoDBClient) PutItemRequest(input *dynamodb.PutItemInput) dynamodb.PutItemRequest {
    mockReq := &mockPutItemRequest{
        result: m.putItemResult,
        err:    m.putItemErr,
    }

    // 将Mock的Send方法绑定到PutItemRequest的Send字段
    return dynamodb.PutItemRequest{
        Send: func(ctx context.Context, opts ...func(*middleware.Stack)) (*dynamodb.PutItemOutput, error) {
            return mockReq.Send(ctx, opts...)
        },
    }
}

3. 编写完整测试用例

现在可以测试成功和失败两种场景了：

// 测试PutItem成功的场景
func TestStoreInDynamoDB_Success(t *testing.T) {
    // 预设成功返回的结果（可根据业务需求填充字段）
    mockResult := &dynamodb.PutItemOutput{}

    mockClient := &mockDynamoDBClient{
        putItemResult: mockResult,
        putItemErr:    nil,
    }

    testData := mydata{} // 填充你的测试数据
    err := MyPutItem(testData, mockClient)

    if err != nil {
        t.Fatalf("预期无错误，实际得到: %v", err)
    }
}

// 测试PutItem失败的场景
func TestStoreInDynamoDB_Failure(t *testing.T) {
    // 预设错误信息
    mockErr := errors.New("模拟DynamoDB写入失败")

    mockClient := &mockDynamoDBClient{
        putItemResult: nil,
        putItemErr:    mockErr,
    }

    testData := mydata{} // 填充你的测试数据
    err := MyPutItem(testData, mockClient)

    if err == nil || err.Error() != mockErr.Error() {
        t.Fatalf("预期错误: %v，实际得到: %v", mockErr, err)
    }
}

关键说明 #

这里的核心是把PutItemRequest的Send函数替换成我们自己的Mock实现，这样业务代码调用req.Send()时，就会执行我们预设的逻辑，而不是真正调用AWS的API。你可以根据需要扩展这个Mock，比如验证输入参数是否符合预期。

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