Function TestReadTimeoutSocketClosed

proxy/clienttimeout_test.go:109–182

(t *testing.T)

}

func TestReadTimeoutSocketClosed(t *testing.T) {
	backend := startTestServer([]byte("backend reply"), 0, func(*http.Request) {})
	defer backend.Close()

	doc := fmt.Sprintf(`
		r1: Path("/timeout") -> readTimeout("10ms") -> status(201) -> "%s";
		r2: Path("/ok") -> status(200) -> inlineContent("OK") -> <shunt>
	`, backend.URL)

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

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

	addr := ps.Listener.Addr().String()

	readResp := func(conn net.Conn) *http.Response {
		t.Helper()
		resp, err := http.ReadResponse(bufio.NewReader(conn), nil)
		if err != nil {
			t.Fatalf("Failed to read response: %v", err)
		}

		_, err = io.ReadAll(resp.Body)
		if err != nil {
			t.Fatalf("Failed to read response body: %v", err)
		}
		resp.Body.Close()

		return resp
	}

	// Verify both routes return their configured status codes (normal client).
	for _, tc := range []struct {
		path string
		want int
	}{
		{"/timeout", http.StatusCreated},
		{"/ok", http.StatusOK},
	} {
		resp, err := ps.Client().Get("http://" + addr + tc.path)
		if err != nil {
			t.Fatalf("GET %s: %v", tc.path, err)
		}
		resp.Body.Close()
		if resp.StatusCode != tc.want {
			t.Errorf("GET %s: got %d, want %d", tc.path, resp.StatusCode, tc.want)
		}
	}

	postConn, err := net.Dial("tcp", addr)
	if err != nil {
		t.Fatalf("dial post conn: %v", err)
	}