github.com/roboticscm/goman@v0.0.0-20210203095141-87c07b4a0a55/src/expvar/expvar_test.go (about) 1 // Copyright 2009 The Go Authors. All rights reserved. 2 // Use of this source code is governed by a BSD-style 3 // license that can be found in the LICENSE file. 4 5 package expvar 6 7 import ( 8 "bytes" 9 "encoding/json" 10 "net/http/httptest" 11 "strconv" 12 "testing" 13 ) 14 15 // RemoveAll removes all exported variables. 16 // This is for tests only. 17 func RemoveAll() { 18 mutex.Lock() 19 defer mutex.Unlock() 20 vars = make(map[string]Var) 21 varKeys = nil 22 } 23 24 func TestInt(t *testing.T) { 25 RemoveAll() 26 reqs := NewInt("requests") 27 if reqs.i != 0 { 28 t.Errorf("reqs.i = %v, want 0", reqs.i) 29 } 30 if reqs != Get("requests").(*Int) { 31 t.Errorf("Get() failed.") 32 } 33 34 reqs.Add(1) 35 reqs.Add(3) 36 if reqs.i != 4 { 37 t.Errorf("reqs.i = %v, want 4", reqs.i) 38 } 39 40 if s := reqs.String(); s != "4" { 41 t.Errorf("reqs.String() = %q, want \"4\"", s) 42 } 43 44 reqs.Set(-2) 45 if reqs.i != -2 { 46 t.Errorf("reqs.i = %v, want -2", reqs.i) 47 } 48 } 49 50 func TestFloat(t *testing.T) { 51 RemoveAll() 52 reqs := NewFloat("requests-float") 53 if reqs.f != 0.0 { 54 t.Errorf("reqs.f = %v, want 0", reqs.f) 55 } 56 if reqs != Get("requests-float").(*Float) { 57 t.Errorf("Get() failed.") 58 } 59 60 reqs.Add(1.5) 61 reqs.Add(1.25) 62 if reqs.f != 2.75 { 63 t.Errorf("reqs.f = %v, want 2.75", reqs.f) 64 } 65 66 if s := reqs.String(); s != "2.75" { 67 t.Errorf("reqs.String() = %q, want \"4.64\"", s) 68 } 69 70 reqs.Add(-2) 71 if reqs.f != 0.75 { 72 t.Errorf("reqs.f = %v, want 0.75", reqs.f) 73 } 74 } 75 76 func TestString(t *testing.T) { 77 RemoveAll() 78 name := NewString("my-name") 79 if name.s != "" { 80 t.Errorf("name.s = %q, want \"\"", name.s) 81 } 82 83 name.Set("Mike") 84 if name.s != "Mike" { 85 t.Errorf("name.s = %q, want \"Mike\"", name.s) 86 } 87 88 if s := name.String(); s != "\"Mike\"" { 89 t.Errorf("reqs.String() = %q, want \"\"Mike\"\"", s) 90 } 91 } 92 93 func TestMapCounter(t *testing.T) { 94 RemoveAll() 95 colors := NewMap("bike-shed-colors") 96 97 colors.Add("red", 1) 98 colors.Add("red", 2) 99 colors.Add("blue", 4) 100 colors.AddFloat(`green "midori"`, 4.125) 101 if x := colors.m["red"].(*Int).i; x != 3 { 102 t.Errorf("colors.m[\"red\"] = %v, want 3", x) 103 } 104 if x := colors.m["blue"].(*Int).i; x != 4 { 105 t.Errorf("colors.m[\"blue\"] = %v, want 4", x) 106 } 107 if x := colors.m[`green "midori"`].(*Float).f; x != 4.125 { 108 t.Errorf("colors.m[`green \"midori\"] = %v, want 3.14", x) 109 } 110 111 // colors.String() should be '{"red":3, "blue":4}', 112 // though the order of red and blue could vary. 113 s := colors.String() 114 var j interface{} 115 err := json.Unmarshal([]byte(s), &j) 116 if err != nil { 117 t.Errorf("colors.String() isn't valid JSON: %v", err) 118 } 119 m, ok := j.(map[string]interface{}) 120 if !ok { 121 t.Error("colors.String() didn't produce a map.") 122 } 123 red := m["red"] 124 x, ok := red.(float64) 125 if !ok { 126 t.Error("red.Kind() is not a number.") 127 } 128 if x != 3 { 129 t.Errorf("red = %v, want 3", x) 130 } 131 } 132 133 func TestFunc(t *testing.T) { 134 RemoveAll() 135 var x interface{} = []string{"a", "b"} 136 f := Func(func() interface{} { return x }) 137 if s, exp := f.String(), `["a","b"]`; s != exp { 138 t.Errorf(`f.String() = %q, want %q`, s, exp) 139 } 140 141 x = 17 142 if s, exp := f.String(), `17`; s != exp { 143 t.Errorf(`f.String() = %q, want %q`, s, exp) 144 } 145 } 146 147 func TestHandler(t *testing.T) { 148 RemoveAll() 149 m := NewMap("map1") 150 m.Add("a", 1) 151 m.Add("z", 2) 152 m2 := NewMap("map2") 153 for i := 0; i < 9; i++ { 154 m2.Add(strconv.Itoa(i), int64(i)) 155 } 156 rr := httptest.NewRecorder() 157 rr.Body = new(bytes.Buffer) 158 expvarHandler(rr, nil) 159 want := `{ 160 "map1": {"a": 1, "z": 2}, 161 "map2": {"0": 0, "1": 1, "2": 2, "3": 3, "4": 4, "5": 5, "6": 6, "7": 7, "8": 8} 162 } 163 ` 164 if got := rr.Body.String(); got != want { 165 t.Errorf("HTTP handler wrote:\n%s\nWant:\n%s", got, want) 166 } 167 }