package common

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"

	"git.5iot.com/iotlcm-public/iotlcm-openapi/internal/library/batchmetadata"
	"git.5iot.com/iotlcm-public/iotlcm-openapi/internal/model/entity"
)

func TestSaveBatchMainExecutionMetadata(t *testing.T) {
	ctx := context.Background()
	if err := gdb.SetConfig(gdb.Config{"default": gdb.ConfigGroup{{Type: "sqlite", Link: "sqlite::@file(:memory:)", MaxOpenConnCount: 1}}}); err != nil {
		t.Fatal(err)
	}
	t.Cleanup(func() { _ = g.DB().Close(ctx) })
	if err := g.DB().GetCore().ClearTableFieldsAll(ctx); err != nil {
		t.Fatal(err)
	}
	if _, err := g.DB().Exec(ctx, "CREATE TABLE batch_main (id INTEGER PRIMARY KEY AUTOINCREMENT, batch_no TEXT, tenant_id TEXT, dept_id INTEGER, source_channel TEXT, execution_mode INTEGER)"); err != nil {
		t.Fatal(err)
	}
	for _, mode := range []int{batchmetadata.Single, batchmetadata.Bulk} {
		batchNo := "SINGLE"
		if mode == batchmetadata.Bulk {
			batchNo = "BULK"
		}
		if err := g.DB().Transaction(ctx, func(ctx context.Context, tx gdb.TX) error {
			return New().SaveBatchMain(ctx, tx, &entity.BatchMain{BatchNo: batchNo, TenantId: "A", DeptId: 7}, mode)
		}); err != nil {
			t.Fatal(err)
		}
		record, err := g.DB().GetOne(ctx, "SELECT * FROM batch_main WHERE batch_no = ?", batchNo)
		if err != nil || record["source_channel"].String() != "api" || record["execution_mode"].Int() != mode || record["tenant_id"].String() != "A" || record["dept_id"].Int() != 7 {
			t.Fatalf("metadata not persisted: record=%+v err=%v", record, err)
		}
	}
	for _, mode := range []int{-1, batchmetadata.Unknown, 3} {
		if err := g.DB().Transaction(ctx, func(ctx context.Context, tx gdb.TX) error {
			return New().SaveBatchMain(ctx, tx, &entity.BatchMain{BatchNo: "INVALID"}, mode)
		}); err == nil {
			t.Fatalf("invalid creation mode accepted: %d", mode)
		}
	}
	count, err := g.DB().GetValue(ctx, "SELECT COUNT(*) FROM batch_main")
	if err != nil || count.Int() != 2 {
		t.Fatalf("invalid batch inserted: count=%v err=%v", count, err)
	}
}