81 lines
1.9 KiB
Go
81 lines
1.9 KiB
Go
package sliderule_test
|
|
|
|
import (
|
|
"bytes"
|
|
"context"
|
|
"io"
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
"github.com/stretchr/testify/require"
|
|
|
|
sr "tildegit.org/tjp/sliderule"
|
|
"tildegit.org/tjp/sliderule/gemini"
|
|
)
|
|
|
|
var h1 = sr.HandlerFunc(func(_ context.Context, _ *sr.Request) *sr.Response {
|
|
return gemini.Success("", &bytes.Buffer{})
|
|
})
|
|
|
|
func mw1(h sr.Handler) sr.Handler {
|
|
return sr.HandlerFunc(func(ctx context.Context, req *sr.Request) *sr.Response {
|
|
resp := h.Handle(ctx, req)
|
|
resp.Body = io.MultiReader(resp.Body, bytes.NewBufferString("\nmiddleware 1"))
|
|
return resp
|
|
})
|
|
}
|
|
|
|
func mw2(h sr.Handler) sr.Handler {
|
|
return sr.HandlerFunc(func(ctx context.Context, req *sr.Request) *sr.Response {
|
|
resp := h.Handle(ctx, req)
|
|
resp.Body = io.MultiReader(resp.Body, bytes.NewBufferString("\nmiddleware 2"))
|
|
return resp
|
|
})
|
|
}
|
|
|
|
func TestRouterUse(t *testing.T) {
|
|
r := &sr.Router{}
|
|
r.Use(mw1)
|
|
r.Use(mw2)
|
|
r.Route("/", h1)
|
|
|
|
request, err := gemini.ParseRequest(bytes.NewBufferString("/\r\n"))
|
|
require.Nil(t, err)
|
|
|
|
response := r.Handler().Handle(context.Background(), request)
|
|
require.NotNil(t, response)
|
|
|
|
body, err := io.ReadAll(response.Body)
|
|
require.Nil(t, err)
|
|
|
|
assert.Equal(t, "\nmiddleware 2\nmiddleware 1", string(body))
|
|
}
|
|
|
|
func TestRouterMount(t *testing.T) {
|
|
outer := &sr.Router{}
|
|
outer.Use(mw2)
|
|
|
|
inner := &sr.Router{}
|
|
inner.Use(mw1)
|
|
inner.Route("/bar", h1)
|
|
|
|
outer.Mount("/foo", inner)
|
|
|
|
request, err := gemini.ParseRequest(bytes.NewBufferString("/foo/bar\r\n"))
|
|
require.Nil(t, err)
|
|
|
|
response := outer.Handler().Handle(context.Background(), request)
|
|
require.NotNil(t, response)
|
|
|
|
body, err := io.ReadAll(response.Body)
|
|
require.Nil(t, err)
|
|
|
|
assert.Equal(t, "\nmiddleware 1\nmiddleware 2", string(body))
|
|
|
|
request, err = gemini.ParseRequest(bytes.NewBufferString("/foo\r\n"))
|
|
require.Nil(t, err)
|
|
|
|
response = outer.Handler().Handle(context.Background(), request)
|
|
require.Nil(t, response)
|
|
}
|