1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
|
// Copyright 2017 The Go 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 pprof
import (
"context"
"reflect"
"sort"
"testing"
)
func labelsSorted(ctx context.Context) []label {
ls := []label{}
ForLabels(ctx, func(key, value string) bool {
ls = append(ls, label{key, value})
return true
})
sort.Sort(labelSorter(ls))
return ls
}
type labelSorter []label
func (s labelSorter) Len() int { return len(s) }
func (s labelSorter) Swap(i, j int) { s[i], s[j] = s[j], s[i] }
func (s labelSorter) Less(i, j int) bool { return s[i].key < s[j].key }
func TestContextLabels(t *testing.T) {
// Background context starts with no labels.
ctx := context.Background()
labels := labelsSorted(ctx)
if len(labels) != 0 {
t.Errorf("labels on background context: want [], got %v ", labels)
}
// Add a single label.
ctx = WithLabels(ctx, Labels("key", "value"))
// Retrieve it with Label.
v, ok := Label(ctx, "key")
if !ok || v != "value" {
t.Errorf(`Label(ctx, "key"): got %v, %v; want "value", ok`, v, ok)
}
gotLabels := labelsSorted(ctx)
wantLabels := []label{{"key", "value"}}
if !reflect.DeepEqual(gotLabels, wantLabels) {
t.Errorf("(sorted) labels on context: got %v, want %v", gotLabels, wantLabels)
}
// Add a label with a different key.
ctx = WithLabels(ctx, Labels("key2", "value2"))
v, ok = Label(ctx, "key2")
if !ok || v != "value2" {
t.Errorf(`Label(ctx, "key2"): got %v, %v; want "value2", ok`, v, ok)
}
gotLabels = labelsSorted(ctx)
wantLabels = []label{{"key", "value"}, {"key2", "value2"}}
if !reflect.DeepEqual(gotLabels, wantLabels) {
t.Errorf("(sorted) labels on context: got %v, want %v", gotLabels, wantLabels)
}
// Add label with first key to test label replacement.
ctx = WithLabels(ctx, Labels("key", "value3"))
v, ok = Label(ctx, "key")
if !ok || v != "value3" {
t.Errorf(`Label(ctx, "key3"): got %v, %v; want "value3", ok`, v, ok)
}
gotLabels = labelsSorted(ctx)
wantLabels = []label{{"key", "value3"}, {"key2", "value2"}}
if !reflect.DeepEqual(gotLabels, wantLabels) {
t.Errorf("(sorted) labels on context: got %v, want %v", gotLabels, wantLabels)
}
// Labels called with two labels with the same key should pick the second.
ctx = WithLabels(ctx, Labels("key4", "value4a", "key4", "value4b"))
v, ok = Label(ctx, "key4")
if !ok || v != "value4b" {
t.Errorf(`Label(ctx, "key4"): got %v, %v; want "value4b", ok`, v, ok)
}
gotLabels = labelsSorted(ctx)
wantLabels = []label{{"key", "value3"}, {"key2", "value2"}, {"key4", "value4b"}}
if !reflect.DeepEqual(gotLabels, wantLabels) {
t.Errorf("(sorted) labels on context: got %v, want %v", gotLabels, wantLabels)
}
}
func TestLabelMapStringer(t *testing.T) {
for _, tbl := range []struct {
m labelMap
expected string
}{
{
m: labelMap{
// empty map
},
expected: "{}",
}, {
m: labelMap{
"foo": "bar",
},
expected: `{"foo":"bar"}`,
}, {
m: labelMap{
"foo": "bar",
"key1": "value1",
"key2": "value2",
"key3": "value3",
"key4WithNewline": "\nvalue4",
},
expected: `{"foo":"bar", "key1":"value1", "key2":"value2", "key3":"value3", "key4WithNewline":"\nvalue4"}`,
},
} {
if got := tbl.m.String(); tbl.expected != got {
t.Errorf("%#v.String() = %q; want %q", tbl.m, got, tbl.expected)
}
}
}
|
