package main

import (
	"net/http"
	"net/http/httptest"
	"net/http/httputil"
	"testing"
)

func serveHTTP(w http.ResponseWriter, r *http.Request) {
	w.Header().Add("Content-Length", "2")
	switch r.Method {
	case http.MethodHead:
	default:
		w.Write([]byte("{}"))
	}
}

func TestDumpResponse(t *testing.T) {
	t.Run("Get", func(t *testing.T) {
		rw := httptest.NewRecorder()
		r := httptest.NewRequest(http.MethodGet, "/", nil)

		serveHTTP(rw, r)
		raw, err := httputil.DumpResponse(rw.Result(), true)
		if err != nil {
			t.Errorf("DumpResponse: (%v)\nRaw: %s", err, string(raw))
		}

	})
	t.Run("Head", func(t *testing.T) {
		rw := httptest.NewRecorder()
		r := httptest.NewRequest(http.MethodHead, "/", nil)

		serveHTTP(rw, r)
		raw, err := httputil.DumpResponse(rw.Result(), true)
		if err != nil {
			t.Errorf("DumpResponse: (%v)\nRaw: %s", err, string(raw))
		}

	})
}