Function TestTracingProxySpanWithLoopbackIfStatusFilter

proxy/tracing_test.go:550–619

(t *testing.T)

}

func TestTracingProxySpanWithLoopbackIfStatusFilter(t *testing.T) {
	// b: backend server that returns 404 to trigger loopbackIfStatus
	// fb: fallback backend server for loopback request
	// We cannot use shunt backends instead because in that case the proxy span will be nil
	b := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
		w.WriteHeader(http.StatusNotFound)
		w.Write([]byte("Hello! I'm main backend!"))
	}))
	defer b.Close()
	fb := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
		w.WriteHeader(http.StatusOK)
		w.Write([]byte("Hello! I'm fallback backend!"))
	}))
	defer fb.Close()

	doc := fmt.Sprintf(`
main: Path("/main") -> loopbackIfStatus(404, "/fallback") -> "%s";
fallback: Path("/fallback") -> "%s";
`, b.URL, fb.URL)

	tracer := tracingtest.NewTracer()
	params := Params{
		OpenTracing: &OpenTracingParams{
			Tracer: tracer,
		},
		Flags: FlagsNone,
	}

	t.Setenv("HOSTNAME", "shadow-tag.tracing.test")

	tp, err := newTestProxyWithParams(doc, params)
	if err != nil {
		t.Fatal(err)
	}
	defer tp.close()

	ps := httptest.NewServer(tp.proxy)
	defer ps.Close()

	req, err := http.NewRequest("GET", ps.URL+"/main", nil)
	if err != nil {
		t.Fatal(err)
	}
	req.Header.Set("X-Flow-Id", "test-flow-id-fallback")

	rsp, err := ps.Client().Do(req)
	if err != nil {
		t.Fatal(err)
	}
	defer rsp.Body.Close()
	io.Copy(io.Discard, rsp.Body)
	t.Logf("got response %d", rsp.StatusCode)

	pxSpanCount := 0
	for _, s := range tracer.FinishedSpans() {
		if s.OperationName == "proxy" {
			pxSpanCount++
			verifyNoTag(t, s, "shadow")