| Server IP : 84.32.84.20 / Your IP : 216.73.217.6 Web Server : LiteSpeed System : Linux id-dci-web1986.main-hosting.eu 5.14.0-611.26.1.el9_7.x86_64 #1 SMP PREEMPT_DYNAMIC Thu Jan 29 05:24:47 EST 2026 x86_64 User : u686484674 ( 686484674) PHP Version : 8.0.30 Disable Function : system, exec, shell_exec, passthru, mysql_list_dbs, ini_alter, dl, symlink, link, chgrp, leak, popen, apache_child_terminate, virtual, mb_send_mail MySQL : OFF | cURL : ON | WGET : ON | Perl : OFF | Python : OFF | Sudo : OFF | Pkexec : OFF Directory : /proc/self/root/opt/go/pkg/mod/github.com/prometheus/common@v0.61.0/route/ |
Upload File : |
// Copyright 2015 The Prometheus Authors
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
// http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.
package route
import (
"net/http"
"net/http/httptest"
"testing"
"github.com/stretchr/testify/require"
)
func TestRedirect(t *testing.T) {
router := New().WithPrefix("/test/prefix")
w := httptest.NewRecorder()
r, err := http.NewRequest("GET", "http://localhost:9090/foo", nil)
require.NoErrorf(t, err, "Error building test request: %s", err)
router.Redirect(w, r, "/some/endpoint", http.StatusFound)
require.Equalf(t, http.StatusFound, w.Code, "Unexpected redirect status code: got %d, want %d", w.Code, http.StatusFound)
want := "/test/prefix/some/endpoint"
got := w.Header()["Location"][0]
require.Equalf(t, want, got, "Unexpected redirect location: got %s, want %s", got, want)
}
func TestContext(t *testing.T) {
router := New()
router.Get("/test/:foo/", func(w http.ResponseWriter, r *http.Request) {
want := "bar"
got := Param(r.Context(), "foo")
require.Equalf(t, want, got, "Unexpected context value: want %q, got %q", want, got)
})
r, err := http.NewRequest("GET", "http://localhost:9090/test/bar/", nil)
require.NoErrorf(t, err, "Error building test request: %s", err)
router.ServeHTTP(nil, r)
}
func TestContextWithValue(t *testing.T) {
router := New()
router.Get("/test/:foo/", func(w http.ResponseWriter, r *http.Request) {
want := "bar"
got := Param(r.Context(), "foo")
require.Equalf(t, want, got, "Unexpected context value: want %q, got %q", want, got)
want = "ipsum"
got = Param(r.Context(), "lorem")
require.Equalf(t, want, got, "Unexpected context value: want %q, got %q", want, got)
want = "sit"
got = Param(r.Context(), "dolor")
require.Equalf(t, want, got, "Unexpected context value: want %q, got %q", want, got)
})
r, err := http.NewRequest("GET", "http://localhost:9090/test/bar/", nil)
require.NoErrorf(t, err, "Error building test request: %s", err)
params := map[string]string{
"lorem": "ipsum",
"dolor": "sit",
}
ctx := r.Context()
for p, v := range params {
ctx = WithParam(ctx, p, v)
}
r = r.WithContext(ctx)
router.ServeHTTP(nil, r)
}
func TestContextWithoutValue(t *testing.T) {
router := New()
router.Get("/test", func(w http.ResponseWriter, r *http.Request) {
want := ""
got := Param(r.Context(), "foo")
require.Equalf(t, want, got, "Unexpected context value: want %q, got %q", want, got)
})
r, err := http.NewRequest("GET", "http://localhost:9090/test", nil)
require.NoErrorf(t, err, "Error building test request: %s", err)
router.ServeHTTP(nil, r)
}
func TestInstrumentation(t *testing.T) {
var got string
cases := []struct {
router *Router
want string
}{
{
router: New(),
want: "",
}, {
router: New().WithInstrumentation(func(handlerName string, handler http.HandlerFunc) http.HandlerFunc {
got = handlerName
return handler
}),
want: "/foo",
},
}
for _, c := range cases {
c.router.Get("/foo", func(w http.ResponseWriter, r *http.Request) {})
r, err := http.NewRequest("GET", "http://localhost:9090/foo", nil)
require.NoErrorf(t, err, "Error building test request: %s", err)
c.router.ServeHTTP(nil, r)
require.Equalf(t, c.want, got, "Unexpected value: want %q, got %q", c.want, got)
}
}
func TestInstrumentations(t *testing.T) {
got := make([]string, 0)
cases := []struct {
router *Router
want []string
}{
{
router: New(),
want: []string{},
}, {
router: New().
WithInstrumentation(
func(handlerName string, handler http.HandlerFunc) http.HandlerFunc {
got = append(got, "1"+handlerName)
return handler
}).
WithInstrumentation(
func(handlerName string, handler http.HandlerFunc) http.HandlerFunc {
got = append(got, "2"+handlerName)
return handler
}).
WithInstrumentation(
func(handlerName string, handler http.HandlerFunc) http.HandlerFunc {
got = append(got, "3"+handlerName)
return handler
}),
want: []string{"1/foo", "2/foo", "3/foo"},
},
}
for _, c := range cases {
c.router.Get("/foo", func(w http.ResponseWriter, r *http.Request) {})
r, err := http.NewRequest("GET", "http://localhost:9090/foo", nil)
require.NoErrorf(t, err, "Error building test request: %s", err)
c.router.ServeHTTP(nil, r)
require.Equalf(t, len(c.want), len(got), "Unexpected value: want %q, got %q", c.want, got)
for i, v := range c.want {
require.Equalf(t, v, got[i], "Unexpected value: want %q, got %q", c.want, got)
}
}
}