时间戳

首页 > 解决方案 > 在 Go 中测试 lambda 处理程序时如何模拟 AWS Lambda 上下文？

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

问题描述

我有一个用 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？

标签： amazon-web-servicesgoaws-lambda

解决方案

'context.Context' 被设计为一个不可变的值（即使它实际上是一个接口）。所以我不会担心嘲笑它。

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

推荐阅读