feat(server/reward): OnRegister 绑定 + 注册段双方发奖 + 自邀请/设备/月封顶防刷

Co-Authored-By: Claude Sonnet 5 <noreply@anthropic.com>
2026-07-13 07:27:10 +08:00
parent 2bd31bbd55
commit 2fb7391a17
2 changed files with 196 additions and 0 deletions
@@ -0,0 +1,80 @@
+package reward
+
+import (
+	"context"
+	"database/sql"
+	"testing"
+	"time"
+
+	"github.com/wangjia/pangolin/server/internal/codes"
+)
+
+func newSvc(t *testing.T, db *sql.DB) *Service {
+	g := codes.NewService(codes.NewStore(db), nil, 5, time.Hour)
+	return NewService(db, NewStore(db), g, Config{RegDays: 3, PaidDays: 7, TgDays: 3, RegMonthlyCap: 10}, time.Now)
+}
+
+func proDays(t *testing.T, db *sql.DB, userID int64) int { // 该用户 pro 剩余天数(粗算)
+	var exp sql.NullTime
+	db.QueryRow(`SELECT expires_at FROM subscriptions s JOIN plans p ON p.id=s.plan_id
+		WHERE s.user_id=? AND p.code='pro' ORDER BY expires_at DESC LIMIT 1`, userID).Scan(&exp)
+	if !exp.Valid {
+		return 0
+	}
+	return int(time.Until(exp.Time).Hours()/24) + 1
+}
+
+func TestOnRegister_BothGetThreeDays(t *testing.T) {
+	db := openDB(t)
+	seedU(t, db, 1, "inviter")
+	seedU(t, db, 2, "invitee")
+	s := newSvc(t, db)
+	code, _ := s.EnsureCode(context.Background(), 1)
+	s.OnRegister(context.Background(), 2, code, "dev-2")
+
+	var status string
+	db.QueryRow(`SELECT status FROM referrals WHERE invitee_id=2`).Scan(&status)
+	if status != "reg_rewarded" {
+		t.Fatalf("status=%q", status)
+	}
+	if d := proDays(t, db, 1); d < 3 {
+		t.Fatalf("inviter pro days=%d, want>=3", d)
+	}
+	if d := proDays(t, db, 2); d < 3 {
+		t.Fatalf("invitee pro days=%d, want>=3", d)
+	}
+}
+
+func TestOnRegister_SelfInviteRejected(t *testing.T) {
+	db := openDB(t)
+	seedU(t, db, 1, "u1")
+	s := newSvc(t, db)
+	code, _ := s.EnsureCode(context.Background(), 1)
+	s.OnRegister(context.Background(), 1, code, "dev-1") // 自邀请
+	var n int
+	db.QueryRow(`SELECT COUNT(*) FROM referrals`).Scan(&n)
+	if n != 0 {
+		t.Fatalf("self-invite created %d referrals, want 0", n)
+	}
+}
+
+func TestOnRegister_DeviceReusedRejectsReward(t *testing.T) {
+	db := openDB(t)
+	seedU(t, db, 1, "inv")
+	seedU(t, db, 2, "old")
+	seedU(t, db, 3, "new")
+	// old(2) 的设备 dev-X 已在 devices
+	db.Exec(`INSERT INTO devices (uuid,user_id,name,platform,last_seen,created_at)
+		VALUES ('dev-X',2,'n','android',?,?)`, time.Now().UTC(), time.Now().UTC())
+	s := newSvc(t, db)
+	code, _ := s.EnsureCode(context.Background(), 1)
+	s.OnRegister(context.Background(), 3, code, "dev-X") // 新用户复用老设备
+	var status string
+	db.QueryRow(`SELECT status FROM referrals WHERE invitee_id=3`).Scan(&status)
+	if status != "rejected" {
+		t.Fatalf("status=%q, want rejected", status)
+	}
+	if d := proDays(t, db, 1); d >= 3 {
+		t.Fatalf("inviter got reward despite device reuse: %d", d)
+	}
+}