199 lines
6.2 KiB
Go
199 lines
6.2 KiB
Go
package lambdarouter
|
|
|
|
import (
|
|
"encoding/json"
|
|
"errors"
|
|
"net/http"
|
|
"strings"
|
|
"testing"
|
|
|
|
"github.com/aws/aws-lambda-go/events"
|
|
. "github.com/smartystreets/goconvey/convey"
|
|
)
|
|
|
|
func TestRouterSpec(t *testing.T) {
|
|
|
|
Convey("Given an instantiated router with an error reporting middleware, headers, and prefix", t, func() {
|
|
request := events.APIGatewayProxyRequest{}
|
|
errorReporter := func(handler APIGHandler) APIGHandler {
|
|
return func(ctx *APIGContext) {
|
|
handler(ctx)
|
|
if ctx.Err != nil {
|
|
ctx.Body, _ = json.Marshal(map[string]string{"error": ctx.Err.Error()})
|
|
if strings.Contains(ctx.Err.Error(), "record not found") {
|
|
ctx.Status = 204
|
|
} else if ctx.Status != 204 && ctx.Status < 400 {
|
|
ctx.Status = 400
|
|
}
|
|
}
|
|
}
|
|
}
|
|
rtr := NewAPIGRouter(&APIGRouterConfig{
|
|
Request: &request,
|
|
Prefix: "/shipping",
|
|
Headers: map[string]string{
|
|
"Access-Control-Allow-Origin": "*",
|
|
"Access-Control-Allow-Credentials": "true",
|
|
},
|
|
Middleware: []APIGMiddleware{
|
|
errorReporter,
|
|
},
|
|
})
|
|
|
|
Convey("When the handler func does NOT return an error", func() {
|
|
hdlrfunc := func(ctx *APIGContext) {
|
|
ctx.Status = http.StatusOK
|
|
ctx.Body = []byte("hello")
|
|
ctx.Err = nil
|
|
}
|
|
|
|
Convey("And a Get handler expecting the pattern /listings/{id}/state/{event} is defined", func() {
|
|
rtr.Get("/listings/{id}/state/{event}", hdlrfunc)
|
|
rtr.Post("/orders", func(ctx *APIGContext) {})
|
|
rtr.Put("/orders", func(ctx *APIGContext) {})
|
|
rtr.Patch("/orders", func(ctx *APIGContext) {})
|
|
rtr.Delete("/orders/{id}", func(ctx *APIGContext) {})
|
|
|
|
Convey("And the request matches the pattern and the path params are filled", func() {
|
|
request.HTTPMethod = http.MethodGet
|
|
request.Path = "/shipping/listings/57/state/list"
|
|
request.PathParameters = map[string]string{
|
|
"id": "57",
|
|
"event": "list",
|
|
}
|
|
request.RequestContext.Authorizer = map[string]interface{}{
|
|
"claims": map[string]interface{}{
|
|
"cognito:username": "mitchell",
|
|
},
|
|
}
|
|
|
|
Convey("The router will return the expected status, body, and headers", func() {
|
|
response := rtr.Respond()
|
|
|
|
So(response.StatusCode, ShouldEqual, http.StatusOK)
|
|
So(response.Body, ShouldEqual, "hello")
|
|
So(response.Headers, ShouldResemble, map[string]string{
|
|
"Access-Control-Allow-Origin": "*",
|
|
"Access-Control-Allow-Credentials": "true",
|
|
})
|
|
})
|
|
})
|
|
|
|
Convey("And the request does NOT match the pattern", func() {
|
|
request.HTTPMethod = http.MethodGet
|
|
request.Path = "/orders/filter"
|
|
|
|
Convey("The router will return an error body and a status not found", func() {
|
|
response := rtr.Respond()
|
|
|
|
So(response.StatusCode, ShouldEqual, http.StatusNotFound)
|
|
So(response.Body, ShouldEqual, "{\"error\":\"no route matching path found\"}")
|
|
So(response.Headers, ShouldResemble, map[string]string{
|
|
"Access-Control-Allow-Origin": "*",
|
|
"Access-Control-Allow-Credentials": "true",
|
|
})
|
|
})
|
|
})
|
|
|
|
Convey("And a Get handler expecting the pattern /listings/{id}/state/{event} is defined AGAIN", func() {
|
|
So(func() {
|
|
rtr.Get("/listings/{id}/state/{event}", hdlrfunc)
|
|
}, ShouldPanicWith, "endpoint already existent")
|
|
})
|
|
|
|
Convey("And a Get handler expecting the pattern /orders/filter", func() {
|
|
rtr.Get("/orders/filter", hdlrfunc)
|
|
|
|
Convey("And the request matches the pattern and the path params are filled", func() {
|
|
request.HTTPMethod = http.MethodGet
|
|
request.Path = "/shipping/orders/filter"
|
|
|
|
Convey("The router will return the expected status and body", func() {
|
|
response := rtr.Respond()
|
|
|
|
So(response.StatusCode, ShouldEqual, http.StatusOK)
|
|
So(response.Body, ShouldEqual, "hello")
|
|
})
|
|
})
|
|
|
|
Convey("And the request does NOT match either of the patterns", func() {
|
|
request.HTTPMethod = http.MethodGet
|
|
request.Path = "/shipping/orders/filter/by_user"
|
|
|
|
Convey("The router will return an error body and a status not found", func() {
|
|
response := rtr.Respond()
|
|
|
|
So(response.StatusCode, ShouldEqual, http.StatusNotFound)
|
|
So(response.Body, ShouldEqual, "{\"error\":\"no route matching path found\"}")
|
|
})
|
|
})
|
|
})
|
|
})
|
|
|
|
})
|
|
|
|
Convey("When the handler func does return a record not found", func() {
|
|
hdlrfunc := func(ctx *APIGContext) {
|
|
ctx.Status = http.StatusNoContent
|
|
ctx.Body = []byte("hello")
|
|
ctx.Err = errors.New("record not found")
|
|
|
|
}
|
|
|
|
Convey("And a Get handler expecting the pattern /listings/{id}/state/{event} is defined", func() {
|
|
rtr.Get("/listings/{id}/state/{event}", hdlrfunc)
|
|
|
|
Convey("And the request matches the pattern and the path params are filled", func() {
|
|
request.HTTPMethod = http.MethodGet
|
|
request.Path = "/shipping/listings/57/state/list"
|
|
request.PathParameters = map[string]string{
|
|
"id": "57",
|
|
"event": "list",
|
|
}
|
|
|
|
Convey("The router will return the expected status and body", func() {
|
|
response := rtr.Respond()
|
|
|
|
So(response.StatusCode, ShouldEqual, http.StatusNoContent)
|
|
So(response.Body, ShouldEqual, "{\"error\":\"record not found\"}")
|
|
})
|
|
})
|
|
})
|
|
})
|
|
|
|
Convey("When the handler func does return a status < 400", func() {
|
|
middlefunc1 := func(handler APIGHandler) APIGHandler {
|
|
return func(ctx *APIGContext) {
|
|
ctx.Status = http.StatusOK
|
|
ctx.Err = errors.New("bad request")
|
|
handler(ctx)
|
|
}
|
|
}
|
|
hdlrfunc := func(ctx *APIGContext) {
|
|
ctx.Status = http.StatusOK
|
|
ctx.Body = []byte("hello")
|
|
}
|
|
|
|
Convey("And a Get handler expecting the pattern /listings/{id}/state/{event} is defined", func() {
|
|
rtr.Get("/listings/{id}/state/{event}", hdlrfunc, middlefunc1)
|
|
|
|
Convey("And the request matches the pattern and the path params are filled", func() {
|
|
request.HTTPMethod = http.MethodGet
|
|
request.Path = "/shipping/listings/57/state/list"
|
|
request.PathParameters = map[string]string{
|
|
"id": "57",
|
|
"event": "list",
|
|
}
|
|
|
|
Convey("The router will return the expected status and body", func() {
|
|
response := rtr.Respond()
|
|
|
|
So(response.StatusCode, ShouldEqual, http.StatusBadRequest)
|
|
So(response.Body, ShouldEqual, "{\"error\":\"bad request\"}")
|
|
})
|
|
})
|
|
})
|
|
})
|
|
})
|
|
}
|