You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
63 lines
1.9 KiB
63 lines
1.9 KiB
7 months ago
|
// Copyright (c) 2023 The Jaeger Authors.
|
||
|
// Copyright (c) 2017 Uber Technologies, Inc.
|
||
|
//
|
||
|
// 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 rpcmetrics
|
||
|
|
||
|
import (
|
||
|
"testing"
|
||
|
|
||
|
"github.com/stretchr/testify/assert"
|
||
|
|
||
|
"github.com/jaegertracing/jaeger/internal/metricstest"
|
||
|
)
|
||
|
|
||
|
// E.g. tags("key", "value", "key", "value")
|
||
|
func tags(kv ...string) map[string]string {
|
||
|
m := make(map[string]string)
|
||
|
for i := 0; i < len(kv)-1; i += 2 {
|
||
|
m[kv[i]] = kv[i+1]
|
||
|
}
|
||
|
return m
|
||
|
}
|
||
|
|
||
|
func endpointTags(endpoint string, kv ...string) map[string]string {
|
||
|
return tags(append([]string{"endpoint", endpoint}, kv...)...)
|
||
|
}
|
||
|
|
||
|
func TestMetricsByEndpoint(t *testing.T) {
|
||
|
met := metricstest.NewFactory(0)
|
||
|
mbe := newMetricsByEndpoint(met, DefaultNameNormalizer, 2)
|
||
|
|
||
|
m1 := mbe.get("abc1")
|
||
|
m2 := mbe.get("abc1") // from cache
|
||
|
m2a := mbe.getWithWriteLock("abc1") // from cache in double-checked lock
|
||
|
assert.Equal(t, m1, m2)
|
||
|
assert.Equal(t, m1, m2a)
|
||
|
|
||
|
m3 := mbe.get("abc3")
|
||
|
m4 := mbe.get("overflow")
|
||
|
m5 := mbe.get("overflow2")
|
||
|
|
||
|
for _, m := range []*Metrics{m1, m2, m2a, m3, m4, m5} {
|
||
|
m.RequestCountSuccess.Inc(1)
|
||
|
}
|
||
|
|
||
|
met.AssertCounterMetrics(t,
|
||
|
metricstest.ExpectedMetric{Name: "requests", Tags: endpointTags("abc1", "error", "false"), Value: 3},
|
||
|
metricstest.ExpectedMetric{Name: "requests", Tags: endpointTags("abc3", "error", "false"), Value: 1},
|
||
|
metricstest.ExpectedMetric{Name: "requests", Tags: endpointTags("other", "error", "false"), Value: 2},
|
||
|
)
|
||
|
}
|