123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159 |
- /*
- Copyright 2019 The Vitess 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 stats
- import (
- "expvar"
- "reflect"
- "testing"
- )
- func clear() {
- defaultVarGroup.vars = make(map[string]expvar.Var)
- defaultVarGroup.newVarHook = nil
- *combineDimensions = ""
- *dropVariables = ""
- combinedDimensions = nil
- droppedVars = nil
- }
- func TestNoHook(t *testing.T) {
- clear()
- v := NewCounter("plainint", "help")
- v.Add(1)
- if v.String() != "1" {
- t.Errorf("want 1, got %s", v.String())
- }
- }
- func TestString(t *testing.T) {
- var gotname string
- var gotv *String
- clear()
- Register(func(name string, v expvar.Var) {
- gotname = name
- gotv = v.(*String)
- })
- v := NewString("String")
- if gotname != "String" {
- t.Errorf("want String, got %s", gotname)
- }
- if gotv != v {
- t.Errorf("want %#v, got %#v", v, gotv)
- }
- v.Set("a\"b")
- if v.Get() != "a\"b" {
- t.Errorf("want \"a\"b\", got %#v", gotv)
- }
- if v.String() != "\"a\\\"b\"" {
- t.Errorf("want \"\"a\\\"b\"\", got %#v", gotv)
- }
- f := StringFunc(func() string {
- return "a"
- })
- if f.String() != "\"a\"" {
- t.Errorf("want \"a\", got %v", f.String())
- }
- }
- type Mystr string
- func (m *Mystr) String() string {
- return string(*m)
- }
- func TestPublish(t *testing.T) {
- var gotname string
- var gotv expvar.Var
- clear()
- Register(func(name string, v expvar.Var) {
- gotname = name
- gotv = v.(*Mystr)
- })
- v := Mystr("abcd")
- Publish("Mystr", &v)
- if gotname != "Mystr" {
- t.Errorf("want Mystr, got %s", gotname)
- }
- if gotv != &v {
- t.Errorf("want %#v, got %#v", &v, gotv)
- }
- }
- func f() string {
- return "abcd"
- }
- type expvarFunc func() string
- func (f expvarFunc) String() string {
- return f()
- }
- func TestPublishFunc(t *testing.T) {
- var gotname string
- var gotv expvarFunc
- clear()
- Register(func(name string, v expvar.Var) {
- gotname = name
- gotv = v.(expvarFunc)
- })
- publish("Myfunc", expvarFunc(f))
- if gotname != "Myfunc" {
- t.Errorf("want Myfunc, got %s", gotname)
- }
- if gotv.String() != f() {
- t.Errorf("want %v, got %#v", f(), gotv())
- }
- }
- func TestDropVariable(t *testing.T) {
- clear()
- *dropVariables = "dropTest"
- // This should not panic.
- _ = NewGaugesWithSingleLabel("dropTest", "help", "label")
- _ = NewGaugesWithSingleLabel("dropTest", "help", "label")
- }
- func TestStringMapToString(t *testing.T) {
- expected1 := "{\"aaa\": \"111\", \"bbb\": \"222\"}"
- expected2 := "{\"bbb\": \"222\", \"aaa\": \"111\"}"
- got := stringMapToString(map[string]string{"aaa": "111", "bbb": "222"})
- if got != expected1 && got != expected2 {
- t.Errorf("expected %v or %v, got %v", expected1, expected2, got)
- }
- }
- func TestParseCommonTags(t *testing.T) {
- res := ParseCommonTags("")
- if len(res) != 0 {
- t.Errorf("expected empty result, got %v", res)
- }
- res = ParseCommonTags("s,a:b ")
- expected1 := map[string]string{"a": "b"}
- if !reflect.DeepEqual(expected1, res) {
- t.Errorf("expected %v, got %v", expected1, res)
- }
- res = ParseCommonTags("a:b, c:d")
- expected2 := map[string]string{"a": "b", "c": "d"}
- if !reflect.DeepEqual(expected2, res) {
- t.Errorf("expected %v, got %v", expected2, res)
- }
- }
|