Function TestClientTimeout

proxy/clienttimeout_test.go:14–64

(t *testing.T)

)

func TestClientTimeout(t *testing.T) {
	testLog := NewTestLog()
	defer testLog.Close()

	d := 200 * time.Millisecond

	payload := []byte("backend reply")
	backend := startTestServer(payload, 0, func(r *http.Request) {
		time.Sleep(2 * d)
	})

	defer backend.Close()

	doc := fmt.Sprintf(`hello: * -> "%s"`, backend.URL)
	tp, err := newTestProxy(doc, FlagsNone)
	if err != nil {
		t.Error()
		return
	}

	ps := httptest.NewServer(tp.proxy)
	defer func() {
		ps.Close()
		tp.close()
	}()

	req, err := http.NewRequest("GET", ps.URL, nil)
	if err != nil {
		t.Errorf("Failed to create request: %v", err)
	}

	rsp, err := (&http.Client{
		Timeout: d,
	}).Do(req)

	if err == nil {
		t.Error("err should not be nil")
	}
	if rsp != nil {
		t.Error("response should be nil")
	}

	const msgErrClientTimeout = "context canceled"
	if err = testLog.WaitFor(msgErrClientTimeout, 3*d); err != nil {
		t.Errorf("log should contain '%s'", msgErrClientTimeout)
	}
	const msgErrClientCanceledAfter = "client canceled after"
	if err = testLog.WaitFor(msgErrClientCanceledAfter, 3*d); err != nil {
		t.Errorf("log should contain '%s'", msgErrClientCanceledAfter)
	}
}

func TestClientCancellation(t *testing.T) {
	testLog := NewTestLog()