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)
	}
}