router_test.go

package dotweb

import (
	"testing"
	"time"

	"github.com/devfeel/dotweb/session"
	"github.com/devfeel/dotweb/test"
)

func TestRouter_ServeHTTP(t *testing.T) {
	param := &InitContextParam{
		t,
		"",
		"",
		test.ToDefault,
	}

	context := initAllContext(param)

	app := New()
	server := app.HttpServer
	r := NewRouter(server)

	r.ServeHTTP(context)
}

func TestWrapRouterHandle(t *testing.T) {
	param := &InitContextParam{
		t,
		"",
		"",
		test.ToDefault,
	}

	context := initAllContext(param)

	app := New()
	server := app.HttpServer
	router := server.Router().(*router)
	// use default config
	server.SetSessionConfig(session.NewDefaultRuntimeConfig())
	handle := router.wrapRouterHandle(Index, false)

	handle(context)
}

func TestLogWebsocketContext(t *testing.T) {
	param := &InitContextParam{
		t,
		"",
		"",
		test.ToDefault,
	}

	context := initAllContext(param)

	log := logWebsocketContext(context, time.Now().Unix())
	t.Log("logContext:", log)
	// test.NotNil(t,log)
	test.Equal(t, "", "")
}

func BenchmarkRouter_MatchPath(b *testing.B) {
	app := New()
	server := app.HttpServer
	r := NewRouter(server)
	r.GET("/1", func(ctx Context) error {
		ctx.WriteString("1")
		return nil
	})
	r.GET("/2", func(ctx Context) error {
		ctx.WriteString("2")
		return nil
	})
	r.POST("/p1", func(ctx Context) error {
		ctx.WriteString("1")
		return nil
	})
	r.POST("/p2", func(ctx Context) error {
		ctx.WriteString("2")
		return nil
	})

	for i := 0; i < b.N; i++ {
		if root := r.Nodes["GET"]; root != nil {
			root.getNode("/1?q=1")
		}
	}
}