micro/util/http/trie_test.go

package http

import (
	"net/http"
	"testing"
)

func TestTrieWildcardPathPrefix(t *testing.T) {
	var err error
	type handler struct {
		name string
	}
	tr := NewTrie()
	if err = tr.Insert([]string{http.MethodPost}, "/v1/update", &handler{name: "post_update"}); err != nil {
		t.Fatal(err)
	}
	if err = tr.Insert([]string{http.MethodPost}, "/v1/*", &handler{name: "post_create"}); err != nil {
		t.Fatal(err)
	}
	h, _, ok := tr.Search(http.MethodPost, "/v1/test/one")
	if !ok {
		t.Fatalf("unexpected error handler not found")
	}
	if h.(*handler).name != "post_create" {
		t.Fatalf("invalid handler %v", h)
	}
	h, _, ok = tr.Search(http.MethodPost, "/v1/update")
	if !ok {
		t.Fatalf("unexpected error")
	}
	if h.(*handler).name != "post_update" {
		t.Fatalf("invalid handler %v", h)
	}
	h, _, ok = tr.Search(http.MethodPost, "/v1/update/some/{x}")
	if !ok {
		t.Fatalf("unexpected error")
	}
	if h.(*handler).name != "post_create" {
		t.Fatalf("invalid handler %v", h)
	}
}

func TestTriePathPrefix(t *testing.T) {
	type handler struct {
		name string
	}
	tr := NewTrie()
	_ = tr.Insert([]string{http.MethodPost}, "/v1/create/{id}", &handler{name: "post_create"})
	_ = tr.Insert([]string{http.MethodPost}, "/v1/update/{id}", &handler{name: "post_update"})
	_ = tr.Insert([]string{http.MethodPost}, "/", &handler{name: "post_wildcard"})
	h, _, ok := tr.Search(http.MethodPost, "/")
	if !ok {
		t.Fatalf("unexpected error")
	}
	if h.(*handler).name != "post_wildcard" {
		t.Fatalf("invalid handler %v", h)
	}
}

func TestTrieFixedPattern(t *testing.T) {
	type handler struct {
		name string
	}
	tr := NewTrie()
	_ = tr.Insert([]string{http.MethodPut}, "/v1/create/{id}", &handler{name: "pattern"})
	_ = tr.Insert([]string{http.MethodPut}, "/v1/create/12", &handler{name: "fixed"})
	h, _, ok := tr.Search(http.MethodPut, "/v1/create/12")
	if !ok {
		t.Fatalf("unexpected error")
	}
	if h.(*handler).name != "fixed" {
		t.Fatalf("invalid handler %v", h)
	}
}

func TestTrieNoMatchMethod(t *testing.T) {
	tr := NewTrie()
	_ = tr.Insert([]string{http.MethodPut}, "/v1/create/{id}", nil)
	_, _, ok := tr.Search(http.MethodPost, "/v1/create")
	if ok {
		t.Fatalf("must be not found error")
	}
}

func TestTrieMatchRegexp(t *testing.T) {
	type handler struct{}
	tr := NewTrie()
	_ = tr.Insert([]string{http.MethodPut}, "/v1/create/{category}/{id:[0-9]+}", &handler{})
	_, params, ok := tr.Search(http.MethodPut, "/v1/create/test_cat/12345")
	switch {
	case !ok:
		t.Fatalf("route not found")
	case len(params) != 2:
		t.Fatalf("param matching error %v", params)
	case params["category"] != "test_cat":
		t.Fatalf("param matching error %v", params)
	}
}

func TestTrieMatchRegexpFail(t *testing.T) {
	type handler struct{}
	tr := NewTrie()
	_ = tr.Insert([]string{http.MethodPut}, "/v1/create/{id:[a-z]+}", &handler{})
	_, _, ok := tr.Search(http.MethodPut, "/v1/create/12345")
	if ok {
		t.Fatalf("route must not be not found")
	}
}

func TestTrieMatchLongest(t *testing.T) {
	type handler struct {
		name string
	}
	tr := NewTrie()
	_ = tr.Insert([]string{http.MethodPut}, "/v1/create", &handler{name: "first"})
	_ = tr.Insert([]string{http.MethodPut}, "/v1/create/{id:[0-9]+}", &handler{name: "second"})
	if h, _, ok := tr.Search(http.MethodPut, "/v1/create/12345"); !ok {
		t.Fatalf("route must be found")
	} else if h.(*handler).name != "second" {
		t.Fatalf("invalid handler found: %s != %s", h.(*handler).name, "second")
	}
	if h, _, ok := tr.Search(http.MethodPut, "/v1/create"); !ok {
		t.Fatalf("route must be found")
	} else if h.(*handler).name != "first" {
		t.Fatalf("invalid handler found: %s != %s", h.(*handler).name, "first")
	}
}