在 Go 中测试 lambda 处理程序时如何模拟 AWS Lambda 上下文？

Question

我有一个用 Go 编写的 S3 触发的 AWS Lambda。我已经能够成功测试所有辅助代码，但是，我一直在尝试测试 lambda 处理程序。

这是我的处理程序的签名：

func HandleRequest(ctx context.Context, s3Event events.S3Event)

这是测试代码：

package main

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

  "github.com/aws/aws-lambda-go/events"
  "github.com/stretchr/testify/assert"
)

func TestHandleRequest(t *testing.T) {
  // 1. read JSON from file
  inputJSON, err := readJSONFromFile("./testdata/s3-event.json")
  if err != nil {
    t.Errorf("could not open test file. details: %v", err)
  }

  // 2. de-serialize into Go object
  var inputEvent events.S3Event
  if err := json.Unmarshal(inputJSON, &inputEvent); err != nil {
    t.Errorf("could not unmarshal event. details: %v", err)
  }

  // 3. How can I mock the context.Context?

  assert.NoError(t, HandleRequest(context.Context, inputEvent))
}

我不知道应该如何模拟 context.Context。我在网上也找不到任何例子。

有人知道吗？我的代码看起来适合测试 S3 触发的 Go Lambda 吗？

Answer 1

\xe2\x80\x98context.Context\xe2\x80\x99 被设计为一个不可变的值（即使它实际上是一个接口）。所以我\xe2\x80\x99不会关心嘲笑它。

\n\n

有两种方法可以创建空上下文（\xe2\x80\x98context.Background()\xe2\x80\x99 和 \xe2\x80\x98context.TODO()\xe2\x80\x99）。我将从这些开始。如果您想在上下文中设置某些内容，请查看上下文包上的文档。

\n