Initial commit

Co-Authored-By: Claude Opus 5 <noreply@anthropic.com>
2026-09-15 13:57:10 +08:00
commit 8679200f41
1897 changed files with 257900 additions and 0 deletions
@@ -0,0 +1,134 @@
+package vidhelpser
+
+import (
+	"testing"
+
+	"91porn-server/models/v/vidmod"
+)
+
+func TestFreeTrialBadgeContextFields(t *testing.T) {
+	const (
+		uid         = uint64(1001)
+		publisherID = uint64(2002)
+	)
+
+	tests := []struct {
+		name          string
+		ctx           FreeTrialBadgeContext
+		newsType      string
+		originCoins   int64
+		freeArea      bool
+		publisherID   uint64
+		wantShow      bool
+		wantRemaining uint64
+		wantCanUse    bool
+	}{
+		{
+			name:          "eligible vip video",
+			ctx:           NewFreeTrialBadgeContext(uid, 3, 2, false, true),
+			newsType:      vidmod.SP,
+			publisherID:   publisherID,
+			wantShow:      true,
+			wantRemaining: 2,
+			wantCanUse:    true,
+		},
+		{
+			name:          "switch only hides badge",
+			ctx:           NewFreeTrialBadgeContext(uid, 3, 2, false, false),
+			newsType:      vidmod.SHORT,
+			publisherID:   publisherID,
+			wantRemaining: 2,
+			wantCanUse:    true,
+		},
+		{
+			name:          "active vip does not use trial",
+			ctx:           NewFreeTrialBadgeContext(uid, 3, 2, true, true),
+			newsType:      vidmod.SP,
+			publisherID:   publisherID,
+			wantRemaining: 2,
+		},
+		{
+			name:          "coin video",
+			ctx:           NewFreeTrialBadgeContext(uid, 3, 2, false, true),
+			newsType:      vidmod.SP,
+			originCoins:   10,
+			publisherID:   publisherID,
+			wantRemaining: 2,
+		},
+		{
+			name:          "free area",
+			ctx:           NewFreeTrialBadgeContext(uid, 3, 2, false, true),
+			newsType:      vidmod.SP,
+			freeArea:      true,
+			publisherID:   publisherID,
+			wantRemaining: 2,
+		},
+		{
+			name:          "own video",
+			ctx:           NewFreeTrialBadgeContext(uid, 3, 2, false, true),
+			newsType:      vidmod.SP,
+			publisherID:   uid,
+			wantRemaining: 2,
+		},
+		{
+			name:          "advertisement",
+			ctx:           NewFreeTrialBadgeContext(uid, 3, 2, false, true),
+			newsType:      vidmod.AD_SP,
+			publisherID:   publisherID,
+			wantRemaining: 2,
+		},
+		{
+			name:          "remaining count is clamped",
+			ctx:           NewFreeTrialBadgeContext(uid, 3, 99, false, true),
+			newsType:      vidmod.SP,
+			publisherID:   publisherID,
+			wantShow:      true,
+			wantRemaining: 3,
+			wantCanUse:    true,
+		},
+		{
+			name:          "no remaining count hides badge",
+			ctx:           NewFreeTrialBadgeContext(uid, 3, 0, false, true),
+			newsType:      vidmod.SP,
+			publisherID:   publisherID,
+			wantRemaining: 0,
+			wantCanUse:    false,
+		},
+	}
+
+	for _, tt := range tests {
+		t.Run(tt.name, func(t *testing.T) {
+			show, remaining, canUse := tt.ctx.Fields(tt.newsType, tt.originCoins, tt.freeArea, tt.publisherID)
+			if show != tt.wantShow || remaining != tt.wantRemaining || canUse != tt.wantCanUse {
+				t.Fatalf(
+					"Fields() = (%v, %d, %v), want (%v, %d, %v)",
+					show,
+					remaining,
+					canUse,
+					tt.wantShow,
+					tt.wantRemaining,
+					tt.wantCanUse,
+				)
+			}
+		})
+	}
+}
+
+func TestApplyFreeTrialBadgeToVideoInfos(t *testing.T) {
+	ctx := NewFreeTrialBadgeContext(1001, 3, 2, false, true)
+	video := &vidmod.VideoInfo{
+		VideoBase: vidmod.VideoBase{
+			NewsType:    vidmod.SP,
+			OriginCoins: 0,
+		},
+		UInfo: vidmod.Publisher{
+			UInfo: vidmod.UInfo{UID: 2002},
+		},
+	}
+
+	ApplyFreeTrialBadgeToVideoInfos(ctx, []*vidmod.VideoInfo{video})
+
+	if !video.ShowFreeTrialBadge || !video.CanUseFreeTrial || video.FreeTrialRemaining != 2 {
+		t.Fatalf("unexpected badge fields: %+v", video.VideoBase)
+	}
+}