diff options
| -rw-r--r-- | examples/fileserver/main.go | 2 | ||||
| -rw-r--r-- | router_test.go | 6 |
2 files changed, 4 insertions, 4 deletions
diff --git a/examples/fileserver/main.go b/examples/fileserver/main.go index df5e81d..c374cc4 100644 --- a/examples/fileserver/main.go +++ b/examples/fileserver/main.go @@ -35,7 +35,7 @@ func main() { router := &sr.Router{} router.Route("/*", handler) - handler = sr.HandlerFunc(router.Handler) + handler = router.Handler() _, infoLog, _, errLog := logging.DefaultLoggers() diff --git a/router_test.go b/router_test.go index 5430de7..ebcbce8 100644 --- a/router_test.go +++ b/router_test.go @@ -42,7 +42,7 @@ func TestRouterUse(t *testing.T) { request, err := gemini.ParseRequest(bytes.NewBufferString("/\r\n")) require.Nil(t, err) - response := r.Handler(context.Background(), request) + response := r.Handler().Handle(context.Background(), request) require.NotNil(t, response) body, err := io.ReadAll(response.Body) @@ -64,7 +64,7 @@ func TestRouterMount(t *testing.T) { request, err := gemini.ParseRequest(bytes.NewBufferString("/foo/bar\r\n")) require.Nil(t, err) - response := outer.Handler(context.Background(), request) + response := outer.Handler().Handle(context.Background(), request) require.NotNil(t, response) body, err := io.ReadAll(response.Body) @@ -75,6 +75,6 @@ func TestRouterMount(t *testing.T) { request, err = gemini.ParseRequest(bytes.NewBufferString("/foo\r\n")) require.Nil(t, err) - response = outer.Handler(context.Background(), request) + response = outer.Handler().Handle(context.Background(), request) require.Nil(t, response) } |