gitlab.com/aquachain/aquachain@v1.17.16-rc3.0.20221018032414-e3ddf1e1c055/common/metrics/json_test.go (about) 1 // Copyright 2018 The aquachain Authors 2 // This file is part of the aquachain library. 3 // 4 // The aquachain library is free software: you can redistribute it and/or modify 5 // it under the terms of the GNU Lesser General Public License as published by 6 // the Free Software Foundation, either version 3 of the License, or 7 // (at your option) any later version. 8 // 9 // The aquachain library is distributed in the hope that it will be useful, 10 // but WITHOUT ANY WARRANTY; without even the implied warranty of 11 // MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the 12 // GNU Lesser General Public License for more details. 13 // 14 // You should have received a copy of the GNU Lesser General Public License 15 // along with the aquachain library. If not, see <http://www.gnu.org/licenses/>. 16 17 package metrics 18 19 import ( 20 "bytes" 21 "encoding/json" 22 "testing" 23 ) 24 25 func TestRegistryMarshallJSON(t *testing.T) { 26 b := &bytes.Buffer{} 27 enc := json.NewEncoder(b) 28 r := NewRegistry() 29 r.Register("counter", NewCounter()) 30 enc.Encode(r) 31 if s := b.String(); "{\"counter\":{\"count\":0}}\n" != s { 32 t.Fatalf(s) 33 } 34 } 35 36 func TestRegistryWriteJSONOnce(t *testing.T) { 37 r := NewRegistry() 38 r.Register("counter", NewCounter()) 39 b := &bytes.Buffer{} 40 WriteJSONOnce(r, b) 41 if s := b.String(); s != "{\"counter\":{\"count\":0}}\n" { 42 t.Fail() 43 } 44 }