GoBlog/blogstats_test.go

package main

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

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

func Test_blogStats(t *testing.T) {

	app := &goBlog{
		cfg: createDefaultTestConfig(t),
	}

	app.cfg.Blogs = map[string]*configBlog{
		"en": {
			Lang: "en",
			BlogStats: &configBlogStats{
				Enabled: true,
				Path:    "/stats",
			},
			Sections: map[string]*configSection{
				"test": {},
			},
		},
	}
	app.cfg.DefaultBlog = "en"

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

	// Insert post

	err := app.createPost(&post{
		Content:    "This is a simple **test** post",
		Blog:       "en",
		Section:    "test",
		Published:  "2020-06-01",
		Status:     statusPublished,
		Visibility: visibilityPublic,
	})
	require.NoError(t, err)

	err = app.createPost(&post{
		Content:    "This is another simple **test** post",
		Blog:       "en",
		Section:    "test",
		Published:  "2021-05-01",
		Status:     statusPublished,
		Visibility: visibilityPublic,
	})
	require.NoError(t, err)

	err = app.createPost(&post{
		Content:    "This is a private post, that doesn't count",
		Blog:       "en",
		Section:    "test",
		Published:  "2021-05-01",
		Status:     statusPublished,
		Visibility: visibilityPrivate,
	})
	require.NoError(t, err)

	err = app.createPost(&post{
		Content:    "Unlisted posts don't count as well",
		Blog:       "en",
		Section:    "test",
		Published:  "2021-05-01",
		Status:     statusPublished,
		Visibility: visibilityUnlisted,
	})
	require.NoError(t, err)

	// Test stats

	sd, err := app.db.getBlogStats("en")
	require.NoError(t, err)
	require.NotNil(t, sd)

	require.NotNil(t, sd.Total)
	assert.Equal(t, "2", sd.Total.Posts)
	assert.Equal(t, "12", sd.Total.Words)
	assert.Equal(t, "48", sd.Total.Chars)

	// 2021
	require.NotNil(t, sd.Years)
	row := sd.Years[0]
	require.NotNil(t, row)
	assert.Equal(t, "2021", row.Name)
	assert.Equal(t, "1", row.Posts)
	assert.Equal(t, "6", row.Words)
	assert.Equal(t, "27", row.Chars)

	// 2021-05
	require.NotNil(t, sd.Months)
	require.NotEmpty(t, sd.Months["2021"])
	row = sd.Months["2021"][0]
	require.NotNil(t, row)
	assert.Equal(t, "05", row.Name)
	assert.Equal(t, "1", row.Posts)
	assert.Equal(t, "6", row.Words)
	assert.Equal(t, "27", row.Chars)

	// 2020
	require.NotNil(t, sd.Years)
	row = sd.Years[1]
	require.NotNil(t, row)
	assert.Equal(t, "2020", row.Name)
	assert.Equal(t, "1", row.Posts)
	assert.Equal(t, "6", row.Words)
	assert.Equal(t, "21", row.Chars)

	// Test if cache exists

	assert.NotNil(t, app.db.loadBlogStatsCache("en"))

	// Test HTML

	t.Run("Test stats table", func(t *testing.T) {
		req := httptest.NewRequest(http.MethodGet, "/abc", nil)
		req = req.WithContext(context.WithValue(req.Context(), blogKey, "en"))

		rec := httptest.NewRecorder()

		app.serveBlogStatsTable(rec, req)

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

		assert.Equal(t, http.StatusOK, res.StatusCode)
		assert.Contains(t, resString, "class=statsyear data-year=2021")
		assert.Contains(t, res.Header.Get(contentType), contenttype.HTML)
	})

	t.Run("Test stats page", func(t *testing.T) {
		req := httptest.NewRequest(http.MethodGet, "/abc", nil)
		req = req.WithContext(context.WithValue(req.Context(), blogKey, "en"))

		rec := httptest.NewRecorder()

		app.serveBlogStats(rec, req)

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

		assert.Equal(t, http.StatusOK, res.StatusCode)
		assert.Contains(t, resString, "data-table=/stats.table.html")
		assert.Contains(t, res.Header.Get(contentType), contenttype.HTML)
	})

}