chore: 并入 iotlcm-worker-svr 历史到 worker-svr/（原提交 96f0f4e）

2026-11-02 13:28:03 +08:00
parent 6de2e503cc 96f0f4eacd
commit 2923c78755
422 changed files with 80993 additions and 0 deletions
@@ -0,0 +1,108 @@
+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)
+		}
+	}
+}