107 lines
2.2 KiB
Go
107 lines
2.2 KiB
Go
package lambdarouter
|
|
|
|
import (
|
|
"context"
|
|
"encoding/json"
|
|
"fmt"
|
|
"net/http"
|
|
"testing"
|
|
|
|
"github.com/aws/aws-lambda-go/events"
|
|
"github.com/aws/aws-lambda-go/lambda"
|
|
"github.com/stretchr/testify/assert"
|
|
)
|
|
|
|
func TestRouter(t *testing.T) {
|
|
a := assert.New(t)
|
|
|
|
desc(t, 0, "Intialize Router and")
|
|
r := New("prefix")
|
|
handler := lambda.NewHandler(handler)
|
|
ctx := context.Background()
|
|
|
|
desc(t, 2, "Get|Post|Put|Patch|Delete method should")
|
|
{
|
|
desc(t, 4, "insert a new route succesfully")
|
|
a.NotPanics(func() {
|
|
r.Get("thing/{id}", handler)
|
|
r.Delete("/thing/{id}/", handler)
|
|
r.Put("thing", handler)
|
|
})
|
|
|
|
desc(t, 4, "panic when inserting the same route")
|
|
a.Panics(func() {
|
|
r.Put("thing", handler)
|
|
})
|
|
|
|
desc(t, 4, "panic when router is uninitalized")
|
|
var r2 Router
|
|
a.Panics(func() {
|
|
r2.Patch("panic", handler)
|
|
})
|
|
|
|
desc(t, 4, "panic when when given an empty path")
|
|
a.Panics(func() {
|
|
r.Post("", handler)
|
|
})
|
|
}
|
|
|
|
desc(t, 2, "PrefixGroup method should")
|
|
{
|
|
desc(t, 4, "insert routes with the specified prefix succesfully")
|
|
r.Group("/ding", func(r *Router) {
|
|
r.Post("dong/{door}", handler)
|
|
})
|
|
}
|
|
|
|
desc(t, 2, "Invoke method should")
|
|
{
|
|
e := events.APIGatewayProxyRequest{
|
|
Path: "/prefix/ding/dong/mitchell",
|
|
HTTPMethod: http.MethodPost,
|
|
PathParameters: map[string]string{"door": "mitchell"},
|
|
}
|
|
|
|
desc(t, 4, "should succesfully route and invoke a defined route")
|
|
ejson, _ := json.Marshal(e)
|
|
|
|
res, err := r.Invoke(ctx, ejson)
|
|
|
|
a.NoError(err)
|
|
a.Exactly("null", string(res))
|
|
|
|
desc(t, 4, "return the expected response when a route is not found")
|
|
e.Path = "thing"
|
|
e.PathParameters = nil
|
|
ejson2, _ := json.Marshal(e)
|
|
eres := events.APIGatewayProxyResponse{
|
|
StatusCode: http.StatusNotFound,
|
|
Body: "not found",
|
|
}
|
|
eresjson, _ := json.Marshal(eres)
|
|
|
|
res, err = r.Invoke(ctx, ejson2)
|
|
|
|
a.NoError(err)
|
|
a.ElementsMatch(eresjson, res)
|
|
|
|
desc(t, 4, "return an error when the there is an issue with the incoming event")
|
|
_, err = r.Invoke(ctx, nil)
|
|
|
|
a.Error(err)
|
|
}
|
|
|
|
}
|
|
|
|
func handler() error {
|
|
return nil
|
|
}
|
|
|
|
func desc(t *testing.T, depth int, str string, args ...interface{}) {
|
|
for i := 0; i < depth; i++ {
|
|
str = " " + str
|
|
}
|
|
|
|
t.Log(fmt.Sprintf(str, args...))
|
|
}
|