mirror of
https://github.com/VictoriaMetrics/VictoriaMetrics.git
synced 2024-12-21 07:56:26 +01:00
7e5555f9c7
Fixed decoding of label values with slash for pushgateway and prometheus golang client compatibility + added some tests. (#4962)
95 lines
3.1 KiB
Go
95 lines
3.1 KiB
Go
package common
|
|
|
|
import (
|
|
"fmt"
|
|
"net/http"
|
|
"sort"
|
|
"strings"
|
|
"testing"
|
|
|
|
"github.com/VictoriaMetrics/VictoriaMetrics/lib/prompbmarshal"
|
|
)
|
|
|
|
func TestGetExtraLabelsSuccess(t *testing.T) {
|
|
f := func(requestURI, expectedLabels string) {
|
|
t.Helper()
|
|
fullURL := "http://fobar" + requestURI
|
|
req, err := http.NewRequest(http.MethodGet, fullURL, nil)
|
|
if err != nil {
|
|
t.Fatalf("cannot parse %q: %s", fullURL, err)
|
|
}
|
|
extraLabels, err := GetExtraLabels(req)
|
|
if err != nil {
|
|
t.Fatalf("unexpected error: %s", err)
|
|
}
|
|
labelsStr := getLabelsString(extraLabels)
|
|
if labelsStr != expectedLabels {
|
|
t.Fatalf("unexpected labels;\ngot\n%s\nwant\n%s", labelsStr, expectedLabels)
|
|
}
|
|
}
|
|
f("", `{}`)
|
|
f("/foo/bar", `{}`)
|
|
f("/foo?extra_label=foo=bar", `{foo="bar"}`)
|
|
f("/foo?extra_label=a=x&extra_label=b=y", `{a="x",b="y"}`)
|
|
f("/metrics/job/foo", `{job="foo"}`)
|
|
f("/metrics/job/foo?extra_label=a=b", `{a="b",job="foo"}`)
|
|
f("/metrics/job/foo/b/bcd?extra_label=a=b&extra_label=qwe=rty", `{a="b",b="bcd",job="foo",qwe="rty"}`)
|
|
f("/metrics/job/titan/name/%CE%A0%CF%81%CE%BF%CE%BC%CE%B7%CE%B8%CE%B5%CF%8D%CF%82", `{job="titan",name="Προμηθεύς"}`)
|
|
f("/metrics/job/titan/name@base64/zqDPgc6_zrzOt864zrXPjc-C", `{job="titan",name="Προμηθεύς"}`)
|
|
}
|
|
|
|
func TestGetPushgatewayLabelsSuccess(t *testing.T) {
|
|
f := func(path, expectedLabels string) {
|
|
t.Helper()
|
|
labels, err := getPushgatewayLabels(path)
|
|
if err != nil {
|
|
t.Fatalf("unexpected error in getPushgatewayLabels(%q): %s", path, err)
|
|
}
|
|
labelsStr := getLabelsString(labels)
|
|
if labelsStr != expectedLabels {
|
|
t.Fatalf("unexpected labels returned from getPushgatewayLabels(%q);\ngot\n%s\nwant\n%s", path, labelsStr, expectedLabels)
|
|
}
|
|
}
|
|
f("", "{}")
|
|
f("/foo/bar", "{}")
|
|
f("/metrics/foo/bar", "{}")
|
|
f("/metrics/job", "{}")
|
|
f("/metrics/job@base64", "{}")
|
|
f("/metrics/job/", "{}")
|
|
f("/metrics/job/foo", `{job="foo"}`)
|
|
f("/foo/metrics/job/foo", `{job="foo"}`)
|
|
f("/api/v1/import/prometheus/metrics/job/foo", `{job="foo"}`)
|
|
f("/foo/metrics/job@base64/Zm9v", `{job="foo"}`)
|
|
f("/foo/metrics/job/x/a/foo/aaa/bar", `{a="foo",aaa="bar",job="x"}`)
|
|
f("/foo/metrics/job/x/a@base64/Zm9v", `{a="foo",job="x"}`)
|
|
f("/metrics/job/test/region@base64/YXotc291dGhlYXN0LTEtZjAxL3d6eS1hei1zb3V0aGVhc3QtMQ", `{job="test",region="az-southeast-1-f01/wzy-az-southeast-1"}`)
|
|
f("/metrics/job/test/empty@base64/=", `{job="test"}`)
|
|
f("/metrics/job/test/test@base64/PT0vPT0", `{job="test",test="==/=="}`)
|
|
}
|
|
|
|
func TestGetPushgatewayLabelsFailure(t *testing.T) {
|
|
f := func(path string) {
|
|
t.Helper()
|
|
labels, err := getPushgatewayLabels(path)
|
|
if err == nil {
|
|
labelsStr := getLabelsString(labels)
|
|
t.Fatalf("expecting non-nil error for getPushgatewayLabels(%q); got labels %s", path, labelsStr)
|
|
}
|
|
}
|
|
// missing bar value
|
|
f("/metrics/job/foo/bar")
|
|
// invalid base64 encoding for job
|
|
f("/metrics/job@base64/#$%")
|
|
// invalid base64 encoding for non-job label
|
|
f("/metrics/job/foo/bar@base64/#$%")
|
|
}
|
|
|
|
func getLabelsString(labels []prompbmarshal.Label) string {
|
|
a := make([]string, len(labels))
|
|
for i, label := range labels {
|
|
a[i] = fmt.Sprintf("%s=%q", label.Name, label.Value)
|
|
}
|
|
sort.Strings(a)
|
|
return "{" + strings.Join(a, ",") + "}"
|
|
}
|