Запрос Mock AWS API Gateway и DynamoDB для функционального блока Golang Lambda Test

Question

Настройка

• Windows 10
• go v1.10.3
• aws cli v1.16.67

Что я пытаюсь сделать

Проверка лямбда-функции AWS, написанной с использованием golang.Функция принимает запрос от API Gateway, а затем выполняет некоторые действия с DynamoDB.Большая часть нижеследующего взята из этой статьи (я новичок в Go)

package main

import (
    "encoding/json"
    "log"
    "net/http"
    "os"
    "regexp"

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

var uuidRegexp = regexp.MustCompile(`\b[0-9a-f]{8}\b-[0-9a-f]{4}-[0-9a-f]{4}-[0-9a-f]{4}-\b[0-9a-f]{12}\b`)
var errorLogger = log.New(os.Stderr, "ERROR ", log.Llongfile)

type job struct {
    ID                string `json:"id"`
    ClientID          string `json:"clientId"`
    Title             string `json:"title"`
    Count             int    `json:"count"`
}

// CreateJobCommand manages interactions with DynamoDB
func CreateJobCommand(req events.APIGatewayProxyRequest) (events.APIGatewayProxyResponse, error) {

    if req.Headers["Content-Type"] != "application/json" {
        return clientError(http.StatusNotAcceptable) //406
    }

    newJob := new(job)
    err := json.Unmarshal([]byte(req.Body), newJob)

    // Ensure request has deserialized correctly
    if err != nil {
        return clientError(http.StatusUnprocessableEntity) //422
    }

    // Validate ID and ClientID attributes match RegEx pattern
    if !uuidRegexp.MatchString(newJob.ID) || !uuidRegexp.MatchString(newJob.ClientID) {
        return clientError(http.StatusBadRequest)
    }

    // Mandatory field check
    if newJob.Title == "" {
        return clientError(http.StatusBadRequest)
    }

    // Put item in database
    err = putItem(newJob) // putItem is defined in another file
    if err != nil {
        return serverError(err)
    }

    return events.APIGatewayProxyResponse{
        StatusCode: 201,
    }, nil
}

// Add a helper for handling errors. This logs any error to os.Stderr
// and returns a 500 Internal Server Error response that the AWS API
// Gateway understands.
func serverError(err error) (events.APIGatewayProxyResponse, error) {
    errorLogger.Println(err.Error())

    return events.APIGatewayProxyResponse{
        StatusCode: http.StatusInternalServerError,
        Body:       http.StatusText(http.StatusInternalServerError),
    }, nil
}

// Similarly add a helper for send responses relating to client errors.
func clientError(status int) (events.APIGatewayProxyResponse, error) {
    return events.APIGatewayProxyResponse{
        StatusCode: status,
        Body:       http.StatusText(status),
    }, nil
}

func putItem(job *job) error {

    // create an aws session
    sess := session.Must(session.NewSession(&aws.Config{
        Region:   aws.String("us-east-1"),
        Endpoint: aws.String("http://localhost:8000"),
    }))

    // create a dynamodb instance
    db := dynamodb.New(sess)

    // marshal the job struct into an aws attribute value object
    jobAVMap, err := dynamodbattribute.MarshalMap(job)
    if err != nil {
        return err
    }

    input := &dynamodb.PutItemInput{
        TableName: aws.String("TEST_TABLE"),
        Item:      jobAVMap,
    }

    _, err = db.PutItem(input)
    return err
}

func main() {
    lambda.Start(CreateJobCommand)
}

Проблема

Я хочунаписать набор модульных тестов для проверки этой функции.На мой взгляд, первое, что мне нужно сделать, это смоделировать запрос API Gateway и таблицу DynamoDB, но я не знаю, как это сделать.

Вопросы

1. Есть ли какая-то насмешливая структура, которую я должен использовать?
2. Если кто-нибудь знает какую-либо документацию, которая может помочь по этой темене могли бы вы указать это, пожалуйста?(Мои навыки Google еще не раскрыты)

Спасибо

Answer 1

Несмотря на то, что mocking является жизнеспособным вариантом, вы можете также рассмотреть возможность тестирования e2e с выделенной учетной записью aws, найти несколько примеров, в том числе DynamoDB и API-шлюз

lambda_e2e

Answer 2

Пожалуйста, проверьте, не поможет ли использование Dynamo-DB в докере реализовать ваш тест.

Проверка: подключение AWS SAM Local с DynamodB в Docker

Вы также можете довольно легко передать событие обработчику в своем тесте.