package asset

import (
	"context"
	"errors"
	"testing"

	bizasset "kra/internal/biz/asset"
	"kra/internal/data/ledger"

	"github.com/glebarez/sqlite"
	"gorm.io/gorm"
)

type testProvider struct{ db *gorm.DB }

func (p testProvider) DB() *gorm.DB { return p.db }

func TestAdjustmentRejectsInsufficientBalanceWithoutAudit(t *testing.T) {
	db, err := gorm.Open(sqlite.Open("file:"+t.Name()+"?mode=memory&cache=shared"), &gorm.Config{})
	if err != nil {
		t.Fatal(err)
	}
	if err = db.AutoMigrate(&ledger.WalletPO{}, &ledger.BalanceAdjustmentPO{}, &ledger.RechargeOrderPO{}, &ledger.GiftSendPO{}); err != nil {
		t.Fatal(err)
	}
	for _, statement := range []string{
		`CREATE TABLE sys_users (id integer primary key, username text, nick_name text, header_img text, enable integer, created_at datetime, deleted_at datetime)`,
		`CREATE TABLE sys_login_logs (id integer primary key, user_id integer, status numeric, created_at datetime)`,
	} {
		if err = db.Exec(statement).Error; err != nil {
			t.Fatal(err)
		}
	}
	if err = db.Exec(`INSERT INTO sys_users(id,username,nick_name,enable,created_at) VALUES (4,'user','User',1,CURRENT_TIMESTAMP)`).Error; err != nil {
		t.Fatal(err)
	}
	if err = db.Create(&ledger.WalletPO{UserID: 4, DiamondBalance: 20}).Error; err != nil {
		t.Fatal(err)
	}
	repo := NewRepo(testProvider{db: db})
	_, err = repo.Adjust(context.Background(), &bizasset.Adjustment{UserID: 4, OperatorUserID: 1, Type: "decrease", Amount: 30, Reason: "test reason"})
	if !errors.Is(err, bizasset.ErrInsufficient) {
		t.Fatalf("adjustment error = %v", err)
	}
	var wallet ledger.WalletPO
	var audits int64
	db.Where("user_id = ?", 4).First(&wallet)
	db.Model(&ledger.BalanceAdjustmentPO{}).Count(&audits)
	if wallet.DiamondBalance != 20 || audits != 0 {
		t.Fatalf("failed adjustment changed state: wallet=%+v audits=%d", wallet, audits)
	}
}