MiFaZhan
46 lines
1.2 KiB
Go
46 lines
1.2 KiB
Go
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)
|
|
}
|
|
}
|