MiFaZhan
109 lines
2.9 KiB
Go
109 lines
2.9 KiB
Go
package renewchange
|
|
|
|
import (
|
|
"context"
|
|
"path/filepath"
|
|
"strings"
|
|
"testing"
|
|
"time"
|
|
|
|
_ "github.com/gogf/gf/contrib/drivers/sqlite/v2"
|
|
"github.com/gogf/gf/v2/database/gdb"
|
|
"github.com/gogf/gf/v2/frame/g"
|
|
|
|
rc "git.5iot.com/iotlcm-public/iotlcm-worker-svr/internal/repository/renewchange"
|
|
)
|
|
|
|
func TestDeductRuntimeRenewBalanceRejectsInsufficientBalanceAtomically(t *testing.T) {
|
|
ctx := context.Background()
|
|
if err := gdb.SetConfig(gdb.Config{
|
|
"default": gdb.ConfigGroup{{Type: "sqlite", Name: filepath.Join(t.TempDir(), "renew-balance.db")}},
|
|
}); err != nil {
|
|
t.Fatalf("configure sqlite: %v", err)
|
|
}
|
|
defer g.DB().Close(ctx)
|
|
|
|
createRenewBalanceTables(t, ctx)
|
|
if _, err := g.DB().Model(rc.TableAgentBalance).Ctx(ctx).Data(g.Map{
|
|
"dept_id": 117,
|
|
"tenant_id": "000000",
|
|
"balance": 100,
|
|
"status": renewFinanceAccountStatusNormal,
|
|
}).Insert(); err != nil {
|
|
t.Fatalf("seed balance: %v", err)
|
|
}
|
|
|
|
main := &rc.BatchMainRow{
|
|
BatchNo: "REN20260806110000TEST",
|
|
TenantId: "000000",
|
|
CreatedBy: 60,
|
|
}
|
|
err := g.DB().Transaction(ctx, func(ctx context.Context, tx gdb.TX) error {
|
|
_, _, _, err := deductRuntimeRenewBalance(ctx, tx, main, 117, 576, time.Date(2026, 8, 6, 11, 0, 0, 0, time.Local))
|
|
return err
|
|
})
|
|
if err == nil || !strings.Contains(err.Error(), "账户余额不足") {
|
|
t.Fatalf("deduct error = %v, want insufficient balance", err)
|
|
}
|
|
|
|
balance, err := g.DB().Model(rc.TableAgentBalance).Ctx(ctx).
|
|
Where("dept_id", 117).
|
|
Where("tenant_id", "000000").
|
|
Value("balance")
|
|
if err != nil {
|
|
t.Fatalf("query balance: %v", err)
|
|
}
|
|
if got := balance.Float64(); got != 100 {
|
|
t.Fatalf("balance = %.2f, want unchanged 100.00", got)
|
|
}
|
|
|
|
count, err := g.DB().Model(rc.TableBalanceDetail).Ctx(ctx).
|
|
Where("serial_no", "RENPAY"+main.BatchNo).
|
|
Count()
|
|
if err != nil {
|
|
t.Fatalf("count balance detail: %v", err)
|
|
}
|
|
if count != 0 {
|
|
t.Fatalf("balance detail count = %d, want 0", count)
|
|
}
|
|
}
|
|
|
|
func createRenewBalanceTables(t *testing.T, ctx context.Context) {
|
|
t.Helper()
|
|
statements := []string{
|
|
`CREATE TABLE t_agent_balance (
|
|
id INTEGER PRIMARY KEY AUTOINCREMENT,
|
|
dept_id INTEGER NOT NULL,
|
|
tenant_id TEXT NOT NULL,
|
|
balance REAL NOT NULL,
|
|
status INTEGER NOT NULL DEFAULT 0,
|
|
updated_by INTEGER,
|
|
updated_at DATETIME
|
|
)`,
|
|
`CREATE TABLE t_balance_detail (
|
|
id INTEGER PRIMARY KEY AUTOINCREMENT,
|
|
dept_id INTEGER NOT NULL,
|
|
serial_no TEXT NOT NULL,
|
|
transaction_time DATETIME,
|
|
income_expense_type INTEGER,
|
|
transaction_type INTEGER,
|
|
transaction_channel INTEGER,
|
|
transaction_status INTEGER,
|
|
amount REAL,
|
|
before_balance REAL,
|
|
balance_after REAL,
|
|
order_sn TEXT,
|
|
remark TEXT,
|
|
tenant_id TEXT NOT NULL,
|
|
created_dept INTEGER,
|
|
created_by INTEGER,
|
|
created_at DATETIME
|
|
)`,
|
|
}
|
|
for _, statement := range statements {
|
|
if _, err := g.DB().Exec(ctx, statement); err != nil {
|
|
t.Fatalf("create renew balance table: %v", err)
|
|
}
|
|
}
|
|
}
|