Function TestShadowSingle

proxy/shadow_test.go:27–105

(t *testing.T)

)

func TestShadowSingle(t *testing.T) {
	s := stdlibhttptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
		w.Header().Add("Backend", "main")
		w.WriteHeader(http.StatusOK)
		w.Write([]byte("OK"))
	}))
	defer s.Close()

	shadow := stdlibhttptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
		w.Header().Add("Backend", "shadow")
		w.WriteHeader(http.StatusOK)
		w.Write([]byte("OK"))
	}))
	defer shadow.Close()

	metrics := &metricstest.MockMetrics{}
	reg := sched.RegistryWith(sched.Options{
		Metrics:                metrics,
		EnableRouteFIFOMetrics: true,
	})
	defer reg.Close()

	fr := make(filters.Registry)
	fr.Register(tee.NewTeeLoopback())
	fr.Register(scheduler.NewFifo())

	doc := fmt.Sprintf(`
main: PathSubtree("/")
  -> "%s";
split: PathSubtree("/") && Traffic(0.5)
  -> teeLoopback("test")
  -> "%s";
shadow: PathSubtree("/") && Tee("test") && True()
  -> "%s";
	`, s.URL, s.URL, shadow.URL)

	dc, err := testdataclient.NewDoc(doc)
	if err != nil {
		t.Fatalf("Failed to create testdataclient: %v", err)
	}
	defer dc.Close()

	ro := routing.Options{
		SignalFirstLoad: true,
		FilterRegistry:  fr,
		DataClients:     []routing.DataClient{dc},
		PostProcessors:  []routing.PostProcessor{reg},
		Predicates: []routing.PredicateSpec{
			teepred.New(),
			traffic.New(),
			primitive.NewTrue(),
		},
	}
	rt := routing.New(ro)
	defer rt.Close()
	<-rt.FirstLoad()

	pr := WithParams(Params{