mirror of https://github.com/XTLS/Xray-core
92 lines
1.8 KiB
Go
92 lines
1.8 KiB
Go
package command_test
|
|
|
|
import (
|
|
"context"
|
|
"testing"
|
|
|
|
"github.com/google/go-cmp/cmp"
|
|
"github.com/google/go-cmp/cmp/cmpopts"
|
|
"github.com/xtls/xray-core/app/stats"
|
|
. "github.com/xtls/xray-core/app/stats/command"
|
|
"github.com/xtls/xray-core/common"
|
|
)
|
|
|
|
func TestGetStats(t *testing.T) {
|
|
m, err := stats.NewManager(context.Background(), &stats.Config{})
|
|
common.Must(err)
|
|
|
|
sc, err := m.RegisterCounter("test_counter")
|
|
common.Must(err)
|
|
|
|
sc.Set(1)
|
|
|
|
s := NewStatsServer(m)
|
|
|
|
testCases := []struct {
|
|
name string
|
|
reset bool
|
|
value int64
|
|
err bool
|
|
}{
|
|
{
|
|
name: "counterNotExist",
|
|
err: true,
|
|
},
|
|
{
|
|
name: "test_counter",
|
|
reset: true,
|
|
value: 1,
|
|
},
|
|
{
|
|
name: "test_counter",
|
|
value: 0,
|
|
},
|
|
}
|
|
for _, tc := range testCases {
|
|
resp, err := s.GetStats(context.Background(), &GetStatsRequest{
|
|
Name: tc.name,
|
|
Reset_: tc.reset,
|
|
})
|
|
if tc.err {
|
|
if err == nil {
|
|
t.Error("nil error: ", tc.name)
|
|
}
|
|
} else {
|
|
common.Must(err)
|
|
if r := cmp.Diff(resp.Stat, &Stat{Name: tc.name, Value: tc.value}, cmpopts.IgnoreUnexported(Stat{})); r != "" {
|
|
t.Error(r)
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
func TestQueryStats(t *testing.T) {
|
|
m, err := stats.NewManager(context.Background(), &stats.Config{})
|
|
common.Must(err)
|
|
|
|
sc1, err := m.RegisterCounter("test_counter")
|
|
common.Must(err)
|
|
sc1.Set(1)
|
|
|
|
sc2, err := m.RegisterCounter("test_counter_2")
|
|
common.Must(err)
|
|
sc2.Set(2)
|
|
|
|
sc3, err := m.RegisterCounter("test_counter_3")
|
|
common.Must(err)
|
|
sc3.Set(3)
|
|
|
|
s := NewStatsServer(m)
|
|
resp, err := s.QueryStats(context.Background(), &QueryStatsRequest{
|
|
Pattern: "counter_",
|
|
})
|
|
common.Must(err)
|
|
if r := cmp.Diff(resp.Stat, []*Stat{
|
|
{Name: "test_counter_2", Value: 2},
|
|
{Name: "test_counter_3", Value: 3},
|
|
}, cmpopts.SortSlices(func(s1, s2 *Stat) bool { return s1.Name < s2.Name }),
|
|
cmpopts.IgnoreUnexported(Stat{})); r != "" {
|
|
t.Error(r)
|
|
}
|
|
}
|