mirror of
https://github.com/prometheus/prometheus.git
synced 2024-12-27 06:29:42 -08:00
57dcdfb15f
* promql: Rewrote tests with testutil for functions_test Signed-off-by: Elif T. Kuş <elifkus@gmail.com> * pkg/relabel: Rewrote tests with testutil for relabel_test Signed-off-by: Elif T. Kuş <elifkus@gmail.com> * discovery/consul: Rewrote tests with testutil for consul_test Signed-off-by: Elif T. Kuş <elifkus@gmail.com> * scrape: Rewrote tests with testutil for manager_test Signed-off-by: Elif T. Kuş <elifkus@gmail.com>
421 lines
10 KiB
Go
421 lines
10 KiB
Go
// 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 relabel
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"github.com/prometheus/common/model"
|
|
|
|
"github.com/prometheus/prometheus/config"
|
|
"github.com/prometheus/prometheus/pkg/labels"
|
|
"github.com/prometheus/prometheus/util/testutil"
|
|
)
|
|
|
|
func TestRelabel(t *testing.T) {
|
|
tests := []struct {
|
|
input labels.Labels
|
|
relabel []*config.RelabelConfig
|
|
output labels.Labels
|
|
}{
|
|
{
|
|
input: labels.FromMap(map[string]string{
|
|
"a": "foo",
|
|
"b": "bar",
|
|
"c": "baz",
|
|
}),
|
|
relabel: []*config.RelabelConfig{
|
|
{
|
|
SourceLabels: model.LabelNames{"a"},
|
|
Regex: config.MustNewRegexp("f(.*)"),
|
|
TargetLabel: "d",
|
|
Separator: ";",
|
|
Replacement: "ch${1}-ch${1}",
|
|
Action: config.RelabelReplace,
|
|
},
|
|
},
|
|
output: labels.FromMap(map[string]string{
|
|
"a": "foo",
|
|
"b": "bar",
|
|
"c": "baz",
|
|
"d": "choo-choo",
|
|
}),
|
|
},
|
|
{
|
|
input: labels.FromMap(map[string]string{
|
|
"a": "foo",
|
|
"b": "bar",
|
|
"c": "baz",
|
|
}),
|
|
relabel: []*config.RelabelConfig{
|
|
{
|
|
SourceLabels: model.LabelNames{"a", "b"},
|
|
Regex: config.MustNewRegexp("f(.*);(.*)r"),
|
|
TargetLabel: "a",
|
|
Separator: ";",
|
|
Replacement: "b${1}${2}m", // boobam
|
|
Action: config.RelabelReplace,
|
|
},
|
|
{
|
|
SourceLabels: model.LabelNames{"c", "a"},
|
|
Regex: config.MustNewRegexp("(b).*b(.*)ba(.*)"),
|
|
TargetLabel: "d",
|
|
Separator: ";",
|
|
Replacement: "$1$2$2$3",
|
|
Action: config.RelabelReplace,
|
|
},
|
|
},
|
|
output: labels.FromMap(map[string]string{
|
|
"a": "boobam",
|
|
"b": "bar",
|
|
"c": "baz",
|
|
"d": "boooom",
|
|
}),
|
|
},
|
|
{
|
|
input: labels.FromMap(map[string]string{
|
|
"a": "foo",
|
|
}),
|
|
relabel: []*config.RelabelConfig{
|
|
{
|
|
SourceLabels: model.LabelNames{"a"},
|
|
Regex: config.MustNewRegexp(".*o.*"),
|
|
Action: config.RelabelDrop,
|
|
}, {
|
|
SourceLabels: model.LabelNames{"a"},
|
|
Regex: config.MustNewRegexp("f(.*)"),
|
|
TargetLabel: "d",
|
|
Separator: ";",
|
|
Replacement: "ch$1-ch$1",
|
|
Action: config.RelabelReplace,
|
|
},
|
|
},
|
|
output: nil,
|
|
},
|
|
{
|
|
input: labels.FromMap(map[string]string{
|
|
"a": "foo",
|
|
"b": "bar",
|
|
}),
|
|
relabel: []*config.RelabelConfig{
|
|
{
|
|
SourceLabels: model.LabelNames{"a"},
|
|
Regex: config.MustNewRegexp(".*o.*"),
|
|
Action: config.RelabelDrop,
|
|
},
|
|
},
|
|
output: nil,
|
|
},
|
|
{
|
|
input: labels.FromMap(map[string]string{
|
|
"a": "abc",
|
|
}),
|
|
relabel: []*config.RelabelConfig{
|
|
{
|
|
SourceLabels: model.LabelNames{"a"},
|
|
Regex: config.MustNewRegexp(".*(b).*"),
|
|
TargetLabel: "d",
|
|
Separator: ";",
|
|
Replacement: "$1",
|
|
Action: config.RelabelReplace,
|
|
},
|
|
},
|
|
output: labels.FromMap(map[string]string{
|
|
"a": "abc",
|
|
"d": "b",
|
|
}),
|
|
},
|
|
{
|
|
input: labels.FromMap(map[string]string{
|
|
"a": "foo",
|
|
}),
|
|
relabel: []*config.RelabelConfig{
|
|
{
|
|
SourceLabels: model.LabelNames{"a"},
|
|
Regex: config.MustNewRegexp("no-match"),
|
|
Action: config.RelabelDrop,
|
|
},
|
|
},
|
|
output: labels.FromMap(map[string]string{
|
|
"a": "foo",
|
|
}),
|
|
},
|
|
{
|
|
input: labels.FromMap(map[string]string{
|
|
"a": "foo",
|
|
}),
|
|
relabel: []*config.RelabelConfig{
|
|
{
|
|
SourceLabels: model.LabelNames{"a"},
|
|
Regex: config.MustNewRegexp("f|o"),
|
|
Action: config.RelabelDrop,
|
|
},
|
|
},
|
|
output: labels.FromMap(map[string]string{
|
|
"a": "foo",
|
|
}),
|
|
},
|
|
{
|
|
input: labels.FromMap(map[string]string{
|
|
"a": "foo",
|
|
}),
|
|
relabel: []*config.RelabelConfig{
|
|
{
|
|
SourceLabels: model.LabelNames{"a"},
|
|
Regex: config.MustNewRegexp("no-match"),
|
|
Action: config.RelabelKeep,
|
|
},
|
|
},
|
|
output: nil,
|
|
},
|
|
{
|
|
input: labels.FromMap(map[string]string{
|
|
"a": "foo",
|
|
}),
|
|
relabel: []*config.RelabelConfig{
|
|
{
|
|
SourceLabels: model.LabelNames{"a"},
|
|
Regex: config.MustNewRegexp("f.*"),
|
|
Action: config.RelabelKeep,
|
|
},
|
|
},
|
|
output: labels.FromMap(map[string]string{
|
|
"a": "foo",
|
|
}),
|
|
},
|
|
{
|
|
// No replacement must be applied if there is no match.
|
|
input: labels.FromMap(map[string]string{
|
|
"a": "boo",
|
|
}),
|
|
relabel: []*config.RelabelConfig{
|
|
{
|
|
SourceLabels: model.LabelNames{"a"},
|
|
Regex: config.MustNewRegexp("f"),
|
|
TargetLabel: "b",
|
|
Replacement: "bar",
|
|
Action: config.RelabelReplace,
|
|
},
|
|
},
|
|
output: labels.FromMap(map[string]string{
|
|
"a": "boo",
|
|
}),
|
|
},
|
|
{
|
|
input: labels.FromMap(map[string]string{
|
|
"a": "foo",
|
|
"b": "bar",
|
|
"c": "baz",
|
|
}),
|
|
relabel: []*config.RelabelConfig{
|
|
{
|
|
SourceLabels: model.LabelNames{"c"},
|
|
TargetLabel: "d",
|
|
Separator: ";",
|
|
Action: config.RelabelHashMod,
|
|
Modulus: 1000,
|
|
},
|
|
},
|
|
output: labels.FromMap(map[string]string{
|
|
"a": "foo",
|
|
"b": "bar",
|
|
"c": "baz",
|
|
"d": "976",
|
|
}),
|
|
},
|
|
{
|
|
input: labels.FromMap(map[string]string{
|
|
"a": "foo",
|
|
"b1": "bar",
|
|
"b2": "baz",
|
|
}),
|
|
relabel: []*config.RelabelConfig{
|
|
{
|
|
Regex: config.MustNewRegexp("(b.*)"),
|
|
Replacement: "bar_${1}",
|
|
Action: config.RelabelLabelMap,
|
|
},
|
|
},
|
|
output: labels.FromMap(map[string]string{
|
|
"a": "foo",
|
|
"b1": "bar",
|
|
"b2": "baz",
|
|
"bar_b1": "bar",
|
|
"bar_b2": "baz",
|
|
}),
|
|
},
|
|
{
|
|
input: labels.FromMap(map[string]string{
|
|
"a": "foo",
|
|
"__meta_my_bar": "aaa",
|
|
"__meta_my_baz": "bbb",
|
|
"__meta_other": "ccc",
|
|
}),
|
|
relabel: []*config.RelabelConfig{
|
|
{
|
|
Regex: config.MustNewRegexp("__meta_(my.*)"),
|
|
Replacement: "${1}",
|
|
Action: config.RelabelLabelMap,
|
|
},
|
|
},
|
|
output: labels.FromMap(map[string]string{
|
|
"a": "foo",
|
|
"__meta_my_bar": "aaa",
|
|
"__meta_my_baz": "bbb",
|
|
"__meta_other": "ccc",
|
|
"my_bar": "aaa",
|
|
"my_baz": "bbb",
|
|
}),
|
|
},
|
|
{ // valid case
|
|
input: labels.FromMap(map[string]string{
|
|
"a": "some-name-value",
|
|
}),
|
|
relabel: []*config.RelabelConfig{
|
|
{
|
|
SourceLabels: model.LabelNames{"a"},
|
|
Regex: config.MustNewRegexp("some-([^-]+)-([^,]+)"),
|
|
Action: config.RelabelReplace,
|
|
Replacement: "${2}",
|
|
TargetLabel: "${1}",
|
|
},
|
|
},
|
|
output: labels.FromMap(map[string]string{
|
|
"a": "some-name-value",
|
|
"name": "value",
|
|
}),
|
|
},
|
|
{ // invalid replacement ""
|
|
input: labels.FromMap(map[string]string{
|
|
"a": "some-name-value",
|
|
}),
|
|
relabel: []*config.RelabelConfig{
|
|
{
|
|
SourceLabels: model.LabelNames{"a"},
|
|
Regex: config.MustNewRegexp("some-([^-]+)-([^,]+)"),
|
|
Action: config.RelabelReplace,
|
|
Replacement: "${3}",
|
|
TargetLabel: "${1}",
|
|
},
|
|
},
|
|
output: labels.FromMap(map[string]string{
|
|
"a": "some-name-value",
|
|
}),
|
|
},
|
|
{ // invalid target_labels
|
|
input: labels.FromMap(map[string]string{
|
|
"a": "some-name-value",
|
|
}),
|
|
relabel: []*config.RelabelConfig{
|
|
{
|
|
SourceLabels: model.LabelNames{"a"},
|
|
Regex: config.MustNewRegexp("some-([^-]+)-([^,]+)"),
|
|
Action: config.RelabelReplace,
|
|
Replacement: "${1}",
|
|
TargetLabel: "${3}",
|
|
},
|
|
{
|
|
SourceLabels: model.LabelNames{"a"},
|
|
Regex: config.MustNewRegexp("some-([^-]+)-([^,]+)"),
|
|
Action: config.RelabelReplace,
|
|
Replacement: "${1}",
|
|
TargetLabel: "0${3}",
|
|
},
|
|
{
|
|
SourceLabels: model.LabelNames{"a"},
|
|
Regex: config.MustNewRegexp("some-([^-]+)-([^,]+)"),
|
|
Action: config.RelabelReplace,
|
|
Replacement: "${1}",
|
|
TargetLabel: "-${3}",
|
|
},
|
|
},
|
|
output: labels.FromMap(map[string]string{
|
|
"a": "some-name-value",
|
|
}),
|
|
},
|
|
{ // more complex real-life like usecase
|
|
input: labels.FromMap(map[string]string{
|
|
"__meta_sd_tags": "path:/secret,job:some-job,label:foo=bar",
|
|
}),
|
|
relabel: []*config.RelabelConfig{
|
|
{
|
|
SourceLabels: model.LabelNames{"__meta_sd_tags"},
|
|
Regex: config.MustNewRegexp("(?:.+,|^)path:(/[^,]+).*"),
|
|
Action: config.RelabelReplace,
|
|
Replacement: "${1}",
|
|
TargetLabel: "__metrics_path__",
|
|
},
|
|
{
|
|
SourceLabels: model.LabelNames{"__meta_sd_tags"},
|
|
Regex: config.MustNewRegexp("(?:.+,|^)job:([^,]+).*"),
|
|
Action: config.RelabelReplace,
|
|
Replacement: "${1}",
|
|
TargetLabel: "job",
|
|
},
|
|
{
|
|
SourceLabels: model.LabelNames{"__meta_sd_tags"},
|
|
Regex: config.MustNewRegexp("(?:.+,|^)label:([^=]+)=([^,]+).*"),
|
|
Action: config.RelabelReplace,
|
|
Replacement: "${2}",
|
|
TargetLabel: "${1}",
|
|
},
|
|
},
|
|
output: labels.FromMap(map[string]string{
|
|
"__meta_sd_tags": "path:/secret,job:some-job,label:foo=bar",
|
|
"__metrics_path__": "/secret",
|
|
"job": "some-job",
|
|
"foo": "bar",
|
|
}),
|
|
},
|
|
{
|
|
input: labels.FromMap(map[string]string{
|
|
"a": "foo",
|
|
"b1": "bar",
|
|
"b2": "baz",
|
|
}),
|
|
relabel: []*config.RelabelConfig{
|
|
{
|
|
Regex: config.MustNewRegexp("(b.*)"),
|
|
Action: config.RelabelLabelKeep,
|
|
},
|
|
},
|
|
output: labels.FromMap(map[string]string{
|
|
"b1": "bar",
|
|
"b2": "baz",
|
|
}),
|
|
},
|
|
{
|
|
input: labels.FromMap(map[string]string{
|
|
"a": "foo",
|
|
"b1": "bar",
|
|
"b2": "baz",
|
|
}),
|
|
relabel: []*config.RelabelConfig{
|
|
{
|
|
Regex: config.MustNewRegexp("(b.*)"),
|
|
Action: config.RelabelLabelDrop,
|
|
},
|
|
},
|
|
output: labels.FromMap(map[string]string{
|
|
"a": "foo",
|
|
}),
|
|
},
|
|
}
|
|
|
|
for _, test := range tests {
|
|
res := Process(test.input, test.relabel...)
|
|
testutil.Equals(t, test.output, res)
|
|
}
|
|
}
|