package lottery

import (
	"context"
	"testing"

	"github.com/DATA-DOG/go-sqlmock"
	"gorm.io/gorm"
)

func TestLedgerService_Reserve_FirstInsertNotExisted(t *testing.T) {
	db, mock, cleanup := newLotteryTestDB(t)
	defer cleanup()

	svc := NewLedgerService()

	mock.ExpectBegin()
	mock.ExpectExec("INSERT INTO `lottery_grant_ledger`").
		WillReturnResult(sqlmock.NewResult(7, 1))
	mock.ExpectQuery("FROM `lottery_grant_ledger`").
		WithArgs("lottery:100:200", 1).
		WillReturnRows(sqlmock.NewRows([]string{"id", "external_ref", "handler_type", "user_id", "activity_id", "draw_id", "amount"}).
			AddRow(int64(7), "lottery:100:200", "vpn_duration", int64(42), int64(100), int64(200), int64(3)))
	mock.ExpectCommit()

	err := db.Transaction(func(tx *gorm.DB) error {
		row, existed, e := svc.Reserve(context.Background(), tx, GrantLedger{
			ExternalRef: "lottery:100:200",
			HandlerType: "vpn_duration",
			UserId:      42,
			ActivityId:  100,
			DrawId:      200,
			Amount:      3,
		})
		if e != nil {
			return e
		}
		if existed {
			t.Fatalf("expected not existed")
		}
		if row.Id != 7 {
			t.Fatalf("expected reloaded id=7, got %d", row.Id)
		}
		return nil
	})
	if err != nil {
		t.Fatalf("tx: %v", err)
	}
	if err := mock.ExpectationsWereMet(); err != nil {
		t.Fatalf("expectations: %v", err)
	}
}

func TestLedgerService_Reserve_DuplicateExisted(t *testing.T) {
	db, mock, cleanup := newLotteryTestDB(t)
	defer cleanup()

	svc := NewLedgerService()

	mock.ExpectBegin()
	mock.ExpectExec("INSERT INTO `lottery_grant_ledger`").
		WillReturnResult(sqlmock.NewResult(0, 0)) // conflict, 0 rows affected
	mock.ExpectQuery("FROM `lottery_grant_ledger`").
		WithArgs("lottery:100:200", 1).
		WillReturnRows(sqlmock.NewRows([]string{"id", "external_ref", "handler_type", "user_id", "activity_id", "draw_id", "amount"}).
			AddRow(int64(9), "lottery:100:200", "vpn_duration", int64(42), int64(100), int64(200), int64(3)))
	mock.ExpectCommit()

	err := db.Transaction(func(tx *gorm.DB) error {
		row, existed, e := svc.Reserve(context.Background(), tx, GrantLedger{
			ExternalRef: "lottery:100:200",
			HandlerType: "vpn_duration",
		})
		if e != nil {
			return e
		}
		if !existed {
			t.Fatalf("expected existed=true when INSERT returns 0 rows affected")
		}
		if row.Id != 9 {
			t.Fatalf("expected stored id=9, got %d", row.Id)
		}
		return nil
	})
	if err != nil {
		t.Fatalf("tx: %v", err)
	}
	if err := mock.ExpectationsWereMet(); err != nil {
		t.Fatalf("expectations: %v", err)
	}
}

func TestLedgerService_Reserve_RequiresTx(t *testing.T) {
	svc := NewLedgerService()
	if _, _, err := svc.Reserve(context.Background(), nil, GrantLedger{ExternalRef: "x"}); err == nil {
		t.Fatalf("expected error when tx is nil")
	}
}

func TestLedgerService_Reserve_RequiresExternalRef(t *testing.T) {
	db, _, cleanup := newLotteryTestDB(t)
	defer cleanup()

	svc := NewLedgerService()
	err := db.Transaction(func(tx *gorm.DB) error {
		_, _, e := svc.Reserve(context.Background(), tx, GrantLedger{})
		return e
	})
	if err == nil {
		t.Fatalf("expected error on empty ExternalRef")
	}
}