Asai Neko
210b8b08ce
All checks were successful
Server Check Build (NixCN CMS) TeamCity build finished
Signed-off-by: Asai Neko <sugar@sne.moe>
67 lines
1.6 KiB
Go
67 lines
1.6 KiB
Go
package stats
|
|
|
|
import (
|
|
"context"
|
|
"encoding/json"
|
|
"net/http"
|
|
"net/http/httptest"
|
|
"testing"
|
|
|
|
"github.com/gin-gonic/gin"
|
|
"github.com/google/uuid"
|
|
"github.com/spf13/viper"
|
|
"github.com/stretchr/testify/assert"
|
|
"github.com/stretchr/testify/require"
|
|
"nixcn-cms/internal/authtoken"
|
|
"nixcn-cms/testutil"
|
|
)
|
|
|
|
func init() { gin.SetMode(gin.TestMode) }
|
|
|
|
func issueToken(t *testing.T, userId uuid.UUID) string {
|
|
t.Helper()
|
|
tok := &authtoken.Token{Application: viper.GetString("server.application")}
|
|
access, _, err := tok.IssueTokens(context.Background(), testutil.TestClientID, userId)
|
|
require.NoError(t, err)
|
|
return access
|
|
}
|
|
|
|
func newStatsRouter(t *testing.T) *gin.Engine {
|
|
t.Helper()
|
|
r := gin.New()
|
|
ApiHandler(r.Group("/stats"))
|
|
return r
|
|
}
|
|
|
|
func TestGlobalStatsHandler(t *testing.T) {
|
|
testutil.SetupWithAuth(t)
|
|
admin := testutil.SeedUser(t, testutil.RandomEmail(), 40)
|
|
token := issueToken(t, admin.UserId)
|
|
r := newStatsRouter(t)
|
|
|
|
req := httptest.NewRequest(http.MethodGet, "/stats/global", nil)
|
|
req.Header.Set("Authorization", "Bearer "+token)
|
|
w := httptest.NewRecorder()
|
|
r.ServeHTTP(w, req)
|
|
|
|
assert.Equal(t, http.StatusOK, w.Code)
|
|
|
|
var resp map[string]any
|
|
require.NoError(t, json.Unmarshal(w.Body.Bytes(), &resp))
|
|
data, ok := resp["data"].(map[string]any)
|
|
require.True(t, ok)
|
|
_, hasTotalUsers := data["total_users"]
|
|
assert.True(t, hasTotalUsers)
|
|
}
|
|
|
|
func TestGlobalStatsHandlerNoAuth(t *testing.T) {
|
|
testutil.Setup(t)
|
|
r := newStatsRouter(t)
|
|
|
|
req := httptest.NewRequest(http.MethodGet, "/stats/global", nil)
|
|
w := httptest.NewRecorder()
|
|
r.ServeHTTP(w, req)
|
|
|
|
assert.Equal(t, http.StatusUnauthorized, w.Code)
|
|
}
|