error-pages/internal/http/handlers/version/handler_test.go

package version_test

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

	"github.com/stretchr/testify/assert"

	"gh.tarampamp.am/error-pages/internal/http/handlers/version"
)

func TestServeHTTP(t *testing.T) {
	t.Parallel()

	var handler = version.New("\t\n foo@bar ")

	t.Run("get", func(t *testing.T) {
		var (
			req = httptest.NewRequest(http.MethodGet, "http://testing", http.NoBody)
			rr  = httptest.NewRecorder()
		)

		handler.ServeHTTP(rr, req)

		assert.Equal(t, rr.Header().Get("Content-Type"), "application/json; charset=utf-8")
		assert.Equal(t, rr.Code, http.StatusOK)
		assert.Equal(t, rr.Body.String(), `{"version":"foo@bar"}`)
	})

	t.Run("head", func(t *testing.T) {
		var (
			req = httptest.NewRequest(http.MethodHead, "http://testing", http.NoBody)
			rr  = httptest.NewRecorder()
		)

		handler.ServeHTTP(rr, req)

		assert.Equal(t, rr.Code, http.StatusOK)
		assert.Empty(t, rr.Header().Get("Content-Type"))
		assert.Empty(t, rr.Body.Bytes())
	})

	t.Run("method not allowed", func(t *testing.T) {
		for _, method := range []string{
			http.MethodDelete,
			http.MethodPatch,
			http.MethodPost,
			http.MethodPut,
		} {
			var (
				req = httptest.NewRequest(method, "http://testing", http.NoBody)
				rr  = httptest.NewRecorder()
			)

			handler.ServeHTTP(rr, req)

			assert.Equal(t, rr.Header().Get("Content-Type"), "text/plain; charset=utf-8")
			assert.Equal(t, rr.Code, http.StatusMethodNotAllowed)
			assert.Equal(t, "Method Not Allowed\n", rr.Body.String())
		}
	})
}