hello-dd-code
97 lines
2.9 KiB
Go
97 lines
2.9 KiB
Go
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)
|
|
}
|
|
}
|