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

worker-svr/internal/repository/monthlyusage/repository_test.go

@@ -0,0 +1,45 @@
+package monthlyusage
+
+import (
+	"context"
+	"testing"
+
+	_ "github.com/gogf/gf/contrib/drivers/sqlite/v2"
+	"github.com/gogf/gf/v2/database/gdb"
+	"github.com/gogf/gf/v2/frame/g"
+)
+
+func TestApplyBatchUpdatesDifferentCardsWithOneCaseStatement(t *testing.T) {
+	if err := gdb.SetConfig(gdb.Config{
+		"default": gdb.ConfigGroup{{Type: "sqlite", Link: "sqlite::@file(:memory:)"}},
+	}); err != nil {
+		t.Fatal(err)
+	}
+	ctx := context.Background()
+	if _, err := g.DB().Exec(ctx, `CREATE TABLE t_sim_card_info (
+id INTEGER PRIMARY KEY,
+used_flow_mb REAL,
+flow_sync_time DATETIME
+)`); err != nil {
+		t.Fatal(err)
+	}
+	if _, err := g.DB().Exec(ctx, `INSERT INTO t_sim_card_info(id, used_flow_mb) VALUES (1, 0), (2, 0), (3, 9)`); err != nil {
+		t.Fatal(err)
+	}
+	if err := NewRepository().ApplyBatch(ctx, []UsageResult{
+		{CardID: 1, UsageMB: 1.5},
+		{CardID: 2, UsageMB: 2.75},
+	}); err != nil {
+		t.Fatal(err)
+	}
+	var rows []struct {
+		ID         int64   `orm:"id"`
+		UsedFlowMB float64 `orm:"used_flow_mb"`
+	}
+	if err := g.DB().Model("t_sim_card_info").Fields("id,used_flow_mb").OrderAsc("id").Scan(&rows); err != nil {
+		t.Fatal(err)
+	}
+	if len(rows) != 3 || rows[0].UsedFlowMB != 1.5 || rows[1].UsedFlowMB != 2.75 || rows[2].UsedFlowMB != 9 {
+		t.Fatalf("rows = %+v", rows)
+	}
+}