54 lines
1.8 KiB
Go
54 lines
1.8 KiB
Go
package asset
|
|
|
|
import (
|
|
"context"
|
|
"errors"
|
|
"testing"
|
|
|
|
bizasset "kra/internal/biz/asset"
|
|
"kra/internal/data/ledger"
|
|
|
|
"github.com/glebarez/sqlite"
|
|
"gorm.io/gorm"
|
|
)
|
|
|
|
type testProvider struct{ db *gorm.DB }
|
|
|
|
func (p testProvider) DB() *gorm.DB { return p.db }
|
|
|
|
func TestAdjustmentRejectsInsufficientBalanceWithoutAudit(t *testing.T) {
|
|
db, err := gorm.Open(sqlite.Open("file:"+t.Name()+"?mode=memory&cache=shared"), &gorm.Config{})
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if err = db.AutoMigrate(&ledger.WalletPO{}, &ledger.BalanceAdjustmentPO{}, &ledger.RechargeOrderPO{}, &ledger.GiftSendPO{}); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
for _, statement := range []string{
|
|
`CREATE TABLE sys_users (id integer primary key, username text, nick_name text, header_img text, enable integer, created_at datetime, deleted_at datetime)`,
|
|
`CREATE TABLE sys_login_logs (id integer primary key, user_id integer, status numeric, created_at datetime)`,
|
|
} {
|
|
if err = db.Exec(statement).Error; err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
}
|
|
if err = db.Exec(`INSERT INTO sys_users(id,username,nick_name,enable,created_at) VALUES (4,'user','User',1,CURRENT_TIMESTAMP)`).Error; err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if err = db.Create(&ledger.WalletPO{UserID: 4, DiamondBalance: 20}).Error; err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
repo := NewRepo(testProvider{db: db})
|
|
_, err = repo.Adjust(context.Background(), &bizasset.Adjustment{UserID: 4, OperatorUserID: 1, Type: "decrease", Amount: 30, Reason: "test reason"})
|
|
if !errors.Is(err, bizasset.ErrInsufficient) {
|
|
t.Fatalf("adjustment error = %v", err)
|
|
}
|
|
var wallet ledger.WalletPO
|
|
var audits int64
|
|
db.Where("user_id = ?", 4).First(&wallet)
|
|
db.Model(&ledger.BalanceAdjustmentPO{}).Count(&audits)
|
|
if wallet.DiamondBalance != 20 || audits != 0 {
|
|
t.Fatalf("failed adjustment changed state: wallet=%+v audits=%d", wallet, audits)
|
|
}
|
|
}
|