router_test.go

package akumu_test

import (
	"net/http"
	"testing"

	"github.com/studiolambda/akumu"
)

func TestRouterHas(t *testing.T) {
	router := akumu.NewRouter()

	router.Get("/", func(request *http.Request) error {
		return akumu.Response(http.StatusOK)
	})

	router.Get("/foo/", func(request *http.Request) error {
		return akumu.Response(http.StatusOK)
	})

	router.Get("/bar", func(request *http.Request) error {
		return akumu.Response(http.StatusOK)
	})

	router.Get("/baz/{other...}", func(request *http.Request) error {
		return akumu.Response(http.StatusOK)
	})

	if route := "/"; !router.Has(http.MethodGet, route) {
		t.Fatalf("router should have the %s route", route)
	}

	if route := "/foo"; !router.Has(http.MethodGet, route) {
		t.Fatalf("router should have the %s route", route)
	}

	if route := "/foo/"; !router.Has(http.MethodGet, route) {
		t.Fatalf("router should have the %s route", route)
	}

	if route := "/bar"; !router.Has(http.MethodGet, route) {
		t.Fatalf("router should have the %s route", route)
	}

	if route := "/bar/"; !router.Has(http.MethodGet, route) {
		t.Fatalf("router should have the %s route", route)
	}

	if route := "/baz"; !router.Has(http.MethodGet, route) {
		t.Fatalf("router should have the %s route", route)
	}

	if route := "/baz/"; !router.Has(http.MethodGet, route) {
		t.Fatalf("router should have the %s route", route)
	}

	if route := "/baz/foo/bar/baz"; !router.Has(http.MethodGet, route) {
		t.Fatalf("router should have the %s route", route)
	}

	if route := "/baz/foo/bar/baz/"; !router.Has(http.MethodGet, route) {
		t.Fatalf("router should have the %s route", route)
	}

	if route := "/not-found"; router.Has(http.MethodGet, route) {
		t.Fatalf("router should not have the %s route", route)
	}
}

func TestRouterMatches(t *testing.T) {
	router := akumu.NewRouter()

	router.Get("/", func(request *http.Request) error {
		return akumu.Response(http.StatusOK)
	})

	request, err := http.NewRequest(http.MethodGet, "/", nil)

	if err != nil {
		t.Fatalf("unable to create http request: %v", err)
	}

	if !router.Matches(request) {
		t.Fatal("router does not have the route")
	}
}

func TestRouterHandler(t *testing.T) {
	router := akumu.NewRouter()

	router.Get("/", func(request *http.Request) error {
		return akumu.Response(http.StatusOK)
	})

	request, err := http.NewRequest(http.MethodGet, "/", nil)

	if err != nil {
		t.Fatalf("failed to create http request: %v", err)
	}

	response := router.Record(request)

	if expected := http.StatusOK; response.Code != expected {
		t.Fatalf("expected status code %d but got %d", expected, response.Code)
	}
}