统一会员门禁逻辑并补权限分支测试

2026-02-28 16:27:48 +08:00
parent 28690e2477
commit 01dd0ccbbe
7 changed files with 229 additions and 28 deletions
@@ -0,0 +1,96 @@
+package service
+
+import (
+	"context"
+	"testing"
+	"time"
+
+	"github.com/DATA-DOG/go-sqlmock"
+
+	usermodel "wx_service/internal/model"
+)
+
+func TestSmokeAIAdviceServiceIsAllowedMember(t *testing.T) {
+	t.Parallel()
+
+	db, mock, cleanup := newMockGormDB(t)
+	defer cleanup()
+
+	svc := &SmokeAIAdviceService{db: db}
+	user := &usermodel.User{ID: 200, MiniProgramID: 100}
+	adviceDate := time.Date(2026, 3, 1, 0, 0, 0, 0, time.Local)
+
+	mock.ExpectQuery("SELECT count\\(\\*\\) FROM `user_memberships`").
+		WithArgs(uint(100), uint(200), "active", sqlmock.AnyArg()).
+		WillReturnRows(sqlmock.NewRows([]string{"count"}).AddRow(1))
+
+	allowed, err := svc.isAllowed(context.Background(), user, adviceDate)
+	if err != nil {
+		t.Fatalf("isAllowed: %v", err)
+	}
+	if !allowed {
+		t.Fatalf("isAllowed got=false, want=true for active member")
+	}
+	if err := mock.ExpectationsWereMet(); err != nil {
+		t.Fatalf("unmet expectations: %v", err)
+	}
+}
+
+func TestSmokeAIAdviceServiceIsAllowedNonMemberLocked(t *testing.T) {
+	t.Parallel()
+
+	db, mock, cleanup := newMockGormDB(t)
+	defer cleanup()
+
+	svc := &SmokeAIAdviceService{db: db}
+	user := &usermodel.User{ID: 201, MiniProgramID: 101}
+	adviceDate := time.Date(2026, 3, 1, 0, 0, 0, 0, time.Local)
+
+	mock.ExpectQuery("SELECT count\\(\\*\\) FROM `user_memberships`").
+		WithArgs(uint(101), uint(201), "active", sqlmock.AnyArg()).
+		WillReturnRows(sqlmock.NewRows([]string{"count"}).AddRow(0))
+	mock.ExpectQuery("SELECT \\* FROM `fa_smoke_ai_advice_unlocks`").
+		WithArgs(201, adviceDate.Format("2006-01-02"), sqlmock.AnyArg()).
+		WillReturnRows(sqlmock.NewRows([]string{"id", "uid", "unlock_date", "ad_watched_at"}))
+
+	allowed, err := svc.isAllowed(context.Background(), user, adviceDate)
+	if err != nil {
+		t.Fatalf("isAllowed: %v", err)
+	}
+	if allowed {
+		t.Fatalf("isAllowed got=true, want=false for non-member locked user")
+	}
+	if err := mock.ExpectationsWereMet(); err != nil {
+		t.Fatalf("unmet expectations: %v", err)
+	}
+}
+
+func TestSmokeAINextSmokeServiceIsAllowedNonMemberUnlocked(t *testing.T) {
+	t.Parallel()
+
+	db, mock, cleanup := newMockGormDB(t)
+	defer cleanup()
+
+	svc := &SmokeAINextSmokeService{db: db}
+	user := &usermodel.User{ID: 202, MiniProgramID: 102}
+	planDate := time.Date(2026, 3, 1, 0, 0, 0, 0, time.Local)
+
+	mock.ExpectQuery("SELECT count\\(\\*\\) FROM `user_memberships`").
+		WithArgs(uint(102), uint(202), "active", sqlmock.AnyArg()).
+		WillReturnRows(sqlmock.NewRows([]string{"count"}).AddRow(0))
+	mock.ExpectQuery("SELECT \\* FROM `fa_smoke_ai_advice_unlocks`").
+		WithArgs(202, planDate.Format("2006-01-02"), sqlmock.AnyArg()).
+		WillReturnRows(sqlmock.NewRows([]string{"id", "uid", "unlock_date", "ad_watched_at"}).
+			AddRow(1, 202, planDate, time.Now()))
+
+	allowed, err := svc.isAllowed(context.Background(), user, planDate)
+	if err != nil {
+		t.Fatalf("isAllowed: %v", err)
+	}
+	if !allowed {
+		t.Fatalf("isAllowed got=false, want=true for non-member unlocked user")
+	}
+	if err := mock.ExpectationsWereMet(); err != nil {
+		t.Fatalf("unmet expectations: %v", err)
+	}
+}