errors_test.go

package main

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

	"github.com/stretchr/testify/assert"
	"go.goblog.app/app/pkgs/contenttype"
)

func Test_errors(t *testing.T) {
	app := &goBlog{
		cfg: createDefaultTestConfig(t),
	}

	_ = app.initConfig(false)
	app.initMarkdown()
	app.initSessions()

	t.Run("Test 404, no HTML", func(t *testing.T) {
		h := http.HandlerFunc(app.serve404)

		req := httptest.NewRequest(http.MethodGet, "/abc", nil)
		req.Header.Set("Accept", contenttype.JSON)

		rec := httptest.NewRecorder()

		h(rec, req)

		res := rec.Result()
		resBody, _ := io.ReadAll(res.Body)
		_ = res.Body.Close()
		resString := string(resBody)

		assert.Equal(t, http.StatusNotFound, res.StatusCode)
		assert.Contains(t, resString, "not found")
		assert.Contains(t, res.Header.Get("Content-Type"), "text/plain")
	})

	t.Run("Test 404, HTML", func(t *testing.T) {
		h := http.HandlerFunc(app.serve404)

		req := httptest.NewRequest(http.MethodGet, "/abc", nil)
		req.Header.Set("Accept", contenttype.HTML)

		rec := httptest.NewRecorder()

		h(rec, req)

		res := rec.Result()
		resBody, _ := io.ReadAll(res.Body)
		_ = res.Body.Close()
		resString := string(resBody)

		assert.Equal(t, http.StatusNotFound, res.StatusCode)
		assert.Contains(t, resString, "not found")
		assert.Contains(t, res.Header.Get("Content-Type"), contenttype.HTML)
	})

	t.Run("Test Method Not Allowed, no HTML", func(t *testing.T) {
		h := http.HandlerFunc(app.serveNotAllowed)

		req := httptest.NewRequest(http.MethodGet, "/abc", nil)
		req.Header.Set("Accept", contenttype.JSON)

		rec := httptest.NewRecorder()

		h(rec, req)

		res := rec.Result()
		resBody, _ := io.ReadAll(res.Body)
		_ = res.Body.Close()
		resString := string(resBody)

		assert.Equal(t, http.StatusMethodNotAllowed, res.StatusCode)
		assert.Contains(t, resString, "Method Not Allowed")
		assert.Contains(t, res.Header.Get("Content-Type"), "text/plain")
	})

	t.Run("Test Method Not Allowed", func(t *testing.T) {
		h := http.HandlerFunc(app.serveNotAllowed)

		req := httptest.NewRequest(http.MethodGet, "/abc", nil)
		req.Header.Set("Accept", contenttype.HTML)

		rec := httptest.NewRecorder()

		h(rec, req)

		res := rec.Result()
		resBody, _ := io.ReadAll(res.Body)
		_ = res.Body.Close()
		resString := string(resBody)

		assert.Equal(t, http.StatusMethodNotAllowed, res.StatusCode)
		assert.Contains(t, resString, "Method Not Allowed")
		assert.Contains(t, res.Header.Get("Content-Type"), contenttype.HTML)
	})
}