router_test.go

package drouter

import (
	"reflect"
	"testing"
)

func TestRouterLookup(t *testing.T) {
	routed := false
	wantHandle := func() {
		routed = true
	}
	wantParams := Params{Param{"name", "gopher"}}

	router := New()

	// try empty router first
	params := make(Params, 0, 1)
	handle, tsr := router.Lookup("/nope", &params)
	if handle != nil {
		t.Fatalf("Got handle for unregistered pattern: %v", handle)
	}
	if tsr {
		t.Error("Got wrong TSR recommendation!")
	}

	// insert route and try again
	router.AddRoute("/user/:name", wantHandle)
	params = make(Params, 0, 1)
	handle, _ = router.Lookup("/user/gopher", &params)
	if handle == nil {
		t.Fatal("Got no handle!")
	} else {
		handle.(func())()
		if !routed {
			t.Fatal("Routing failed!")
		}
	}
	if !reflect.DeepEqual(params, wantParams) {
		t.Fatalf("Wrong parameter values: want %v, got %v", wantParams, params)
	}
	routed = false

	// route without param
	router.AddRoute("/user", wantHandle)
	params = nil
	handle, _ = router.Lookup("/user", &params)
	if handle == nil {
		t.Fatal("Got no handle!")
	} else {
		handle.(func())()
		if !routed {
			t.Fatal("Routing failed!")
		}
	}
	if params != nil {
		t.Fatalf("Wrong parameter values: want %v, got %v", nil, params)
	}

	params = make(Params, 0, 1)
	handle, tsr = router.Lookup("/user/gopher/", &params)
	if handle != nil {
		t.Fatalf("Got handle for unregistered pattern: %v", handle)
	}
	if !tsr {
		t.Error("Got no TSR recommendation!")
	}

	handle, tsr = router.Lookup("/nope", &params)
	if handle != nil {
		t.Fatalf("Got handle for unregistered pattern: %v", handle)
	}
	if tsr {
		t.Error("Got wrong TSR recommendation!")
	}
}