shanshanzhong147
ce3babcc33
Closes HIF-3 Stage 1 完整闭环 PR C：用户 API + 后台 CRUD + 抽奖事务服务 + 审计 + QA curl。合并后 Stage 1 可交测试。 架构师 review R1（rulecaps depth≤8/nodes≤64/bytes≤8KB）+ R2（InviteHook source_ref 加 order: 前缀）已全部落地。 - 迁移 02158_admin_action_log：后台写操作审计 - xerr 100xxx 段：抽奖错误码（NotEligible/NoChances/ActivityEnded/RateLimited/NotClaimable/InternalError/RuleTooDeep/RuleTooMany/RuleTooLarge） - feature flag config.Lottery.Enable 默认 false，合并后线上零副作用 - draw service：feature flag → rate limit → pre-tx reads → nonce dedupe → Consume → Pick → 乐观扣库存 → 双快照 → Dispatch → finalize - 用户 API 4 个：GET /config、POST /draw、GET /records、POST /claim（Stage 1 返回 100010） - 后台 CRUD：活动 / 奖品 / rules PUT（rulecaps gate）/ chances/grant - audit.WriteAdminAction：与调用方 tx 同生共死，SHA1 body 摘要 - QA 脚本：qa/lottery/stage1_curl.sh 全链路 curl 测试覆盖：model 76.5% / draw 69% / handler 68.3% / hook 91.9% / rulecaps 87.8% / audit 100% 100 并发抢库存 + 10000 次概率分布 e2e 推 QA 环境（sqlmock 无法忠实模拟 InnoDB 行锁） CI 全绿：构建/Vet/测试 + golangci-lint
112 lines
2.9 KiB
Go
112 lines
2.9 KiB
Go
package audit
|
|
|
|
import (
|
|
"context"
|
|
"errors"
|
|
"fmt"
|
|
"strings"
|
|
"testing"
|
|
|
|
"github.com/DATA-DOG/go-sqlmock"
|
|
"gorm.io/driver/mysql"
|
|
"gorm.io/gorm"
|
|
)
|
|
|
|
func newTestDB(t *testing.T) (*gorm.DB, sqlmock.Sqlmock, func()) {
|
|
t.Helper()
|
|
sqlDB, mock, err := sqlmock.New(sqlmock.QueryMatcherOption(sqlmock.QueryMatcherFunc(func(expected, actual string) error {
|
|
if strings.Contains(actual, expected) {
|
|
return nil
|
|
}
|
|
return fmt.Errorf("actual sql %q does not contain %q", actual, expected)
|
|
})))
|
|
if err != nil {
|
|
t.Fatalf("create sqlmock: %v", err)
|
|
}
|
|
db, err := gorm.Open(mysql.New(mysql.Config{Conn: sqlDB, SkipInitializeWithVersion: true}), &gorm.Config{SkipDefaultTransaction: true})
|
|
if err != nil {
|
|
_ = sqlDB.Close()
|
|
t.Fatalf("open gorm: %v", err)
|
|
}
|
|
return db, mock, func() { _ = sqlDB.Close() }
|
|
}
|
|
|
|
func TestWriteAdminAction_RequiresTx(t *testing.T) {
|
|
err := WriteAdminAction(context.Background(), nil, Entry{ActorUserId: 1, Action: "x"})
|
|
if err == nil {
|
|
t.Fatalf("expected error on nil tx")
|
|
}
|
|
}
|
|
|
|
func TestWriteAdminAction_RequiresActorAndAction(t *testing.T) {
|
|
db, _, cleanup := newTestDB(t)
|
|
defer cleanup()
|
|
|
|
if err := WriteAdminAction(context.Background(), db, Entry{Action: "x"}); err == nil {
|
|
t.Fatal("expected error when ActorUserId=0")
|
|
}
|
|
if err := WriteAdminAction(context.Background(), db, Entry{ActorUserId: 1}); err == nil {
|
|
t.Fatal("expected error when Action empty")
|
|
}
|
|
}
|
|
|
|
func TestWriteAdminAction_InsertsRow(t *testing.T) {
|
|
db, mock, cleanup := newTestDB(t)
|
|
defer cleanup()
|
|
|
|
mock.ExpectExec("INSERT INTO `admin_action_log`").
|
|
WillReturnResult(sqlmock.NewResult(1, 1))
|
|
|
|
err := WriteAdminAction(context.Background(), db, Entry{
|
|
ActorUserId: 42,
|
|
Action: ActionLotteryActivityCreate,
|
|
TargetIds: "[1,2,3]",
|
|
RequestBody: []byte(`{"title":"test"}`),
|
|
IP: "127.0.0.1",
|
|
UserAgent: "curl/7.85",
|
|
})
|
|
if err != nil {
|
|
t.Fatalf("WriteAdminAction: %v", err)
|
|
}
|
|
if err := mock.ExpectationsWereMet(); err != nil {
|
|
t.Fatalf("expectations: %v", err)
|
|
}
|
|
}
|
|
|
|
func TestHashBody(t *testing.T) {
|
|
if got := hashBody(nil); got != "" {
|
|
t.Fatalf("nil body should hash to empty, got %q", got)
|
|
}
|
|
if got := hashBody([]byte("")); got != "" {
|
|
t.Fatalf("empty body should hash to empty, got %q", got)
|
|
}
|
|
if got := hashBody([]byte("abc")); len(got) != 40 {
|
|
t.Fatalf("expected 40-char sha1 hex, got %q", got)
|
|
}
|
|
}
|
|
|
|
func TestTruncate(t *testing.T) {
|
|
if got := truncate("hello", 10); got != "hello" {
|
|
t.Fatalf("short strings pass through, got %q", got)
|
|
}
|
|
if got := truncate("hello world", 5); got != "hello" {
|
|
t.Fatalf("expected truncation to 5, got %q", got)
|
|
}
|
|
}
|
|
|
|
func TestWriteAdminAction_DBError(t *testing.T) {
|
|
db, mock, cleanup := newTestDB(t)
|
|
defer cleanup()
|
|
|
|
mock.ExpectExec("INSERT INTO `admin_action_log`").
|
|
WillReturnError(errors.New("db down"))
|
|
|
|
err := WriteAdminAction(context.Background(), db, Entry{
|
|
ActorUserId: 1,
|
|
Action: "test",
|
|
})
|
|
if err == nil {
|
|
t.Fatal("expected error propagation from DB")
|
|
}
|
|
}
|