route_test.go

package hime

import (
	"net/http/httptest"
	"testing"

	"github.com/stretchr/testify/assert"
)

func TestRoute(t *testing.T) {
	t.Run("clone nil", func(t *testing.T) {
		assert.Nil(t, cloneRoutes(nil))
	})

	t.Run("clone empty", func(t *testing.T) {
		r := Routes{}
		assert.Equal(t, cloneRoutes(r), Routes{})
	})

	t.Run("clone data", func(t *testing.T) {
		r := Routes{
			"a": "/b",
			"b": "/cd",
		}

		p := cloneRoutes(r)
		assert.Equal(t, p, r)

		p["a"] = "/a"
		assert.NotEqual(t, p, r)
	})

	t.Run("App", func(t *testing.T) {
		t.Run("panic when retrieve route from empty route app", func(t *testing.T) {
			app := New()

			assert.Panics(t, func() { app.Route("r1") })
		})

		t.Run("retrieve valid route", func(t *testing.T) {
			app := New()
			app.Routes(Routes{
				"a": "/b",
				"b": "/cd",
			})

			assert.Equal(t, app.Route("a"), "/b")
			assert.Equal(t, app.Route("b"), "/cd")
		})
	})

	t.Run("panic when retrieve not exists route", func(t *testing.T) {
		app := New()
		app.Routes(Routes{
			"a": "/b",
			"b": "/cd",
		})

		assert.Panics(t, func() { app.Route("c") })
	})

	t.Run("Context", func(t *testing.T) {
		t.Run("retrieve route from context", func(t *testing.T) {
			app := New()
			app.Routes(Routes{
				"a": "/b",
				"b": "/cd",
			})

			w := httptest.NewRecorder()
			r := httptest.NewRequest("GET", "/", nil)

			app.Handler(Handler(func(ctx *Context) error {
				assert.Equal(t, ctx.Route("a"), "/b")
				assert.Equal(t, ctx.Route("b"), "/cd")
				assert.Equal(t, Route(ctx, "a"), "/b")
				assert.Equal(t, Route(ctx, "b"), "/cd")
				return nil
			}))
			app.ServeHTTP(w, r)
		})

		t.Run("panic when retrieve not exists route", func(t *testing.T) {
			app := New()
			app.Routes(Routes{
				"a": "/b",
				"b": "/cd",
			})

			w := httptest.NewRecorder()
			r := httptest.NewRequest("GET", "/", nil)

			app.Handler(Handler(func(ctx *Context) error {
				assert.Panics(t, func() { ctx.Route("c") })
				return nil
			}))
			app.ServeHTTP(w, r)
		})
	})
}