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
|
// Copyright 2016 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 httptrace
import (
"bytes"
"context"
"testing"
)
func TestWithClientTrace(t *testing.T) {
var buf bytes.Buffer
connectStart := func(b byte) func(network, addr string) {
return func(network, addr string) {
buf.WriteByte(b)
}
}
ctx := context.Background()
oldtrace := &ClientTrace{
ConnectStart: connectStart('O'),
}
ctx = WithClientTrace(ctx, oldtrace)
newtrace := &ClientTrace{
ConnectStart: connectStart('N'),
}
ctx = WithClientTrace(ctx, newtrace)
trace := ContextClientTrace(ctx)
buf.Reset()
trace.ConnectStart("net", "addr")
if got, want := buf.String(), "NO"; got != want {
t.Errorf("got %q; want %q", got, want)
}
}
func TestCompose(t *testing.T) {
var buf bytes.Buffer
var testNum int
connectStart := func(b byte) func(network, addr string) {
return func(network, addr string) {
if addr != "addr" {
t.Errorf(`%d. args for %q case = %q, %q; want addr of "addr"`, testNum, b, network, addr)
}
buf.WriteByte(b)
}
}
tests := [...]struct {
trace, old *ClientTrace
want string
}{
0: {
want: "T",
trace: &ClientTrace{
ConnectStart: connectStart('T'),
},
},
1: {
want: "TO",
trace: &ClientTrace{
ConnectStart: connectStart('T'),
},
old: &ClientTrace{ConnectStart: connectStart('O')},
},
2: {
want: "O",
trace: &ClientTrace{},
old: &ClientTrace{ConnectStart: connectStart('O')},
},
}
for i, tt := range tests {
testNum = i
buf.Reset()
tr := *tt.trace
tr.compose(tt.old)
if tr.ConnectStart != nil {
tr.ConnectStart("net", "addr")
}
if got := buf.String(); got != tt.want {
t.Errorf("%d. got = %q; want %q", i, got, tt.want)
}
}
}
|
