Golang如何在lambda集成测试或单元测试中模拟aws sns

Question

我想本地测试 lambda 函数调用，而不是使用 aws sns 服务。

func Handler(request events.APIGatewayProxyRequest) (events.APIGatewayProxyResponse, error){

    ...do some logic and db opeations.

    err := db.Create(something)

    if err != nil {

       msg := "something error."

       target := "sns:arn"

       // do sns notification

       sess := session.Must(session.NewSession())

       svc := sns.New(sess,&cfg)

       err = PublishNotification(svc, msg, target)

    }}

func PublishNotification(svc *sns.SNS, message string, target string) error {

      params := sns.PublishInput{

                Message:   &message,

                Subject:   &subject,

                TargetArn: &target,

}

      req, _ := svc.PublishRequest(&params)

      err := req.Send()

      if err != nil {

         return err

}

      return nil

}

func main(){

    lambda.Start(Handler)

}

现在一切正常。我想对这个 lambda 处理程序进行单元测试或集成测试。

func TestLabmdaWithSns(t *testing.T) {

        tests := []struct {

        request events.APIGatewayProxyRequest

}{

    request: events.APIGatewayProxyRequest{Body:"something"}

 }

 

 for _, test := range tests {

    response, err := Handler(test.request)

    // assert something

}

}

而且我不想在测试时发送 sns 通知。

我应该如何模拟这个 svc 或 PublishNotification 函数在本地测试时不向 sns 发布消息？

Answer 1

您应该在 PublishNotification 中使用接口 snsiface.SNSAPI 而不是 *sns.SNS。然后，您可以将 snsiface.SNSAPI 的模拟实现注入单元测试，如下所述：https ://docs.aws.amazon.com/sdk-for-go/api/service/sns/snsiface/