group_test.go

package goapi_test

import (
	"net/http"
	"testing"
	"time"

	"github.com/NaturalSelectionLabs/goapi"
	"github.com/NaturalSelectionLabs/goapi/lib/middlewares"
	"github.com/ysmood/got"
)

func TestPrefix(t *testing.T) {
	g := got.T(t)

	g.Eq(goapi.New().Group("/test").Prefix(), "/test")
}

func TestMultipleGroups(t *testing.T) {
	g := got.T(t)

	tr := setupRouter(g, func(r *goapi.Group) {
		ga := r.Router().Group("/a")
		ga.Use(middlewares.Func(func(next http.Handler) http.Handler {
			return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
				next.ServeHTTP(w, r)
			})
		}))
		ga.GET("/users", func() res { return resOK{Data: "a"} })

		gb := r.Group("/b")

		gb.GET("/users", func() res { return resOK{Data: "b"} })
		gb.POST("/users", func() res { return resOK{Data: "post"} })
		gb.PUT("/users", func() res { return resOK{Data: "b"} })
		gb.PATCH("/users", func() res { return resOK{Data: "b"} })
		gb.DELETE("/users", func() res { return resOK{Data: "b"} })
		gb.HEAD("/users", func() res { return resOK{Data: "b"} })
		gb.OPTIONS("/users", func() res { return resOK{Data: "b"} })

		g.Eq(g.Panic(func() {
			gb.Group("user")
		}), "expect prefix to start with '/', but got: user")

		g.Eq(g.Panic(func() {
			gb.Group("/user/")
		}), "expect prefix to not end with '/', but got: /user/")

		g.Eq(g.Panic(func() {
			gb.Group("/{user}")
		}), "expect prefix not contains braces, but got: /{user}")
	})

	g.Eq(g.Req("", tr.URL("/a/users")).JSON(), map[string]any{
		"data": "a",
	})
	g.Eq(g.Req("", tr.URL("/b/users")).JSON(), map[string]any{
		"data": "b",
	})
	g.Eq(g.Req(http.MethodPost, tr.URL("/b/users")).JSON(), map[string]any{
		"data": "post",
	})
}

func TestStart(t *testing.T) {
	g := got.T(t)
	r := goapi.New()
	r.GET("/", func() resOK { return resOK{Data: "ok"} })

	go func() { _ = r.Start(":41375") }()

	time.Sleep(300 * time.Millisecond)

	g.Eq(g.Req("", "http://localhost:41375").JSON(), map[string]interface{}{
		"data": "ok",
	})

	g.E(r.Shutdown(g.Context()))
}

func TestGroupAsMiddleware(t *testing.T) {
	g := got.T(t)
	tr := g.Serve()

	r := goapi.New()

	h := r.Handler(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
		_, _ = w.Write([]byte("ok"))
	}))

	tr.Mux.Handle("/", h)

	g.Eq(g.Req("", tr.URL("/")).String(), "ok")
}

func TestGroupTrailingSlash(t *testing.T) {
	g := got.T(t)

	tr := setupRouter(g, func(r *goapi.Group) {
		r.GET("/", func() resOK { return resOK{} })

		ga := r.Group("/test")
		ga.GET("/", func() resOK { return resOK{} })

		ga.GET("/x", func() resOK { return resOK{Data: "x"} })
		ga.GET("/x/", func() resOK { return resOK{Data: "x/"} })
	})

	g.Eq(g.Req("", tr.URL("")).StatusCode, 200)
	g.Eq(g.Req("", tr.URL("/")).StatusCode, 200)

	g.Eq(g.Req("", tr.URL("/test")).StatusCode, 200)
	g.Eq(g.Req("", tr.URL("/test/")).StatusCode, 200)

	g.Eq(g.Req("", tr.URL("/test/x")).JSON(), map[string]interface{}{"data": "x"})
	g.Eq(g.Req("", tr.URL("/test/x/")).JSON(), map[string]interface{}{"data": "x/"})
}

func double(num int) int {
	return num * 2
}

func TestAdd(t *testing.T) {
	g := got.T(t)

	tr := setupRouter(g, func(r *goapi.Group) {
		goapi.Add(r, double)
	})

	g.Eq(g.Req(http.MethodPost, tr.URL("/double"), "3").JSON(), 6)
}