Retro68/gcc/libgo/go/net/http/httptrace/trace_test.go
2017-10-07 02:16:47 +02:00

90 lines
1.8 KiB
Go

// 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)
}
}
}