GoBlog/httpLogs_test.go

package main

import (
	"io"
	"net/http"
	"net/http/httptest"
	"os"
	"path/filepath"
	"testing"

	"github.com/stretchr/testify/assert"
	"github.com/stretchr/testify/require"
)

func initTestHttpLogs(logFile string) (http.Handler, error) {

	app := &goBlog{
		cfg: &config{
			Server: &configServer{
				Logging: true,
				LogFile: logFile,
			},
		},
	}

	err := app.initHTTPLog()
	if err != nil {
		return nil, err
	}

	return app.logMiddleware(testHttpHandler()), nil

}

func testHttpHandler() http.Handler {
	return http.HandlerFunc(func(rw http.ResponseWriter, r *http.Request) {
		_, _ = rw.Write([]byte("Test"))
	})
}

func Test_httpLogs(t *testing.T) {

	// Init

	logFile := filepath.Join(t.TempDir(), "access.log")
	handler, err := initTestHttpLogs(logFile)

	require.NoError(t, err)

	// Do fake request

	req := httptest.NewRequest(http.MethodGet, "/testpath", nil)
	rec := httptest.NewRecorder()

	handler.ServeHTTP(rec, req)

	// Check response

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

	assert.Equal(t, http.StatusOK, res.StatusCode)
	assert.Contains(t, resBodyStr, "Test")

	// Check log

	logBytes, err := os.ReadFile(logFile)
	require.NoError(t, err)

	logString := string(logBytes)
	assert.Contains(t, logString, "GET /testpath")

}

func Benchmark_httpLogs(b *testing.B) {

	// Init

	logFile := filepath.Join(b.TempDir(), "access.log")
	logHandler, err := initTestHttpLogs(logFile)
	require.NoError(b, err)

	noLogHandler := testHttpHandler()

	// Run benchmarks

	b.Run("With logging", func(b *testing.B) {
		b.RunParallel(func(p *testing.PB) {
			for p.Next() {
				logHandler.ServeHTTP(httptest.NewRecorder(), httptest.NewRequest(http.MethodGet, "/testpath", nil))
			}
		})

	})

	b.Run("Without logging", func(b *testing.B) {
		b.RunParallel(func(p *testing.PB) {
			for p.Next() {
				noLogHandler.ServeHTTP(httptest.NewRecorder(), httptest.NewRequest(http.MethodGet, "/testpath", nil))
			}
		})
	})

}