You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
75 lines
1.7 KiB
Go
75 lines
1.7 KiB
Go
package metric
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"github.com/prometheus/client_golang/prometheus/testutil"
|
|
"github.com/stretchr/testify/assert"
|
|
"github.com/zeromicro/go-zero/core/proc"
|
|
)
|
|
|
|
func TestNewGaugeVec(t *testing.T) {
|
|
gaugeVec := NewGaugeVec(&GaugeVecOpts{
|
|
Namespace: "rpc_server",
|
|
Subsystem: "requests",
|
|
Name: "duration",
|
|
Help: "rpc server requests duration(ms).",
|
|
})
|
|
defer gaugeVec.close()
|
|
gaugeVecNil := NewGaugeVec(nil)
|
|
assert.NotNil(t, gaugeVec)
|
|
assert.Nil(t, gaugeVecNil)
|
|
|
|
proc.Shutdown()
|
|
}
|
|
|
|
func TestGaugeInc(t *testing.T) {
|
|
startAgent()
|
|
gaugeVec := NewGaugeVec(&GaugeVecOpts{
|
|
Namespace: "rpc_client2",
|
|
Subsystem: "requests",
|
|
Name: "duration_ms",
|
|
Help: "rpc server requests duration(ms).",
|
|
Labels: []string{"path"},
|
|
})
|
|
defer gaugeVec.close()
|
|
gv, _ := gaugeVec.(*promGaugeVec)
|
|
gv.Inc("/users")
|
|
gv.Inc("/users")
|
|
r := testutil.ToFloat64(gv.gauge)
|
|
assert.Equal(t, float64(2), r)
|
|
}
|
|
|
|
func TestGaugeAdd(t *testing.T) {
|
|
startAgent()
|
|
gaugeVec := NewGaugeVec(&GaugeVecOpts{
|
|
Namespace: "rpc_client",
|
|
Subsystem: "request",
|
|
Name: "duration_ms",
|
|
Help: "rpc server requests duration(ms).",
|
|
Labels: []string{"path"},
|
|
})
|
|
defer gaugeVec.close()
|
|
gv, _ := gaugeVec.(*promGaugeVec)
|
|
gv.Add(-10, "/classroom")
|
|
gv.Add(30, "/classroom")
|
|
r := testutil.ToFloat64(gv.gauge)
|
|
assert.Equal(t, float64(20), r)
|
|
}
|
|
|
|
func TestGaugeSet(t *testing.T) {
|
|
startAgent()
|
|
gaugeVec := NewGaugeVec(&GaugeVecOpts{
|
|
Namespace: "http_client",
|
|
Subsystem: "request",
|
|
Name: "duration_ms",
|
|
Help: "rpc server requests duration(ms).",
|
|
Labels: []string{"path"},
|
|
})
|
|
gaugeVec.close()
|
|
gv, _ := gaugeVec.(*promGaugeVec)
|
|
gv.Set(666, "/users")
|
|
r := testutil.ToFloat64(gv.gauge)
|
|
assert.Equal(t, float64(666), r)
|
|
}
|