1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
|
package gus_test
import (
"bytes"
"context"
"io"
"testing"
"github.com/stretchr/testify/assert"
"github.com/stretchr/testify/require"
"tildegit.org/tjp/gus"
"tildegit.org/tjp/gus/gemini"
)
var h1 = gus.HandlerFunc(func(_ context.Context, _ *gus.Request) *gus.Response {
return gemini.Success("", &bytes.Buffer{})
})
func mw1(h gus.Handler) gus.Handler {
return gus.HandlerFunc(func(ctx context.Context, req *gus.Request) *gus.Response {
resp := h.Handle(ctx, req)
resp.Body = io.MultiReader(resp.Body, bytes.NewBufferString("\nmiddleware 1"))
return resp
})
}
func mw2(h gus.Handler) gus.Handler {
return gus.HandlerFunc(func(ctx context.Context, req *gus.Request) *gus.Response {
resp := h.Handle(ctx, req)
resp.Body = io.MultiReader(resp.Body, bytes.NewBufferString("\nmiddleware 2"))
return resp
})
}
func TestRouterUse(t *testing.T) {
r := &gus.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(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 := &gus.Router{}
outer.Use(mw2)
inner := &gus.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(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(context.Background(), request)
require.Nil(t, response)
}
|
