mirror of https://github.com/knative/pkg.git
172 lines
4.1 KiB
Go
172 lines
4.1 KiB
Go
/*
|
|
Copyright 2019 The Knative Authors
|
|
|
|
Licensed under the Apache License, Version 2.0 (the "License");
|
|
you may not use this file except in compliance with the License.
|
|
You may obtain a copy of the License at
|
|
|
|
http://www.apache.org/licenses/LICENSE-2.0
|
|
|
|
Unless required by applicable law or agreed to in writing, software
|
|
distributed under the License is distributed on an "AS IS" BASIS,
|
|
WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
See the License for the specific language governing permissions and
|
|
limitations under the License.
|
|
*/
|
|
|
|
package tracing_test
|
|
|
|
import (
|
|
"net/http"
|
|
"net/url"
|
|
"testing"
|
|
|
|
"github.com/google/go-cmp/cmp"
|
|
. "knative.dev/pkg/tracing"
|
|
"knative.dev/pkg/tracing/config"
|
|
. "knative.dev/pkg/tracing/testing"
|
|
)
|
|
|
|
type fakeWriter struct {
|
|
lastWrite *[]byte
|
|
}
|
|
|
|
func (fw fakeWriter) Header() http.Header {
|
|
return http.Header{}
|
|
}
|
|
|
|
func (fw fakeWriter) Write(data []byte) (int, error) {
|
|
*fw.lastWrite = data
|
|
return len(data), nil
|
|
}
|
|
|
|
func (fw fakeWriter) WriteHeader(statusCode int) {
|
|
}
|
|
|
|
type testHandler struct {
|
|
}
|
|
|
|
func (th *testHandler) ServeHTTP(w http.ResponseWriter, r *http.Request) {
|
|
w.Write([]byte("fake"))
|
|
}
|
|
|
|
func TestHTTPSpanMiddleware(t *testing.T) {
|
|
cfg := config.Config{
|
|
Backend: config.Zipkin,
|
|
Debug: true,
|
|
}
|
|
|
|
// Create tracer with reporter recorder
|
|
reporter, co := FakeZipkinExporter()
|
|
defer reporter.Close()
|
|
oct := NewOpenCensusTracer(co)
|
|
defer oct.Finish()
|
|
|
|
if err := oct.ApplyConfig(&cfg); err != nil {
|
|
t.Errorf("Failed to apply tracer config: %v", err)
|
|
}
|
|
|
|
next := testHandler{}
|
|
middleware := HTTPSpanMiddleware(&next)
|
|
|
|
var lastWrite []byte
|
|
fw := fakeWriter{lastWrite: &lastWrite}
|
|
|
|
req, err := http.NewRequest("GET", "http://test.example.com", nil)
|
|
if err != nil {
|
|
t.Errorf("Failed to make fake request: %v", err)
|
|
}
|
|
traceID := "821e0d50d931235a5ba3fa42eddddd8f"
|
|
req.Header["X-B3-Traceid"] = []string{traceID}
|
|
req.Header["X-B3-Spanid"] = []string{"b3bd5e1c4318c78a"}
|
|
|
|
middleware.ServeHTTP(fw, req)
|
|
|
|
// Assert our next handler was called
|
|
if diff := cmp.Diff([]byte("fake"), lastWrite); diff != "" {
|
|
t.Errorf("Got http response (-want, +got) = %v", diff)
|
|
}
|
|
|
|
spans := reporter.Flush()
|
|
if len(spans) != 1 {
|
|
t.Errorf("Got %d spans, expected 1: spans = %v", len(spans), spans)
|
|
}
|
|
if got := spans[0].TraceID.String(); got != traceID {
|
|
t.Errorf("spans[0].TraceID = %s, want %s", got, traceID)
|
|
}
|
|
}
|
|
|
|
func TestHTTPSpanIgnoringPaths(t *testing.T) {
|
|
cfg := config.Config{
|
|
Backend: config.Zipkin,
|
|
Debug: true,
|
|
}
|
|
|
|
// Create tracer with reporter recorder
|
|
reporter, co := FakeZipkinExporter()
|
|
defer reporter.Close()
|
|
oct := NewOpenCensusTracer(co)
|
|
defer oct.Finish()
|
|
|
|
if err := oct.ApplyConfig(&cfg); err != nil {
|
|
t.Errorf("Failed to apply tracer config: %v", err)
|
|
}
|
|
|
|
paths := []string{"/readyz"}
|
|
middleware := HTTPSpanIgnoringPaths(paths...)(&testHandler{})
|
|
|
|
testCases := map[string]struct {
|
|
path string
|
|
traced bool
|
|
}{
|
|
"traced": {
|
|
path: "/",
|
|
traced: true,
|
|
},
|
|
"ignored": {
|
|
path: paths[0],
|
|
traced: false,
|
|
},
|
|
}
|
|
for n, tc := range testCases {
|
|
t.Run(n, func(t *testing.T) {
|
|
var lastWrite []byte
|
|
fw := fakeWriter{lastWrite: &lastWrite}
|
|
|
|
u := &url.URL{
|
|
Scheme: "http",
|
|
Host: "test.example.com",
|
|
Path: tc.path,
|
|
}
|
|
req, err := http.NewRequest("GET", u.String(), nil)
|
|
if err != nil {
|
|
t.Errorf("Failed to make fake request: %v", err)
|
|
}
|
|
traceID := "821e0d50d931235a5ba3fa42eddddd8f"
|
|
req.Header.Set("X-B3-Traceid", traceID)
|
|
req.Header.Set("X-B3-Spanid", "b3bd5e1c4318c78a")
|
|
|
|
middleware.ServeHTTP(fw, req)
|
|
|
|
// Assert our next handler was called
|
|
if diff := cmp.Diff([]byte("fake"), lastWrite); diff != "" {
|
|
t.Errorf("Got http response (-want, +got) = %v", diff)
|
|
}
|
|
|
|
spans := reporter.Flush()
|
|
if tc.traced {
|
|
if len(spans) != 1 {
|
|
t.Errorf("Got %d spans, expected 1: spans = %v", len(spans), spans)
|
|
}
|
|
if got := spans[0].TraceID.String(); got != traceID {
|
|
t.Errorf("spans[0].TraceID = %s, want %s", got, traceID)
|
|
}
|
|
} else {
|
|
if len(spans) != 0 {
|
|
t.Errorf("Got %d spans, expected 0: spans = %v", len(spans), spans)
|
|
}
|
|
}
|
|
})
|
|
}
|
|
}
|