TUN-528: Move cloudflared into a separate repo

vendor/github.com/opentracing/opentracing-go/propagation_test.go

@@ -0,0 +1,93 @@
+package opentracing
+
+import (
+	"net/http"
+	"strconv"
+	"testing"
+)
+
+const testHeaderPrefix = "testprefix-"
+
+func TestTextMapCarrierInject(t *testing.T) {
+	m := make(map[string]string)
+	m["NotOT"] = "blah"
+	m["opname"] = "AlsoNotOT"
+	tracer := testTracer{}
+	span := tracer.StartSpan("someSpan")
+	fakeID := span.Context().(testSpanContext).FakeID
+
+	carrier := TextMapCarrier(m)
+	if err := span.Tracer().Inject(span.Context(), TextMap, carrier); err != nil {
+		t.Fatal(err)
+	}
+
+	if len(m) != 3 {
+		t.Errorf("Unexpected header length: %v", len(m))
+	}
+	// The prefix comes from just above; the suffix comes from
+	// testTracer.Inject().
+	if m["testprefix-fakeid"] != strconv.Itoa(fakeID) {
+		t.Errorf("Could not find fakeid at expected key")
+	}
+}
+
+func TestTextMapCarrierExtract(t *testing.T) {
+	m := make(map[string]string)
+	m["NotOT"] = "blah"
+	m["opname"] = "AlsoNotOT"
+	m["testprefix-fakeid"] = "42"
+	tracer := testTracer{}
+
+	carrier := TextMapCarrier(m)
+	extractedContext, err := tracer.Extract(TextMap, carrier)
+	if err != nil {
+		t.Fatal(err)
+	}
+
+	if extractedContext.(testSpanContext).FakeID != 42 {
+		t.Errorf("Failed to read testprefix-fakeid correctly")
+	}
+}
+
+func TestHTTPHeaderInject(t *testing.T) {
+	h := http.Header{}
+	h.Add("NotOT", "blah")
+	h.Add("opname", "AlsoNotOT")
+	tracer := testTracer{}
+	span := tracer.StartSpan("someSpan")
+	fakeID := span.Context().(testSpanContext).FakeID
+
+	// Use HTTPHeadersCarrier to wrap around `h`.
+	carrier := HTTPHeadersCarrier(h)
+	if err := span.Tracer().Inject(span.Context(), HTTPHeaders, carrier); err != nil {
+		t.Fatal(err)
+	}
+
+	if len(h) != 3 {
+		t.Errorf("Unexpected header length: %v", len(h))
+	}
+	// The prefix comes from just above; the suffix comes from
+	// testTracer.Inject().
+	if h.Get("testprefix-fakeid") != strconv.Itoa(fakeID) {
+		t.Errorf("Could not find fakeid at expected key")
+	}
+}
+
+func TestHTTPHeaderExtract(t *testing.T) {
+	h := http.Header{}
+	h.Add("NotOT", "blah")
+	h.Add("opname", "AlsoNotOT")
+	h.Add("testprefix-fakeid", "42")
+	tracer := testTracer{}
+
+	// Use HTTPHeadersCarrier to wrap around `h`.
+	carrier := HTTPHeadersCarrier(h)
+	spanContext, err := tracer.Extract(HTTPHeaders, carrier)
+	if err != nil {
+		t.Fatal(err)
+	}
+
+	if spanContext.(testSpanContext).FakeID != 42 {
+		t.Errorf("Failed to read testprefix-fakeid correctly")
+	}
+}