Ejemplo n.º 1
0
func TestCatena(t *testing.T) {
	c0 := catena.New(handlerWrapper0)
	c1 := c0.Append(handlerWrapper1, handlerWrapperA)
	cBefore0 := catena.New(handlerWrapper1)
	c0 = cBefore0.Merge(c0)
	m := http.NewServeMux()
	r0 := "/0"
	r1 := "/1"
	m.Handle(r0, c0.EndFn(endHandler))
	m.Handle(r1, c1.EndFn(endHandler))
	s := httptest.NewServer(m)

	tMap := map[string]string{
		"/0": bTxt1 + bTxt0 + bTxtEnd + bTxt0 + bTxt1,
		"/1": bTxt0 + bTxt1 + bTxtA + bTxtEnd + bTxtA + bTxt1 + bTxt0,
	}

	for k, v := range tMap {
		rb, err := getReqBody(s.URL + k)
		if err != nil {
			t.Error(err)
		}
		want := v
		got := rb
		if got != want {
			t.Errorf("Body = %v, want %v", got, want)
		}
	}
}
Ejemplo n.º 2
0
func TestNilEnd(t *testing.T) {
	c0 := catena.New(emptyHandlerWrapper)
	m := http.NewServeMux()
	r0 := "/0"
	r1 := "/1"
	m.Handle(r0, c0.End(nil))
	m.Handle(r1, c0.EndFn(nil))
	s := httptest.NewServer(m)

	tMap := map[string]int{
		"/0": http.StatusOK,
		"/1": http.StatusOK,
	}

	for k, v := range tMap {
		rs, err := getReqStatus(s.URL + k)
		if err != nil {
			t.Error(err)
		}
		want := v
		got := rs
		if got != want {
			t.Errorf("Status Code = %v, want %v", got, want)
		}
	}
}
Ejemplo n.º 3
0
func Example() {
	// Each wrapper writes either "0", "1", or "A" to the response body before
	// and after ServeHTTP() is called.
	// endHandler writes "_END_" to the response body and returns.
	cat00 := catena.New(handlerWrapper0, handlerWrapper0)
	cat00A1 := cat00.Append(handlerWrapperA, handlerWrapper1)

	cat100A1 := catena.New(handlerWrapper1)
	cat100A1 = cat100A1.Merge(cat00A1)

	mux := http.NewServeMux()
	mux.Handle("/path_implies_body/00_End", cat00.EndFn(endHandler))
	mux.Handle("/path_implies_body/00A1_End", cat00A1.EndFn(endHandler))
	mux.Handle("/path_implies_body/100A1_End", cat100A1.EndFn(endHandler))

	server := httptest.NewServer(mux)

	rBody0, err := getReqBody(server.URL + "/path_implies_body/00_End")
	if err != nil {
		fmt.Println(err)
	}
	rBody1, err := getReqBody(server.URL + "/path_implies_body/00A1_End")
	if err != nil {
		fmt.Println(err)
	}
	rBody2, err := getReqBody(server.URL + "/path_implies_body/100A1_End")
	if err != nil {
		fmt.Println(err)
	}

	fmt.Println("Catena 0 Body:", rBody0)
	fmt.Println("Catena 1 Body:", rBody1)
	fmt.Println("Catena 2 Body:", rBody2)

	// Output:
	// Catena 0 Body: 00_END_00
	// Catena 1 Body: 00A1_END_1A00
	// Catena 2 Body: 100A1_END_1A001
}
Ejemplo n.º 4
0
func BenchmarkCatena10(b *testing.B) {
	c0 := catena.New(emptyHandlerWrapper,
		emptyHandlerWrapper, emptyHandlerWrapper, emptyHandlerWrapper,
		emptyHandlerWrapper, emptyHandlerWrapper, emptyHandlerWrapper,
		emptyHandlerWrapper, emptyHandlerWrapper, emptyHandlerWrapper)
	m := http.NewServeMux()
	m.Handle("/", c0.EndFn(nilHandler))
	s := httptest.NewServer(m)

	b.ResetTimer()

	for n := 0; n < b.N; n++ {
		re0, err := http.Get(s.URL + "/")
		if err != nil {
			b.Error(err)
		}
		_ = re0.Body.Close()
	}
}