boulder/web/context_test.go

package web

import (
	"bytes"
	"context"
	"crypto/tls"
	"fmt"
	"net/http"
	"net/http/httptest"
	"strings"
	"testing"
	"time"

	"github.com/letsencrypt/boulder/features"
	blog "github.com/letsencrypt/boulder/log"
	"github.com/letsencrypt/boulder/test"
)

type myHandler struct{}

func (m myHandler) ServeHTTP(e *RequestEvent, w http.ResponseWriter, r *http.Request) {
	w.WriteHeader(201)
	e.Endpoint = "/endpoint"
	_, _ = w.Write([]byte("hi"))
}

func TestLogCode(t *testing.T) {
	mockLog := blog.UseMock()
	th := NewTopHandler(mockLog, myHandler{})
	req, err := http.NewRequest("GET", "/thisisignored", &bytes.Reader{})
	if err != nil {
		t.Fatal(err)
	}
	th.ServeHTTP(httptest.NewRecorder(), req)
	expected := `INFO: GET /endpoint 0 201 0 0.0.0.0 JSON={}`
	if len(mockLog.GetAllMatching(expected)) != 1 {
		t.Errorf("Expected exactly one log line matching %q. Got \n%s",
			expected, strings.Join(mockLog.GetAllMatching(".*"), "\n"))
	}
}

type codeHandler struct{}

func (ch codeHandler) ServeHTTP(e *RequestEvent, w http.ResponseWriter, r *http.Request) {
	e.Endpoint = "/endpoint"
	_, _ = w.Write([]byte("hi"))
}

func TestStatusCodeLogging(t *testing.T) {
	mockLog := blog.UseMock()
	th := NewTopHandler(mockLog, codeHandler{})
	req, err := http.NewRequest("GET", "/thisisignored", &bytes.Reader{})
	if err != nil {
		t.Fatal(err)
	}
	th.ServeHTTP(httptest.NewRecorder(), req)
	expected := `INFO: GET /endpoint 0 200 0 0.0.0.0 JSON={}`
	if len(mockLog.GetAllMatching(expected)) != 1 {
		t.Errorf("Expected exactly one log line matching %q. Got \n%s",
			expected, strings.Join(mockLog.GetAllMatching(".*"), "\n"))
	}
}

func TestOrigin(t *testing.T) {
	mockLog := blog.UseMock()
	th := NewTopHandler(mockLog, myHandler{})
	req, err := http.NewRequest("GET", "/thisisignored", &bytes.Reader{})
	if err != nil {
		t.Fatal(err)
	}
	req.Header.Add("Origin", "https://example.com")
	th.ServeHTTP(httptest.NewRecorder(), req)
	expected := `INFO: GET /endpoint 0 201 0 0.0.0.0 JSON={.*"Origin":"https://example.com"}`
	if len(mockLog.GetAllMatching(expected)) != 1 {
		t.Errorf("Expected exactly one log line matching %q. Got \n%s",
			expected, strings.Join(mockLog.GetAllMatching(".*"), "\n"))
	}
}

type hostHeaderHandler struct {
	f func(*RequestEvent, http.ResponseWriter, *http.Request)
}

func (hhh hostHeaderHandler) ServeHTTP(e *RequestEvent, w http.ResponseWriter, r *http.Request) {
	hhh.f(e, w, r)
}

func TestHostHeaderRewrite(t *testing.T) {
	mockLog := blog.UseMock()
	hhh := hostHeaderHandler{f: func(_ *RequestEvent, _ http.ResponseWriter, r *http.Request) {
		t.Helper()
		test.AssertEquals(t, r.Host, "localhost")
	}}
	th := NewTopHandler(mockLog, &hhh)

	req, err := http.NewRequest("GET", "/", &bytes.Reader{})
	test.AssertNotError(t, err, "http.NewRequest failed")
	req.Host = "localhost:80"
	fmt.Println("here")
	th.ServeHTTP(httptest.NewRecorder(), req)

	req, err = http.NewRequest("GET", "/", &bytes.Reader{})
	test.AssertNotError(t, err, "http.NewRequest failed")
	req.Host = "localhost:443"
	req.TLS = &tls.ConnectionState{}
	th.ServeHTTP(httptest.NewRecorder(), req)

	req, err = http.NewRequest("GET", "/", &bytes.Reader{})
	test.AssertNotError(t, err, "http.NewRequest failed")
	req.Host = "localhost:443"
	req.TLS = nil
	th.ServeHTTP(httptest.NewRecorder(), req)

	hhh.f = func(_ *RequestEvent, _ http.ResponseWriter, r *http.Request) {
		t.Helper()
		test.AssertEquals(t, r.Host, "localhost:123")
	}
	req, err = http.NewRequest("GET", "/", &bytes.Reader{})
	test.AssertNotError(t, err, "http.NewRequest failed")
	req.Host = "localhost:123"
	th.ServeHTTP(httptest.NewRecorder(), req)
}

type cancelHandler struct {
	res chan string
}

func (ch cancelHandler) ServeHTTP(e *RequestEvent, w http.ResponseWriter, r *http.Request) {
	select {
	case <-r.Context().Done():
		ch.res <- r.Context().Err().Error()
	case <-time.After(300 * time.Millisecond):
		ch.res <- "300 ms passed"
	}
}

func TestPropagateCancel(t *testing.T) {
	mockLog := blog.UseMock()
	res := make(chan string)
	features.Set(features.Config{PropagateCancels: true})
	th := NewTopHandler(mockLog, cancelHandler{res})
	ctx, cancel := context.WithCancel(context.Background())
	go func() {
		req, err := http.NewRequestWithContext(ctx, "GET", "/thisisignored", &bytes.Reader{})
		if err != nil {
			t.Error(err)
		}
		th.ServeHTTP(httptest.NewRecorder(), req)
	}()
	cancel()
	result := <-res
	if result != "context canceled" {
		t.Errorf("expected 'context canceled', got %q", result)
	}
}