270 lines
11 KiB
Go
270 lines
11 KiB
Go
package vote
|
||
|
||
import (
|
||
stdctx "context"
|
||
"encoding/json"
|
||
"strings"
|
||
"testing"
|
||
"time"
|
||
|
||
"github.com/tamakyi/TamaBox/internal/conf"
|
||
"github.com/tamakyi/TamaBox/internal/db"
|
||
"github.com/tamakyi/TamaBox/internal/dbutil"
|
||
"github.com/tamakyi/TamaBox/internal/i18n"
|
||
)
|
||
|
||
// mockVoteSubmissions 仅实现 aggregateFormFieldStats 依赖的 ListByVoteID。
|
||
type mockVoteSubmissions struct {
|
||
submissions []*db.VoteSubmission
|
||
}
|
||
|
||
func (m *mockVoteSubmissions) ListByVoteID(ctx stdctx.Context, voteID uint, cursor *dbutil.Cursor) ([]*db.VoteSubmission, error) {
|
||
return m.submissions, nil
|
||
}
|
||
func (m *mockVoteSubmissions) ListByVoteIDs(ctx stdctx.Context, voteIDs []uint) (map[uint][]*db.VoteSubmission, error) {
|
||
out := make(map[uint][]*db.VoteSubmission)
|
||
for _, s := range m.submissions {
|
||
out[s.VoteID] = append(out[s.VoteID], s)
|
||
}
|
||
return out, nil
|
||
}
|
||
func (m *mockVoteSubmissions) Create(ctx stdctx.Context, opts db.CreateVoteSubmissionOptions) (*db.VoteSubmission, error) {
|
||
return nil, nil
|
||
}
|
||
func (m *mockVoteSubmissions) GetByID(ctx stdctx.Context, id uint) (*db.VoteSubmission, error) {
|
||
return nil, nil
|
||
}
|
||
func (m *mockVoteSubmissions) CountByVoteID(ctx stdctx.Context, voteID uint) (int64, error) {
|
||
return 0, nil
|
||
}
|
||
func (m *mockVoteSubmissions) CountByVoteIDs(ctx stdctx.Context, voteIDs []uint) (map[uint]int64, error) {
|
||
return nil, nil
|
||
}
|
||
func (m *mockVoteSubmissions) CountByOptionID(ctx stdctx.Context, voteID, optionID uint) (int64, error) {
|
||
return 0, nil
|
||
}
|
||
func (m *mockVoteSubmissions) CountGroupByOptionID(ctx stdctx.Context, voteID uint) (map[uint]int64, error) {
|
||
return nil, nil
|
||
}
|
||
func (m *mockVoteSubmissions) CountByIP(ctx stdctx.Context, voteID uint, ip string, since time.Time) (int64, error) {
|
||
return 0, nil
|
||
}
|
||
func (m *mockVoteSubmissions) HasUserVoted(ctx stdctx.Context, voteID, userID uint) (bool, error) {
|
||
return false, nil
|
||
}
|
||
func (m *mockVoteSubmissions) HasTokenVoted(ctx stdctx.Context, voteID uint, token string) (bool, error) {
|
||
return false, nil
|
||
}
|
||
func (m *mockVoteSubmissions) DeleteByVoteID(ctx stdctx.Context, voteID uint) error { return nil }
|
||
|
||
func formData(t *testing.T, v map[string]interface{}) []byte {
|
||
t.Helper()
|
||
b, err := json.Marshal(v)
|
||
if err != nil {
|
||
t.Fatalf("marshal form data: %v", err)
|
||
}
|
||
return b
|
||
}
|
||
|
||
func TestParseMultipleOptionIDsRejectsDuplicates(t *testing.T) {
|
||
valid := map[uint]bool{1: true, 2: true, 3: true}
|
||
chosen, err := parseMultipleOptionIDs([]string{"1", "2"}, 2, valid, i18n.Default)
|
||
if err != nil {
|
||
t.Fatalf("valid multiple options rejected: %v", err)
|
||
}
|
||
if len(chosen) != 2 || chosen[0] != 1 || chosen[1] != 2 {
|
||
t.Fatalf("unexpected parsed options: %#v", chosen)
|
||
}
|
||
|
||
cases := []struct {
|
||
name string
|
||
ids []string
|
||
want string
|
||
}{
|
||
{"duplicate", []string{"1", "1"}, "选项不能重复选择"},
|
||
{"too-many", []string{"1", "2", "3"}, "最多只能选择 2 个选项"},
|
||
{"invalid", []string{"1", "99"}, "存在无效的选项"},
|
||
{"overflow", []string{"18446744073709551616"}, "存在无效的选项"},
|
||
}
|
||
for _, tc := range cases {
|
||
t.Run(tc.name, func(t *testing.T) {
|
||
_, err := parseMultipleOptionIDs(tc.ids, 2, valid, i18n.Default)
|
||
if err == nil || err.Error() != tc.want {
|
||
t.Fatalf("parseMultipleOptionIDs(%v) error = %v, want %q", tc.ids, err, tc.want)
|
||
}
|
||
})
|
||
}
|
||
}
|
||
|
||
func TestVoterTokenSignatureValidation(t *testing.T) {
|
||
oldKey := conf.Server.XSRFKey
|
||
conf.Server.XSRFKey = "test-voter-token-secret"
|
||
defer func() { conf.Server.XSRFKey = oldKey }()
|
||
|
||
token := newVoterToken()
|
||
if !validVoterToken(token) {
|
||
t.Fatalf("new voter token should validate: %q", token)
|
||
}
|
||
if len(token) != 108 {
|
||
t.Fatalf("new voter token length = %d, want 108", len(token))
|
||
}
|
||
if validVoterToken(token + "x") {
|
||
t.Fatal("tampered voter token should be rejected")
|
||
}
|
||
parts := strings.Split(token, ".")
|
||
if len(parts) != 2 || validVoterToken(parts[0]+"."+strings.Repeat("A", len(parts[1]))) {
|
||
t.Fatal("token with forged signature should be rejected")
|
||
}
|
||
}
|
||
|
||
func TestAggregateFormFieldStats(t *testing.T) {
|
||
fields := []*db.VoteFormField{
|
||
{Model: dbutil.Model{ID: 1}, VoteID: 9, FieldType: db.VoteFormFieldTypeRadio, Label: "满意度", Options: `["满意","一般"]`},
|
||
{Model: dbutil.Model{ID: 2}, VoteID: 9, FieldType: db.VoteFormFieldTypeCheckbox, Label: "多选", Options: `["A","B"]`},
|
||
{Model: dbutil.Model{ID: 3}, VoteID: 9, FieldType: db.VoteFormFieldTypeRating, Label: "评分"},
|
||
{Model: dbutil.Model{ID: 4}, VoteID: 9, FieldType: db.VoteFormFieldTypeRank, Label: "排序", Options: `["X","Y","Z"]`},
|
||
{Model: dbutil.Model{ID: 5}, VoteID: 9, FieldType: db.VoteFormFieldTypeNumber, Label: "数量"},
|
||
{Model: dbutil.Model{ID: 6}, VoteID: 9, FieldType: db.VoteFormFieldTypeTextarea, Label: "建议"},
|
||
}
|
||
old := db.VoteSubmissions
|
||
defer func() { db.VoteSubmissions = old }()
|
||
|
||
subs := []*db.VoteSubmission{
|
||
{VoteID: 9, FormData: formData(t, map[string]interface{}{
|
||
"满意度": "满意", "多选": []string{"A", "B"}, "评分": 9, "排序": map[string]float64{"X": 1, "Y": 2, "Z": 3}, "数量": 3, "建议": "很好",
|
||
})},
|
||
{VoteID: 9, FormData: formData(t, map[string]interface{}{
|
||
"满意度": "满意", "多选": []string{"A"}, "评分": 7, "排序": map[string]float64{"X": 2, "Y": 1, "Z": 3}, "数量": 5, "建议": "",
|
||
})},
|
||
{VoteID: 9, FormData: formData(t, map[string]interface{}{
|
||
"满意度": "一般", "评分": 8, "排序": map[string]float64{"X": 3, "Y": 2, "Z": 1}, "数量": 10,
|
||
})},
|
||
}
|
||
db.VoteSubmissions = &mockVoteSubmissions{submissions: subs}
|
||
|
||
stats := aggregateFormFieldStatsWithSubs(fields, subs)
|
||
if len(stats) != 6 {
|
||
t.Fatalf("expected 6 field stats, got %d", len(stats))
|
||
}
|
||
|
||
// radio：满意 2 / 一般 1
|
||
radio := stats[0]
|
||
if radio.Total != 3 || len(radio.OptionStats) != 2 {
|
||
t.Fatalf("radio stats wrong: total=%d opts=%d", radio.Total, len(radio.OptionStats))
|
||
}
|
||
if radio.OptionStats[0].Count != 2 || radio.OptionStats[1].Count != 1 {
|
||
t.Fatalf("radio counts wrong: %+v", radio.OptionStats)
|
||
}
|
||
|
||
// checkbox：A 2 / B 1，回答数 2，总勾选数 3（A、B、A）
|
||
cb := stats[1]
|
||
if cb.Total != 2 || cb.OptionStats[0].Count != 2 || cb.OptionStats[1].Count != 1 {
|
||
t.Fatalf("checkbox stats wrong: total=%d %+v", cb.Total, cb.OptionStats)
|
||
}
|
||
if cb.TotalSelections != 3 {
|
||
t.Fatalf("checkbox TotalSelections wrong: got %d, want 3", cb.TotalSelections)
|
||
}
|
||
|
||
// rating：均值 (9+7+8)/3=8.0，总 3
|
||
rating := stats[2]
|
||
if rating.Total != 3 || rating.AvgRating < 7.99 || rating.AvgRating > 8.01 {
|
||
t.Fatalf("rating stats wrong: total=%d avg=%f", rating.Total, rating.AvgRating)
|
||
}
|
||
if len(rating.RatingCounts) != 10 {
|
||
t.Fatalf("rating counts must have 10 buckets, got %d", len(rating.RatingCounts))
|
||
}
|
||
|
||
// rank：X 平均 2，Y 平均 5/3≈1.67，Z 平均 7/3≈2.33，整体平均 2.0
|
||
rank := stats[3]
|
||
if rank.Total != 3 || rank.AvgRank < 1.99 || rank.AvgRank > 2.01 {
|
||
t.Fatalf("rank avg wrong: total=%d avg=%f", rank.Total, rank.AvgRank)
|
||
}
|
||
if rank.OptionStats[0].AvgRank < 1.99 || rank.OptionStats[0].AvgRank > 2.01 {
|
||
t.Fatalf("rank option X avg wrong: %f", rank.OptionStats[0].AvgRank)
|
||
}
|
||
if rank.OptionStats[1].AvgRank < 1.66 || rank.OptionStats[1].AvgRank > 1.68 {
|
||
t.Fatalf("rank option Y avg wrong: %f", rank.OptionStats[1].AvgRank)
|
||
}
|
||
|
||
// number：均值 (3+5+10)/3=6
|
||
num := stats[4]
|
||
if num.Total != 3 || num.AvgNumber < 5.99 || num.AvgNumber > 6.01 {
|
||
t.Fatalf("number stats wrong: total=%d avg=%f", num.Total, num.AvgNumber)
|
||
}
|
||
|
||
// textarea：填写 1；文本字段的"回答数"(Total) 应与"已填写数"(TextCount) 一致
|
||
text := stats[5]
|
||
if text.TextCount != 1 {
|
||
t.Fatalf("textarea count wrong: %d", text.TextCount)
|
||
}
|
||
if text.Total != text.TextCount {
|
||
t.Fatalf("textarea total should equal filled count, got total=%d filled=%d", text.Total, text.TextCount)
|
||
}
|
||
}
|
||
|
||
func TestAggregateFormFieldStatsEmpty(t *testing.T) {
|
||
fields := []*db.VoteFormField{
|
||
{Model: dbutil.Model{ID: 1}, VoteID: 9, FieldType: db.VoteFormFieldTypeRadio, Label: "满意度", Options: `["满意","一般"]`},
|
||
}
|
||
old := db.VoteSubmissions
|
||
defer func() { db.VoteSubmissions = old }()
|
||
db.VoteSubmissions = &mockVoteSubmissions{submissions: []*db.VoteSubmission{}}
|
||
|
||
stats := aggregateFormFieldStatsWithSubs(fields, nil)
|
||
if len(stats) != 1 || stats[0].Total != 0 || stats[0].OptionStats[0].Count != 0 {
|
||
t.Fatalf("empty stats wrong: %+v", stats)
|
||
}
|
||
}
|
||
|
||
// TestAggregateFormFieldStatsCustom 验证"允许补充自定义选项"字段：非预置值归入 CustomStats（按文本去重计数）。
|
||
func TestAggregateFormFieldStatsCustom(t *testing.T) {
|
||
fields := []*db.VoteFormField{
|
||
{Model: dbutil.Model{ID: 1}, VoteID: 9, FieldType: db.VoteFormFieldTypeRadio, Label: "满意度", Options: `["满意","一般"]`, AllowCustom: true},
|
||
{Model: dbutil.Model{ID: 2}, VoteID: 9, FieldType: db.VoteFormFieldTypeCheckbox, Label: "多选", Options: `["A","B"]`, AllowCustom: true},
|
||
// 未开启 AllowCustom 的字段：非预置值仍按普通值计数（不归入 CustomStats）
|
||
{Model: dbutil.Model{ID: 3}, VoteID: 9, FieldType: db.VoteFormFieldTypeRadio, Label: "无自定义", Options: `["X"]`},
|
||
}
|
||
subs := []*db.VoteSubmission{
|
||
{VoteID: 9, FormData: formData(t, map[string]interface{}{
|
||
"满意度": "满意", "多选": []string{"A", "自定义甲"}, "无自定义": "X",
|
||
})},
|
||
{VoteID: 9, FormData: formData(t, map[string]interface{}{
|
||
"满意度": "自定义乙", "多选": []string{"自定义甲", "B"},
|
||
})},
|
||
{VoteID: 9, FormData: formData(t, map[string]interface{}{
|
||
"满意度": "自定义乙", "多选": []string{"A"},
|
||
})},
|
||
}
|
||
stats := aggregateFormFieldStatsWithSubs(fields, subs)
|
||
|
||
// radio（AllowCustom）：满意 1 / 一般 0 / 自定义乙 2 / 自定义甲 0 → CustomStats 仅"自定义乙"计 2
|
||
radio := stats[0]
|
||
if radio.Total != 3 {
|
||
t.Fatalf("radio total wrong: %d", radio.Total)
|
||
}
|
||
if radio.OptionStats[0].Count != 1 || radio.OptionStats[1].Count != 0 {
|
||
t.Fatalf("radio preset counts wrong: %+v", radio.OptionStats)
|
||
}
|
||
if len(radio.CustomStats) != 1 || radio.CustomStats[0].Text != "自定义乙" || radio.CustomStats[0].Count != 2 {
|
||
t.Fatalf("radio custom stats wrong: %+v", radio.CustomStats)
|
||
}
|
||
|
||
// checkbox（AllowCustom）：A 2 / B 1 / 自定义甲 2（两条提交各选一次）→ CustomStats 计 2
|
||
cb := stats[1]
|
||
if cb.Total != 3 || cb.TotalSelections != 5 {
|
||
t.Fatalf("checkbox total/selections wrong: total=%d sel=%d", cb.Total, cb.TotalSelections)
|
||
}
|
||
if cb.OptionStats[0].Count != 2 || cb.OptionStats[1].Count != 1 {
|
||
t.Fatalf("checkbox preset counts wrong: %+v", cb.OptionStats)
|
||
}
|
||
if len(cb.CustomStats) != 1 || cb.CustomStats[0].Text != "自定义甲" || cb.CustomStats[0].Count != 2 {
|
||
t.Fatalf("checkbox custom stats wrong: %+v", cb.CustomStats)
|
||
}
|
||
|
||
// 未开启 AllowCustom 的字段：非预置值不应出现（提交校验已拦截），即使出现也只按普通值计数
|
||
noCustom := stats[2]
|
||
if noCustom.Total != 1 || len(noCustom.CustomStats) != 0 {
|
||
t.Fatalf("no-custom field should not produce CustomStats: %+v", noCustom)
|
||
}
|
||
}
|