util/clientmetric: add tests omitted from earlier commit

These were supposed to be part of
3b541c833e but I guess I forgot to "git
add" them. Whoops.

Updates #3307

Change-Id: I8c768a61ec7102a01799e81dc502a22399b9e9f0
Signed-off-by: Brad Fitzpatrick <bradfitz@tailscale.com>
This commit is contained in:
Brad Fitzpatrick 2021-11-22 21:47:30 -08:00
parent ad5e04249b
commit 0de1b74fbb

View File

@ -0,0 +1,75 @@
// Copyright (c) 2021 Tailscale Inc & AUTHORS All rights reserved.
// Use of this source code is governed by a BSD-style
// license that can be found in the LICENSE file.
package clientmetric
import (
"testing"
"time"
)
func TestDeltaEncBuf(t *testing.T) {
var enc deltaEncBuf
enc.writeName("one_one")
enc.writeValue(1, 1)
enc.writeName("two_zero")
enc.writeValue(2, 0)
enc.writeDelta(1, 63)
enc.writeDelta(2, 64)
enc.writeDelta(1, -65)
enc.writeDelta(2, -64)
got := enc.buf.String()
const want = "N0eone_oneS0202N10two_zeroS0400I027eI048001I028101I047f"
if got != want {
t.Errorf("error\n got %q\nwant %q\n", got, want)
}
}
func clearMetrics() {
mu.Lock()
defer mu.Unlock()
metrics = map[string]*Metric{}
numWireID = 0
lastDelta = time.Time{}
sorted = nil
lastLogVal = nil
unsorted = nil
}
func advanceTime() {
mu.Lock()
defer mu.Unlock()
lastDelta = time.Time{}
}
func TestEncodeLogTailMetricsDelta(t *testing.T) {
clearMetrics()
c1 := NewCounter("foo")
c2 := NewCounter("bar")
c1.Add(123)
if got, want := EncodeLogTailMetricsDelta(), "N06fooS02f601"; got != want {
t.Errorf("first = %q; want %q", got, want)
}
c2.Add(456)
advanceTime()
if got, want := EncodeLogTailMetricsDelta(), "N06barS049007"; got != want {
t.Errorf("second = %q; want %q", got, want)
}
advanceTime()
if got, want := EncodeLogTailMetricsDelta(), ""; got != want {
t.Errorf("with no changes = %q; want %q", got, want)
}
c1.Add(1)
c2.Add(2)
advanceTime()
if got, want := EncodeLogTailMetricsDelta(), "I0202I0404"; got != want {
t.Errorf("with increments = %q; want %q", got, want)
}
}