transaction_test.go

/*
Copyright 2024 Blnk Finance Authors.

Licensed under the Apache License, Version 2.0 (the "License");
you may not use this file except in compliance with the License.
You may obtain a copy of the License at

    http://www.apache.org/licenses/LICENSE-2.0

Unless required by applicable law or agreed to in writing, software
distributed under the License is distributed on an "AS IS" BASIS,
WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
See the License for the specific language governing permissions and
limitations under the License.
*/

package blnk

import (
	"context"
	"encoding/json"
	"fmt"
	"math/big"
	"regexp"
	"sort"
	"strings"
	"sync"
	"testing"
	"time"

	"github.com/blnkfinance/blnk/config"
	"github.com/blnkfinance/blnk/database"
	"github.com/blnkfinance/blnk/database/mocks"
	redlock "github.com/blnkfinance/blnk/internal/lock"
	redis_db "github.com/blnkfinance/blnk/internal/redis-db"
	"github.com/blnkfinance/blnk/model"
	"github.com/go-redis/redismock/v9"
	"github.com/hibiken/asynq"

	"github.com/brianvoe/gofakeit/v6"

	"github.com/DATA-DOG/go-sqlmock"

	"github.com/stretchr/testify/assert"
	"github.com/stretchr/testify/mock"
	"github.com/stretchr/testify/require"
)

func TestAcquireTransactionLocker(t *testing.T) {
	t.Run("fail fast when wait timeout disabled", func(t *testing.T) {
		db, mock := redismock.NewClientMock()
		blnk := &Blnk{
			redis: db,
			config: &config.Configuration{
				Transaction: config.TransactionConfig{
					LockDuration: 5 * time.Second,
				},
			},
		}
		locker := redlock.NewMultiLocker(db, []string{"key-b", "key-a"}, "test-value")

		mock.ExpectSetNX("key-a", "test-value", 5*time.Second).SetVal(true)
		mock.ExpectSetNX("key-b", "test-value", 5*time.Second).SetVal(true)

		err := blnk.acquireTransactionLocker(context.Background(), locker)
		require.NoError(t, err)
		require.NoError(t, mock.ExpectationsWereMet())
	})

	t.Run("waits when configured", func(t *testing.T) {
		db, mock := redismock.NewClientMock()
		blnk := &Blnk{
			redis: db,
			config: &config.Configuration{
				Transaction: config.TransactionConfig{
					LockDuration:    5 * time.Second,
					LockWaitTimeout: time.Second,
				},
			},
		}
		locker := redlock.NewMultiLocker(db, []string{"key-b", "key-a"}, "test-value")

		mock.ExpectSetNX("key-a", "test-value", 5*time.Second).SetVal(false)
		mock.ExpectSetNX("key-a", "test-value", 5*time.Second).SetVal(true)
		mock.ExpectSetNX("key-b", "test-value", 5*time.Second).SetVal(true)

		err := blnk.acquireTransactionLocker(context.Background(), locker)
		require.NoError(t, err)
		require.NoError(t, mock.ExpectationsWereMet())
	})
}

// pollForTransactionStatus polls for a transaction until it reaches the expected status or a timeout occurs.
func pollForTransactionStatus(ctx context.Context, ds database.IDataSource, transactionRef string, expectedStatus string, pollInterval time.Duration, timeoutDuration time.Duration) (*model.Transaction, error) {
	timeoutCtx, cancel := context.WithTimeout(ctx, timeoutDuration)
	defer cancel()

	ticker := time.NewTicker(pollInterval)
	defer ticker.Stop()

	for {
		select {
		case <-timeoutCtx.Done():
			return nil, fmt.Errorf("timed out waiting for transaction ref %s to reach status %s: %w", transactionRef, expectedStatus, timeoutCtx.Err())
		case <-ticker.C:
			txn, err := ds.GetTransactionByRef(timeoutCtx, transactionRef) // Use timeoutCtx for the DB call as well
			if err != nil {
				// If the error is that the transaction is not found, we should continue polling.
				// For other errors, the outer timeout will eventually catch persistent issues.
				continue
			}
			// If err is nil, txn is a valid model.Transaction struct.
			if txn.Status == expectedStatus {
				return &txn, nil // Return a pointer to the transaction
			}
		}
	}
}

func TestRecordTransaction(t *testing.T) {
	cnf := &config.Configuration{
		Redis: config.RedisConfig{
			Dns: "localhost:6379",
		},
		Queue: config.QueueConfig{
			WebhookQueue:     "webhook_queue",
			TransactionQueue: "transaction_queue",
			IndexQueue:       "index_queue",
			NumberOfQueues:   1,
		},
		Server: config.ServerConfig{SecretKey: "some-secret"},
		AccountNumberGeneration: config.AccountNumberGenerationConfig{
			HttpService: config.AccountGenerationHttpService{
				Url: "http://example.com/generateAccount",
			},
		},
	}

	config.ConfigStore.Store(cnf)
	datasource, mock, err := newTestDataSource()
	assert.NoError(t, err)

	// Important: Set ExpectationsWereMet to ensure execution occurs in order of appearance
	mock.MatchExpectationsInOrder(false)

	d, err := NewBlnk(datasource)
	assert.NoError(t, err)

	// Use fixed UUIDs for better predictability
	source := "source-balance-id-123"
	destination := "destination-balance-id-456"
	reference := "transaction-reference-789"

	txn := &model.Transaction{
		Reference:      reference,
		Source:         source,
		Destination:    destination,
		Rate:           1,
		Amount:         10,
		AllowOverdraft: false,
		Precision:      100,
		Currency:       "NGN",
	}

	// First, check if transaction exists
	mock.ExpectQuery(regexp.QuoteMeta(`
        SELECT EXISTS(SELECT 1 FROM blnk.transactions WHERE reference = $1)
    `)).WithArgs(txn.Reference).WillReturnRows(sqlmock.NewRows([]string{"exists"}).AddRow(false))

	// Set up source balance response
	sourceBalanceRows := sqlmock.NewRows([]string{"balance_id", "indicator", "currency", "currency_multiplier", "ledger_id", "balance", "credit_balance", "debit_balance", "inflight_balance", "inflight_credit_balance", "inflight_debit_balance", "created_at", "version", "track_fund_lineage", "allocation_strategy", "identity_id"}).
		AddRow(source, "NGN", "", 1, "ledger-id-source", int64(10000), int64(10000), 0, 0, 0, 0, time.Now(), 0, false, "FIFO", "")

	// Set up destination balance response
	destinationBalanceRows := sqlmock.NewRows([]string{"balance_id", "indicator", "currency", "currency_multiplier", "ledger_id", "balance", "credit_balance", "debit_balance", "inflight_balance", "inflight_credit_balance", "inflight_debit_balance", "created_at", "version", "track_fund_lineage", "allocation_strategy", "identity_id"}).
		AddRow(destination, "", "NGN", 1, "ledger-id-destination", 0, 0, 0, 0, 0, 0, time.Now(), 0, false, "FIFO", "")

	balanceQuery := `SELECT balance_id, indicator, currency, currency_multiplier, ledger_id, balance, credit_balance, debit_balance, inflight_balance, inflight_credit_balance, inflight_debit_balance, created_at, version, track_fund_lineage, COALESCE\(allocation_strategy, 'FIFO'\) as allocation_strategy, COALESCE\(identity_id, ''\) as identity_id FROM blnk.balances WHERE balance_id = \$1`
	balanceQueryPattern := regexp.MustCompile(`\s+`).ReplaceAllString(balanceQuery, `\s*`)

	// Expect balance queries with the source/destination IDs (order can vary)
	mock.ExpectQuery(balanceQueryPattern).WithArgs(source).WillReturnRows(sourceBalanceRows)
	mock.ExpectQuery(balanceQueryPattern).WithArgs(destination).WillReturnRows(destinationBalanceRows)

	// Start atomic transaction for balance updates AND transaction insert
	mock.ExpectBegin()

	// Update source and destination balances (order doesn't matter since MatchExpectationsInOrder is false)
	mock.ExpectExec(regexp.QuoteMeta(`
	  UPDATE blnk.balances
	  SET balance = $2, credit_balance = $3, debit_balance = $4, inflight_balance = $5, inflight_credit_balance = $6, inflight_debit_balance = $7, currency = $8, currency_multiplier = $9, ledger_id = $10, created_at = $11, version = version + 1
	  WHERE balance_id = $1 AND version = $12
	`)).WithArgs(
		source,           // balance_id
		sqlmock.AnyArg(), // balance
		sqlmock.AnyArg(), // credit_balance
		sqlmock.AnyArg(), // debit_balance
		sqlmock.AnyArg(), // inflight_balance
		sqlmock.AnyArg(), // inflight_credit_balance
		sqlmock.AnyArg(), // inflight_debit_balance
		sqlmock.AnyArg(), // currency
		sqlmock.AnyArg(), // currency_multiplier
		sqlmock.AnyArg(), // ledger_id
		sqlmock.AnyArg(), // created_at
		sqlmock.AnyArg(), // version
	).WillReturnResult(sqlmock.NewResult(1, 1))

	mock.ExpectExec(regexp.QuoteMeta(`
	  UPDATE blnk.balances
	  SET balance = $2, credit_balance = $3, debit_balance = $4, inflight_balance = $5, inflight_credit_balance = $6, inflight_debit_balance = $7, currency = $8, currency_multiplier = $9, ledger_id = $10, created_at = $11, version = version + 1
	  WHERE balance_id = $1 AND version = $12
	`)).WithArgs(
		destination,      // balance_id
		sqlmock.AnyArg(), // balance
		sqlmock.AnyArg(), // credit_balance
		sqlmock.AnyArg(), // debit_balance
		sqlmock.AnyArg(), // inflight_balance
		sqlmock.AnyArg(), // inflight_credit_balance
		sqlmock.AnyArg(), // inflight_debit_balance
		sqlmock.AnyArg(), // currency
		sqlmock.AnyArg(), // currency_multiplier
		sqlmock.AnyArg(), // ledger_id
		sqlmock.AnyArg(), // created_at
		sqlmock.AnyArg(), // version
	).WillReturnResult(sqlmock.NewResult(1, 1))

	// Expect transaction insertion within the same atomic transaction
	expectedSQL := `INSERT INTO blnk.transactions(transaction_id, parent_transaction, source, reference, amount, precise_amount, precision, rate, currency, destination, description, status, created_at, meta_data, scheduled_for, hash, effective_date) VALUES ($1, $2, $3, $4, $5, $6, $7, $8, $9, $10, $11, $12, $13, $14, $15, $16, $17)`
	mock.ExpectExec(regexp.QuoteMeta(expectedSQL)).WithArgs(
		sqlmock.AnyArg(), // transaction_id
		sqlmock.AnyArg(), // parent_transaction
		source,           // source
		reference,        // reference
		txn.AmountString, // amount
		"1000",           // precise amount
		txn.Precision,    // precision
		float64(1),       // rate
		txn.Currency,     // currency
		destination,      // destination
		sqlmock.AnyArg(), // description
		sqlmock.AnyArg(), // status
		sqlmock.AnyArg(), // created_at
		sqlmock.AnyArg(), // meta_data
		sqlmock.AnyArg(), // scheduled_for
		sqlmock.AnyArg(), // hash
		sqlmock.AnyArg(), // effective_date
	).WillReturnResult(sqlmock.NewResult(1, 1))

	// Commit atomic transaction (balance updates + transaction insert)
	mock.ExpectCommit()

	// Execute the function being tested
	result, err := d.RecordTransaction(context.Background(), txn)

	// Assert no errors
	assert.NoError(t, err)

	// Check if all expectations were met
	if err := mock.ExpectationsWereMet(); err != nil {
		t.Errorf("there were unfulfilled expectations: %s", err)
	}

	// Optionally check the result if needed
	assert.NotNil(t, result)
}

func TestRecordTransactionWithRate(t *testing.T) {
	cnf := &config.Configuration{
		Redis: config.RedisConfig{
			Dns: "localhost:6379",
		},
		Queue: config.QueueConfig{
			WebhookQueue:     "webhook_queue",
			TransactionQueue: "transaction_queue",
			NumberOfQueues:   1,
		},
		Server: config.ServerConfig{SecretKey: "some-secret"},
		AccountNumberGeneration: config.AccountNumberGenerationConfig{
			HttpService: config.AccountGenerationHttpService{
				Url: "http://example.com/generateAccount",
			},
		},
	}

	config.ConfigStore.Store(cnf)
	datasource, mock, err := newTestDataSource()
	assert.NoError(t, err)

	// Important: Set ExpectationsWereMet to ensure execution occurs in order of appearance
	mock.MatchExpectationsInOrder(false)

	d, err := NewBlnk(datasource)
	assert.NoError(t, err)

	// Use fixed UUIDs for better predictability
	source := "source-balance-id-123"
	destination := "destination-balance-id-456"
	reference := "transaction-reference-789"

	txn := &model.Transaction{
		Reference:      reference,
		Source:         source,
		Destination:    destination,
		Amount:         1000000,
		Rate:           1300,
		AllowOverdraft: true,
		Precision:      100,
		Currency:       "NGN",
	}

	// First, check if transaction exists
	mock.ExpectQuery(regexp.QuoteMeta(`
        SELECT EXISTS(SELECT 1 FROM blnk.transactions WHERE reference = $1)
    `)).WithArgs(txn.Reference).WillReturnRows(sqlmock.NewRows([]string{"exists"}).AddRow(false))

	// Set up source balance response - Note this is USD
	sourceBalanceRows := sqlmock.NewRows([]string{"balance_id", "indicator", "currency", "currency_multiplier", "ledger_id", "balance", "credit_balance", "debit_balance", "inflight_balance", "inflight_credit_balance", "inflight_debit_balance", "created_at", "version", "track_fund_lineage", "allocation_strategy", "identity_id"}).
		AddRow(source, "", "USD", 1, "ledger-id-source", 0, 0, 0, 0, 0, 0, time.Now(), 0, false, "FIFO", "")

	// Set up destination balance response - This is NGN
	destinationBalanceRows := sqlmock.NewRows([]string{"balance_id", "indicator", "currency", "currency_multiplier", "ledger_id", "balance", "credit_balance", "debit_balance", "inflight_balance", "inflight_credit_balance", "inflight_debit_balance", "created_at", "version", "track_fund_lineage", "allocation_strategy", "identity_id"}).
		AddRow(destination, "", "NGN", 1, "ledger-id-destination", 0, 0, 0, 0, 0, 0, time.Now(), 0, false, "FIFO", "")

	balanceQuery := `SELECT balance_id, indicator, currency, currency_multiplier, ledger_id, balance, credit_balance, debit_balance, inflight_balance, inflight_credit_balance, inflight_debit_balance, created_at, version, track_fund_lineage, COALESCE\(allocation_strategy, 'FIFO'\) as allocation_strategy, COALESCE\(identity_id, ''\) as identity_id FROM blnk.balances WHERE balance_id = \$1`
	balanceQueryPattern := regexp.MustCompile(`\s+`).ReplaceAllString(balanceQuery, `\s*`)

	// Expect balance queries with the source/destination IDs (order can vary)
	mock.ExpectQuery(balanceQueryPattern).WithArgs(source).WillReturnRows(sourceBalanceRows)
	mock.ExpectQuery(balanceQueryPattern).WithArgs(destination).WillReturnRows(destinationBalanceRows)

	// Start atomic transaction for balance updates AND transaction insert
	mock.ExpectBegin()

	// Update source balance
	mock.ExpectExec(regexp.QuoteMeta(`
        UPDATE blnk.balances
        SET balance = $2, credit_balance = $3, debit_balance = $4, inflight_balance = $5, inflight_credit_balance = $6, inflight_debit_balance = $7, currency = $8, currency_multiplier = $9, ledger_id = $10, created_at = $11, version = version + 1
        WHERE balance_id = $1 AND version = $12
    `)).WithArgs(
		source,                          // $1
		big.NewInt(-100000000).String(), // $2
		big.NewInt(0).String(),          // $3
		big.NewInt(100000000).String(),  // $4
		big.NewInt(0).String(),          // $5
		big.NewInt(0).String(),          // $6
		big.NewInt(0).String(),          // $7
		"USD",                           // $8
		float64(1),                      // $9
		"ledger-id-source",              // $10
		sqlmock.AnyArg(),                // $11
		0,                               // $12
	).WillReturnResult(sqlmock.NewResult(1, 1))

	// Update destination balance
	mock.ExpectExec(regexp.QuoteMeta(`
        UPDATE blnk.balances
        SET balance = $2, credit_balance = $3, debit_balance = $4, inflight_balance = $5, inflight_credit_balance = $6, inflight_debit_balance = $7, currency = $8, currency_multiplier = $9, ledger_id = $10, created_at = $11, version = version + 1
        WHERE balance_id = $1 AND version = $12
    `)).WithArgs(
		destination,                       // $1
		big.NewInt(130000000000).String(), // $2
		big.NewInt(130000000000).String(), // $3
		big.NewInt(0).String(),            // $4
		big.NewInt(0).String(),            // $5
		big.NewInt(0).String(),            // $6
		big.NewInt(0).String(),            // $7
		"NGN",                             // $8
		float64(1),                        // $9
		"ledger-id-destination",           // $10
		sqlmock.AnyArg(),                  // $11
		0,                                 // $12
	).WillReturnResult(sqlmock.NewResult(1, 1))

	// Expect transaction insertion within the same atomic transaction
	expectedSQL := `INSERT INTO blnk.transactions(transaction_id, parent_transaction, source, reference, amount, precise_amount, precision, rate, currency, destination, description, status, created_at, meta_data, scheduled_for, hash, effective_date) VALUES ($1, $2, $3, $4, $5, $6, $7, $8, $9, $10, $11, $12, $13, $14, $15, $16, $17)`
	mock.ExpectExec(regexp.QuoteMeta(expectedSQL)).WithArgs(
		sqlmock.AnyArg(), // transaction_id
		sqlmock.AnyArg(), // parent_transaction
		source,           // source
		reference,        // reference
		txn.AmountString, // amount
		"100000000",      // precise amount
		txn.Precision,    // precision
		float64(1300),    // rate
		txn.Currency,     // currency
		destination,      // destination
		sqlmock.AnyArg(), // description
		sqlmock.AnyArg(), // status
		sqlmock.AnyArg(), // created_at
		sqlmock.AnyArg(), // meta_data
		sqlmock.AnyArg(), // scheduled_for
		sqlmock.AnyArg(), // hash
		sqlmock.AnyArg(), // effective_date
	).WillReturnResult(sqlmock.NewResult(1, 1))

	// Commit atomic transaction (balance updates + transaction insert)
	mock.ExpectCommit()

	// Execute the function being tested
	_, err = d.RecordTransaction(context.Background(), txn)

	// Assert no errors
	assert.NoError(t, err)

	// Check if all expectations were met
	if err := mock.ExpectationsWereMet(); err != nil {
		t.Errorf("there were unfulfilled expectations: %s", err)
	}
}

func TestRecordTransaction_AtomicRollback_InsertFails(t *testing.T) {
	cnf := &config.Configuration{
		Redis: config.RedisConfig{
			Dns: "localhost:6379",
		},
		Queue: config.QueueConfig{
			WebhookQueue:     "webhook_queue",
			TransactionQueue: "transaction_queue",
			IndexQueue:       "index_queue",
			NumberOfQueues:   1,
		},
		Server: config.ServerConfig{SecretKey: "some-secret"},
		AccountNumberGeneration: config.AccountNumberGenerationConfig{
			HttpService: config.AccountGenerationHttpService{
				Url: "http://example.com/generateAccount",
			},
		},
	}

	config.ConfigStore.Store(cnf)
	datasource, mock, err := newTestDataSource()
	assert.NoError(t, err)

	mock.MatchExpectationsInOrder(false)

	d, err := NewBlnk(datasource)
	assert.NoError(t, err)

	source := "source-balance-id-atomicity-test"
	destination := "destination-balance-id-atomicity-test"
	reference := "transaction-reference-atomicity-test"

	txn := &model.Transaction{
		Reference:      reference,
		Source:         source,
		Destination:    destination,
		Rate:           1,
		Amount:         100,
		AllowOverdraft: true,
		Precision:      100,
		Currency:       "NGN",
	}

	mock.ExpectQuery(regexp.QuoteMeta(`
        SELECT EXISTS(SELECT 1 FROM blnk.transactions WHERE reference = $1)
    `)).WithArgs(txn.Reference).WillReturnRows(sqlmock.NewRows([]string{"exists"}).AddRow(false))

	sourceBalanceRows := sqlmock.NewRows([]string{"balance_id", "indicator", "currency", "currency_multiplier", "ledger_id", "balance", "credit_balance", "debit_balance", "inflight_balance", "inflight_credit_balance", "inflight_debit_balance", "created_at", "version", "track_fund_lineage", "allocation_strategy", "identity_id"}).
		AddRow(source, "", "NGN", 1, "ledger-id-source", int64(100000), int64(100000), 0, 0, 0, 0, time.Now(), 0, false, "FIFO", "")

	destinationBalanceRows := sqlmock.NewRows([]string{"balance_id", "indicator", "currency", "currency_multiplier", "ledger_id", "balance", "credit_balance", "debit_balance", "inflight_balance", "inflight_credit_balance", "inflight_debit_balance", "created_at", "version", "track_fund_lineage", "allocation_strategy", "identity_id"}).
		AddRow(destination, "", "NGN", 1, "ledger-id-destination", 0, 0, 0, 0, 0, 0, time.Now(), 0, false, "FIFO", "")

	balanceQuery := `SELECT balance_id, indicator, currency, currency_multiplier, ledger_id, balance, credit_balance, debit_balance, inflight_balance, inflight_credit_balance, inflight_debit_balance, created_at, version, track_fund_lineage, COALESCE\(allocation_strategy, 'FIFO'\) as allocation_strategy, COALESCE\(identity_id, ''\) as identity_id FROM blnk.balances WHERE balance_id = \$1`
	balanceQueryPattern := regexp.MustCompile(`\s+`).ReplaceAllString(balanceQuery, `\s*`)

	mock.ExpectQuery(balanceQueryPattern).WithArgs(source).WillReturnRows(sourceBalanceRows)
	mock.ExpectQuery(balanceQueryPattern).WithArgs(destination).WillReturnRows(destinationBalanceRows)

	mock.ExpectBegin()

	mock.ExpectExec(regexp.QuoteMeta(`
	  UPDATE blnk.balances
	  SET balance = $2, credit_balance = $3, debit_balance = $4, inflight_balance = $5, inflight_credit_balance = $6, inflight_debit_balance = $7, currency = $8, currency_multiplier = $9, ledger_id = $10, created_at = $11, version = version + 1
	  WHERE balance_id = $1 AND version = $12
	`)).WithArgs(
		source,
		sqlmock.AnyArg(), sqlmock.AnyArg(), sqlmock.AnyArg(), sqlmock.AnyArg(),
		sqlmock.AnyArg(), sqlmock.AnyArg(), sqlmock.AnyArg(), sqlmock.AnyArg(),
		sqlmock.AnyArg(), sqlmock.AnyArg(), sqlmock.AnyArg(),
	).WillReturnResult(sqlmock.NewResult(1, 1))

	mock.ExpectExec(regexp.QuoteMeta(`
	  UPDATE blnk.balances
	  SET balance = $2, credit_balance = $3, debit_balance = $4, inflight_balance = $5, inflight_credit_balance = $6, inflight_debit_balance = $7, currency = $8, currency_multiplier = $9, ledger_id = $10, created_at = $11, version = version + 1
	  WHERE balance_id = $1 AND version = $12
	`)).WithArgs(
		destination,
		sqlmock.AnyArg(), sqlmock.AnyArg(), sqlmock.AnyArg(), sqlmock.AnyArg(),
		sqlmock.AnyArg(), sqlmock.AnyArg(), sqlmock.AnyArg(), sqlmock.AnyArg(),
		sqlmock.AnyArg(), sqlmock.AnyArg(), sqlmock.AnyArg(),
	).WillReturnResult(sqlmock.NewResult(1, 1))

	expectedSQL := `INSERT INTO blnk.transactions(transaction_id, parent_transaction, source, reference, amount, precise_amount, precision, rate, currency, destination, description, status, created_at, meta_data, scheduled_for, hash, effective_date) VALUES ($1, $2, $3, $4, $5, $6, $7, $8, $9, $10, $11, $12, $13, $14, $15, $16, $17)`
	mock.ExpectExec(regexp.QuoteMeta(expectedSQL)).WithArgs(
		sqlmock.AnyArg(), sqlmock.AnyArg(), source, reference,
		sqlmock.AnyArg(), sqlmock.AnyArg(), sqlmock.AnyArg(), sqlmock.AnyArg(),
		sqlmock.AnyArg(), destination, sqlmock.AnyArg(), sqlmock.AnyArg(),
		sqlmock.AnyArg(), sqlmock.AnyArg(), sqlmock.AnyArg(), sqlmock.AnyArg(), sqlmock.AnyArg(),
	).WillReturnError(fmt.Errorf("database connection lost - simulated failure"))

	mock.ExpectRollback()

	_, err = d.RecordTransaction(context.Background(), txn)

	assert.Error(t, err)
	assert.Contains(t, err.Error(), "failed to persist transaction with balances")

	if err := mock.ExpectationsWereMet(); err != nil {
		t.Errorf("there were unfulfilled expectations: %s", err)
	}
}

func TestRecordTransaction_AtomicRollback_SecondBalanceUpdateFails(t *testing.T) {
	cnf := &config.Configuration{
		Redis: config.RedisConfig{
			Dns: "localhost:6379",
		},
		Queue: config.QueueConfig{
			WebhookQueue:     "webhook_queue",
			TransactionQueue: "transaction_queue",
			IndexQueue:       "index_queue",
			NumberOfQueues:   1,
		},
		Server: config.ServerConfig{SecretKey: "some-secret"},
		AccountNumberGeneration: config.AccountNumberGenerationConfig{
			HttpService: config.AccountGenerationHttpService{
				Url: "http://example.com/generateAccount",
			},
		},
	}

	config.ConfigStore.Store(cnf)
	datasource, mock, err := newTestDataSource()
	assert.NoError(t, err)

	mock.MatchExpectationsInOrder(true)

	d, err := NewBlnk(datasource)
	assert.NoError(t, err)

	source := "source-balance-id-atomicity-test-2"
	destination := "destination-balance-id-atomicity-test-2"
	reference := "transaction-reference-atomicity-test-2"

	txn := &model.Transaction{
		Reference:      reference,
		Source:         source,
		Destination:    destination,
		Rate:           1,
		Amount:         100,
		AllowOverdraft: true,
		Precision:      100,
		Currency:       "NGN",
	}

	mock.ExpectQuery(regexp.QuoteMeta(`
        SELECT EXISTS(SELECT 1 FROM blnk.transactions WHERE reference = $1)
    `)).WithArgs(txn.Reference).WillReturnRows(sqlmock.NewRows([]string{"exists"}).AddRow(false))

	sourceBalanceRows := sqlmock.NewRows([]string{"balance_id", "indicator", "currency", "currency_multiplier", "ledger_id", "balance", "credit_balance", "debit_balance", "inflight_balance", "inflight_credit_balance", "inflight_debit_balance", "created_at", "version", "track_fund_lineage", "allocation_strategy", "identity_id"}).
		AddRow(source, "", "NGN", 1, "ledger-id-source", int64(100000), int64(100000), 0, 0, 0, 0, time.Now(), 0, false, "FIFO", "")

	destinationBalanceRows := sqlmock.NewRows([]string{"balance_id", "indicator", "currency", "currency_multiplier", "ledger_id", "balance", "credit_balance", "debit_balance", "inflight_balance", "inflight_credit_balance", "inflight_debit_balance", "created_at", "version", "track_fund_lineage", "allocation_strategy", "identity_id"}).
		AddRow(destination, "", "NGN", 1, "ledger-id-destination", 0, 0, 0, 0, 0, 0, time.Now(), 0, false, "FIFO", "")

	balanceQuery := `SELECT balance_id, indicator, currency, currency_multiplier, ledger_id, balance, credit_balance, debit_balance, inflight_balance, inflight_credit_balance, inflight_debit_balance, created_at, version, track_fund_lineage, COALESCE\(allocation_strategy, 'FIFO'\) as allocation_strategy, COALESCE\(identity_id, ''\) as identity_id FROM blnk.balances WHERE balance_id = \$1`
	balanceQueryPattern := regexp.MustCompile(`\s+`).ReplaceAllString(balanceQuery, `\s*`)

	mock.ExpectQuery(balanceQueryPattern).WithArgs(source).WillReturnRows(sourceBalanceRows)
	mock.ExpectQuery(balanceQueryPattern).WithArgs(destination).WillReturnRows(destinationBalanceRows)

	mock.ExpectBegin()

	mock.ExpectExec(regexp.QuoteMeta(`
        UPDATE blnk.balances
        SET balance = $2, credit_balance = $3, debit_balance = $4, inflight_balance = $5, inflight_credit_balance = $6, inflight_debit_balance = $7, currency = $8, currency_multiplier = $9, ledger_id = $10, created_at = $11, version = version + 1
        WHERE balance_id = $1 AND version = $12
    `)).WithArgs(
		source,
		sqlmock.AnyArg(), sqlmock.AnyArg(), sqlmock.AnyArg(), sqlmock.AnyArg(),
		sqlmock.AnyArg(), sqlmock.AnyArg(), sqlmock.AnyArg(), sqlmock.AnyArg(),
		sqlmock.AnyArg(), sqlmock.AnyArg(), sqlmock.AnyArg(),
	).WillReturnResult(sqlmock.NewResult(1, 1))

	mock.ExpectExec(regexp.QuoteMeta(`
        UPDATE blnk.balances
        SET balance = $2, credit_balance = $3, debit_balance = $4, inflight_balance = $5, inflight_credit_balance = $6, inflight_debit_balance = $7, currency = $8, currency_multiplier = $9, ledger_id = $10, created_at = $11, version = version + 1
        WHERE balance_id = $1 AND version = $12
    `)).WithArgs(
		destination,
		sqlmock.AnyArg(), sqlmock.AnyArg(), sqlmock.AnyArg(), sqlmock.AnyArg(),
		sqlmock.AnyArg(), sqlmock.AnyArg(), sqlmock.AnyArg(), sqlmock.AnyArg(),
		sqlmock.AnyArg(), sqlmock.AnyArg(), sqlmock.AnyArg(),
	).WillReturnError(fmt.Errorf("optimistic lock failure - balance was modified by another transaction"))

	mock.ExpectRollback()

	_, err = d.RecordTransaction(context.Background(), txn)

	assert.Error(t, err)
	assert.Contains(t, err.Error(), "failed to persist transaction with balances")

	if err := mock.ExpectationsWereMet(); err != nil {
		t.Errorf("there were unfulfilled expectations: %s", err)
	}
}

func TestRecordTransaction_AtomicRollback_FirstBalanceUpdateFails(t *testing.T) {
	cnf := &config.Configuration{
		Redis: config.RedisConfig{
			Dns: "localhost:6379",
		},
		Queue: config.QueueConfig{
			WebhookQueue:     "webhook_queue",
			TransactionQueue: "transaction_queue",
			IndexQueue:       "index_queue",
			NumberOfQueues:   1,
		},
		Server: config.ServerConfig{SecretKey: "some-secret"},
		AccountNumberGeneration: config.AccountNumberGenerationConfig{
			HttpService: config.AccountGenerationHttpService{
				Url: "http://example.com/generateAccount",
			},
		},
	}

	config.ConfigStore.Store(cnf)
	datasource, mock, err := newTestDataSource()
	assert.NoError(t, err)

	mock.MatchExpectationsInOrder(true)

	d, err := NewBlnk(datasource)
	assert.NoError(t, err)

	source := "source-balance-id-atomicity-test-3"
	destination := "destination-balance-id-atomicity-test-3"
	reference := "transaction-reference-atomicity-test-3"

	txn := &model.Transaction{
		Reference:      reference,
		Source:         source,
		Destination:    destination,
		Rate:           1,
		Amount:         100,
		AllowOverdraft: true,
		Precision:      100,
		Currency:       "NGN",
	}

	mock.ExpectQuery(regexp.QuoteMeta(`
        SELECT EXISTS(SELECT 1 FROM blnk.transactions WHERE reference = $1)
    `)).WithArgs(txn.Reference).WillReturnRows(sqlmock.NewRows([]string{"exists"}).AddRow(false))

	sourceBalanceRows := sqlmock.NewRows([]string{"balance_id", "indicator", "currency", "currency_multiplier", "ledger_id", "balance", "credit_balance", "debit_balance", "inflight_balance", "inflight_credit_balance", "inflight_debit_balance", "created_at", "version", "track_fund_lineage", "allocation_strategy", "identity_id"}).
		AddRow(source, "", "NGN", 1, "ledger-id-source", int64(100000), int64(100000), 0, 0, 0, 0, time.Now(), 0, false, "FIFO", "")

	destinationBalanceRows := sqlmock.NewRows([]string{"balance_id", "indicator", "currency", "currency_multiplier", "ledger_id", "balance", "credit_balance", "debit_balance", "inflight_balance", "inflight_credit_balance", "inflight_debit_balance", "created_at", "version", "track_fund_lineage", "allocation_strategy", "identity_id"}).
		AddRow(destination, "", "NGN", 1, "ledger-id-destination", 0, 0, 0, 0, 0, 0, time.Now(), 0, false, "FIFO", "")

	balanceQuery := `SELECT balance_id, indicator, currency, currency_multiplier, ledger_id, balance, credit_balance, debit_balance, inflight_balance, inflight_credit_balance, inflight_debit_balance, created_at, version, track_fund_lineage, COALESCE\(allocation_strategy, 'FIFO'\) as allocation_strategy, COALESCE\(identity_id, ''\) as identity_id FROM blnk.balances WHERE balance_id = \$1`
	balanceQueryPattern := regexp.MustCompile(`\s+`).ReplaceAllString(balanceQuery, `\s*`)

	mock.ExpectQuery(balanceQueryPattern).WithArgs(source).WillReturnRows(sourceBalanceRows)
	mock.ExpectQuery(balanceQueryPattern).WithArgs(destination).WillReturnRows(destinationBalanceRows)

	mock.ExpectBegin()

	mock.ExpectExec(regexp.QuoteMeta(`
        UPDATE blnk.balances
        SET balance = $2, credit_balance = $3, debit_balance = $4, inflight_balance = $5, inflight_credit_balance = $6, inflight_debit_balance = $7, currency = $8, currency_multiplier = $9, ledger_id = $10, created_at = $11, version = version + 1
        WHERE balance_id = $1 AND version = $12
    `)).WithArgs(
		source,
		sqlmock.AnyArg(), sqlmock.AnyArg(), sqlmock.AnyArg(), sqlmock.AnyArg(),
		sqlmock.AnyArg(), sqlmock.AnyArg(), sqlmock.AnyArg(), sqlmock.AnyArg(),
		sqlmock.AnyArg(), sqlmock.AnyArg(), sqlmock.AnyArg(),
	).WillReturnError(fmt.Errorf("database write error - disk full"))

	mock.ExpectRollback()

	_, err = d.RecordTransaction(context.Background(), txn)

	assert.Error(t, err)
	assert.Contains(t, err.Error(), "failed to persist transaction with balances")

	if err := mock.ExpectationsWereMet(); err != nil {
		t.Errorf("there were unfulfilled expectations: %s", err)
	}
}

func TestVoidInflightTransaction_Negative(t *testing.T) {
	datasource, mock, err := newTestDataSource()
	assert.NoError(t, err)

	d, err := NewBlnk(datasource)
	assert.NoError(t, err)

	source := gofakeit.UUID()
	destination := gofakeit.UUID()
	metaDataJSON, _ := json.Marshal(map[string]interface{}{"key": "value"})
	t.Run("Transaction not in INFLIGHT status", func(t *testing.T) {
		transactionID := gofakeit.UUID()

		mock.ExpectQuery(regexp.QuoteMeta(`SELECT transaction_id, source, reference, amount, precise_amount, precision, currency, destination, description, status, created_at, meta_data, parent_transaction, hash FROM blnk.transactions WHERE transaction_id = $1`)).
			WithArgs(transactionID).
			WillReturnRows(sqlmock.NewRows([]string{"transaction_id", "source", "reference", "amount", "precise_amount", "precision", "currency", "destination", "description", "status", "created_at", "meta_data", "parent_transaction", "hash"}).
				AddRow(transactionID, source, gofakeit.UUID(), 100.0, 10000, 100, "USD", destination, gofakeit.UUID(), "APPLIED", time.Now(), metaDataJSON, "", ""))

		_, err := d.VoidInflightTransaction(context.Background(), transactionID)
		assert.Error(t, err)
		assert.Contains(t, err.Error(), "transaction is not in inflight status")
	})

	t.Run("Transaction already voided", func(t *testing.T) {
		transactionID := gofakeit.UUID()

		mock.ExpectQuery(regexp.QuoteMeta(`SELECT transaction_id, source, reference, amount, precise_amount, precision, currency, destination, description, status, created_at, meta_data, parent_transaction, hash FROM blnk.transactions WHERE transaction_id = $1`)).
			WithArgs(transactionID).
			WillReturnRows(sqlmock.NewRows([]string{"transaction_id", "source", "reference", "amount", "precise_amount", "precision", "currency", "destination", "description", "status", "created_at", "meta_data", "parent_transaction", "hash"}).
				AddRow(transactionID, source, gofakeit.UUID(), 100.0, 10000, 100, "USD", destination, gofakeit.UUID(), "INFLIGHT", time.Now(), metaDataJSON, "", ""))

		// Mock IsParentTransactionVoid
		mock.ExpectQuery(regexp.QuoteMeta(`SELECT EXISTS ( SELECT 1 FROM blnk.transactions WHERE parent_transaction = $1 AND status = 'VOID' )`)).
			WithArgs(transactionID).
			WillReturnRows(sqlmock.NewRows([]string{"exists"}).AddRow(true))

		_, err := d.VoidInflightTransaction(context.Background(), transactionID)
		assert.Error(t, err)
		assert.Contains(t, err.Error(), "transaction has already been voided")
	})

	// Verify all expectations were met
	if err := mock.ExpectationsWereMet(); err != nil {
		t.Errorf("there were unfulfilled expectations: %s", err)
	}
}

func TestQueueTransactionFlow(t *testing.T) {
	// Skip in short mode as this is a long-running test
	if testing.Short() {
		t.Skip("Skipping queue flow test in short mode")
	}

	// Initialize test context
	ctx := context.Background()

	// Setup real test configuration
	cnf := &config.Configuration{
		Redis: config.RedisConfig{
			Dns: "localhost:6379",
		},
		DataSource: config.DataSourceConfig{
			Dns: "postgres://postgres:password@localhost:5432/blnk?sslmode=disable",
		},
		Queue: config.QueueConfig{
			WebhookQueue:     "webhook_queue_test",
			IndexQueue:       "index_queue_test",
			TransactionQueue: "transaction_queue_test",
			NumberOfQueues:   1,
		},
		Server: config.ServerConfig{
			SecretKey: "test-secret",
		},
		Transaction: config.TransactionConfig{
			BatchSize:        100,
			MaxQueueSize:     1000,
			LockDuration:     time.Second * 30,
			IndexQueuePrefix: "test_index",
		},
	}
	config.ConfigStore.Store(cnf)

	// Create real datasource for integration test
	ds, err := database.NewDataSource(cnf)
	require.NoError(t, err, "Failed to create datasource")

	// Create real Blnk instance
	blnk, err := NewBlnk(ds)
	require.NoError(t, err, "Failed to create Blnk instance")

	txnRef := "txn_" + model.GenerateUUIDWithSuffix("test")

	// Create source balance
	sourceBalance := &model.Balance{
		Currency: "USD",
		LedgerID: "general_ledger_id",
	}

	// Create destination balance
	destBalance := &model.Balance{
		Currency: "USD",
		LedgerID: "general_ledger_id",
	}

	// Create balances in database
	source, err := ds.CreateBalance(*sourceBalance)
	require.NoError(t, err, "Failed to create source balance")

	dest, err := ds.CreateBalance(*destBalance)
	require.NoError(t, err, "Failed to create destination balance")

	sourceID := source.BalanceID
	destID := dest.BalanceID

	// Create transaction
	txn := &model.Transaction{
		Reference:      txnRef,
		Source:         sourceID,
		Destination:    destID,
		Amount:         500, // $5.00
		Currency:       "USD",
		AllowOverdraft: true,
		Precision:      100,
		MetaData:       map[string]interface{}{"test": true},
	}

	// Queue the transaction
	queuedTxn, err := blnk.QueueTransaction(ctx, txn)
	require.NoError(t, err, "Failed to queue transaction")

	// Verify initial transaction state
	require.Equal(t, StatusQueued, queuedTxn.Status, "Initial transaction should be QUEUED")

	// Store the original transaction ID
	originalTxnID := queuedTxn.TransactionID

	queueCopy := createQueueCopy(queuedTxn, queuedTxn.Reference)
	_, err = blnk.RecordTransaction(ctx, queueCopy)
	assert.NoError(t, err)

	ref := fmt.Sprintf("%s_%s", txnRef, "q")
	// Verify the processed transaction
	appliedTxn, err := ds.GetTransactionByRef(ctx, ref)
	require.NoError(t, err, "Failed to get processed transactions")
	require.NotEmpty(t, appliedTxn, "Should have processed transactions")

	// Verify the applied transaction
	require.NotNil(t, appliedTxn, "Should have an APPLIED transaction")
	require.Equal(t, originalTxnID, appliedTxn.ParentTransaction, "Applied transaction should reference original transaction")
	require.Equal(t, txn.Amount, appliedTxn.Amount, "Amount should match")

	// Verify final balance states
	updatedSource, err := ds.GetBalanceByIDLite(sourceID)
	require.NoError(t, err, "Failed to get updated source balance")

	updatedDest, err := ds.GetBalanceByIDLite(destID)
	require.NoError(t, err, "Failed to get updated destination balance")

	fmt.Println("Updated Source Balance: ", updatedSource.Balance)
	fmt.Println("Updated Destination Balance: ", updatedDest.Balance)

	require.Equal(t, 0, updatedSource.Balance.Cmp(updatedSource.Balance),
		"Source balance should be reduced by transaction amount")
	require.Equal(t, 0, updatedDest.Balance.Cmp(updatedDest.Balance),
		"Destination balance should be increased by transaction amount")
}

func TestQueueTransactionFlowWithSkipQueue(t *testing.T) {
	// Skip in short mode
	if testing.Short() {
		t.Skip("Skipping queue flow test in short mode")
	}

	ctx := context.Background()
	cnf := &config.Configuration{
		Redis: config.RedisConfig{
			Dns: "localhost:6379",
		},
		DataSource: config.DataSourceConfig{
			Dns: "postgres://postgres:password@localhost:5432/blnk?sslmode=disable",
		},
		Queue: config.QueueConfig{
			WebhookQueue:     "webhook_queue_test",
			IndexQueue:       "index_queue_test",
			TransactionQueue: "transaction_queue_test",
			NumberOfQueues:   1,
		},
		Server: config.ServerConfig{
			SecretKey: "test-secret",
		},
		Transaction: config.TransactionConfig{
			BatchSize:        100,
			MaxQueueSize:     1000,
			LockDuration:     time.Second * 30,
			IndexQueuePrefix: "test_index",
		},
	}
	config.ConfigStore.Store(cnf)

	ds, err := database.NewDataSource(cnf)
	require.NoError(t, err, "Failed to create datasource")

	blnk, err := NewBlnk(ds)
	require.NoError(t, err, "Failed to create Blnk instance")

	txnRef := "txn_" + model.GenerateUUIDWithSuffix("test")

	// Create test balances
	sourceBalance := &model.Balance{
		Currency: "USD",
		LedgerID: "general_ledger_id",
	}
	destBalance := &model.Balance{
		Currency: "USD",
		LedgerID: "general_ledger_id",
	}

	source, err := ds.CreateBalance(*sourceBalance)
	require.NoError(t, err, "Failed to create source balance")

	dest, err := ds.CreateBalance(*destBalance)
	require.NoError(t, err, "Failed to create destination balance")

	// Create transaction with skip_queue set to true
	txn := &model.Transaction{
		Reference:      txnRef,
		Source:         source.BalanceID,
		Destination:    dest.BalanceID,
		Amount:         500,
		Currency:       "USD",
		AllowOverdraft: true,
		Precision:      100,
		MetaData:       map[string]interface{}{"test": true},
		SkipQueue:      true, // Enable skip queue
	}

	// Queue the transaction
	queuedTxn, err := blnk.QueueTransaction(ctx, txn)
	require.NoError(t, err, "Failed to queue transaction")

	// Verify that the transaction was processed immediately
	require.Equal(t, StatusApplied, queuedTxn.Status, "Transaction should be APPLIED immediately when skip_queue is true")

	// Verify balances were updated immediately
	updatedSource, err := ds.GetBalanceByIDLite(source.BalanceID)
	require.NoError(t, err, "Failed to get updated source balance")

	updatedDest, err := ds.GetBalanceByIDLite(dest.BalanceID)
	require.NoError(t, err, "Failed to get updated destination balance")

	// Calculate expected balance changes
	expectedDebit := big.NewInt(int64(-500) * 100) // Amount * precision
	expectedCredit := big.NewInt(int64(500) * 100)

	// Verify balance changes
	require.Equal(t, 0, updatedSource.Balance.Cmp(expectedDebit),
		"Source balance should be immediately reduced by transaction amount")
	require.Equal(t, 0, updatedDest.Balance.Cmp(expectedCredit),
		"Destination balance should be immediately increased by transaction amount")

	// Verify no queued entry exists
	queuedEntry, err := ds.GetTransactionByRef(ctx, txnRef)
	require.NoError(t, err, "Failed to get queued transaction entry")
	require.Equal(t, StatusApplied, queuedEntry.Status, "Should have an APPLIED transaction entry")
}

func TestInflightTransactionFlowWithSkipQueue(t *testing.T) {
	// Skip in short mode
	if testing.Short() {
		t.Skip("Skipping queue flow test in short mode")
	}

	ctx := context.Background()
	cnf := &config.Configuration{
		Redis: config.RedisConfig{
			Dns: "localhost:6379",
		},
		DataSource: config.DataSourceConfig{
			Dns: "postgres://postgres:password@localhost:5432/blnk?sslmode=disable",
		},
		Queue: config.QueueConfig{
			WebhookQueue:     "webhook_queue_test",
			IndexQueue:       "index_queue_test",
			TransactionQueue: "transaction_queue_test",
			NumberOfQueues:   1,
		},
		Server: config.ServerConfig{
			SecretKey: "test-secret",
		},
		Transaction: config.TransactionConfig{
			BatchSize:        100,
			MaxQueueSize:     1000,
			LockDuration:     time.Second * 30,
			IndexQueuePrefix: "test_index",
		},
	}
	config.ConfigStore.Store(cnf)

	ds, err := database.NewDataSource(cnf)
	require.NoError(t, err, "Failed to create datasource")

	blnk, err := NewBlnk(ds)
	require.NoError(t, err, "Failed to create Blnk instance")

	txnRef := "txn_" + model.GenerateUUIDWithSuffix("test")

	// Create test balances
	sourceBalance := &model.Balance{
		Currency: "USD",
		LedgerID: "general_ledger_id",
	}
	destBalance := &model.Balance{
		Currency: "USD",
		LedgerID: "general_ledger_id",
	}

	source, err := ds.CreateBalance(*sourceBalance)
	require.NoError(t, err, "Failed to create source balance")

	dest, err := ds.CreateBalance(*destBalance)
	require.NoError(t, err, "Failed to create destination balance")

	// Create transaction with skip_queue set to true
	txn := &model.Transaction{
		Reference:      txnRef,
		Source:         source.BalanceID,
		Destination:    dest.BalanceID,
		Amount:         500,
		Inflight:       true,
		Currency:       "USD",
		AllowOverdraft: true,
		Precision:      100,
		MetaData:       map[string]interface{}{"test": true},
		SkipQueue:      true, // Enable skip queue
	}

	// Queue the transaction
	queuedTxn, err := blnk.QueueTransaction(ctx, txn)
	require.NoError(t, err, "Failed to queue transaction")

	// Verify that the transaction was processed immediately
	require.Equal(t, StatusInflight, queuedTxn.Status, "Transaction should be INFLIGHT immediately when skip_queue is true")

	// Verify balances were updated immediately
	updatedSource, err := ds.GetBalanceByIDLite(source.BalanceID)
	require.NoError(t, err, "Failed to get updated source balance")

	updatedDest, err := ds.GetBalanceByIDLite(dest.BalanceID)
	require.NoError(t, err, "Failed to get updated destination balance")

	// Calculate expected balance changes
	expectedDebit := big.NewInt(int64(-500) * 100) // Amount * precision
	expectedCredit := big.NewInt(int64(500) * 100)

	// Verify balance changes
	require.Equal(t, 0, updatedSource.InflightBalance.Cmp(expectedDebit),
		"Source balance should be immediately reduced by transaction amount")
	require.Equal(t, 0, updatedDest.InflightBalance.Cmp(expectedCredit),
		"Destination balance should be immediately increased by transaction amount")

	// Verify no queued entry exists
	queuedEntry, err := ds.GetTransactionByRef(ctx, txnRef)
	require.NoError(t, err, "Failed to get queued transaction entry")
	require.Equal(t, StatusInflight, queuedEntry.Status, "Should have an INFLIGHT transaction entry")
}

func TestInflightTransactionFlowWithSkipQueueThenCommit(t *testing.T) {
	// Skip in short mode
	if testing.Short() {
		t.Skip("Skipping queue flow test in short mode")
	}

	ctx := context.Background()
	cnf := &config.Configuration{
		Redis: config.RedisConfig{
			Dns: "localhost:6379",
		},
		DataSource: config.DataSourceConfig{
			Dns: "postgres://postgres:password@localhost:5432/blnk?sslmode=disable",
		},
		Queue: config.QueueConfig{
			WebhookQueue:     "webhook_queue_test",
			IndexQueue:       "index_queue_test",
			TransactionQueue: "transaction_queue_test",
			NumberOfQueues:   1,
		},
		Server: config.ServerConfig{
			SecretKey: "test-secret",
		},
		Transaction: config.TransactionConfig{
			BatchSize:        100,
			MaxQueueSize:     1000,
			LockDuration:     time.Second * 30,
			IndexQueuePrefix: "test_index",
		},
	}
	config.ConfigStore.Store(cnf)

	ds, err := database.NewDataSource(cnf)
	require.NoError(t, err, "Failed to create datasource")

	blnk, err := NewBlnk(ds)
	require.NoError(t, err, "Failed to create Blnk instance")

	txnRef := "txn_" + model.GenerateUUIDWithSuffix("test")

	// Create test balances
	sourceBalance := &model.Balance{
		Currency: "USD",
		LedgerID: "general_ledger_id",
	}
	destBalance := &model.Balance{
		Currency: "USD",
		LedgerID: "general_ledger_id",
	}

	source, err := ds.CreateBalance(*sourceBalance)
	require.NoError(t, err, "Failed to create source balance")

	dest, err := ds.CreateBalance(*destBalance)
	require.NoError(t, err, "Failed to create destination balance")

	// Create transaction with skip_queue set to true
	txn := &model.Transaction{
		Reference:      txnRef,
		Source:         source.BalanceID,
		Destination:    dest.BalanceID,
		Amount:         500,
		Inflight:       true,
		Currency:       "USD",
		AllowOverdraft: true,
		Precision:      100,
		MetaData:       map[string]interface{}{"test": true},
		SkipQueue:      true, // Enable skip queue
	}

	// Queue the transaction
	queuedTxn, err := blnk.QueueTransaction(ctx, txn)
	require.NoError(t, err, "Failed to queue transaction")

	// Verify that the transaction was processed immediately
	require.Equal(t, StatusInflight, queuedTxn.Status, "Transaction should be INFLIGHT immediately when skip_queue is true")

	// Verify balances were updated immediately
	updatedSource, err := ds.GetBalanceByIDLite(source.BalanceID)
	require.NoError(t, err, "Failed to get updated source balance")

	updatedDest, err := ds.GetBalanceByIDLite(dest.BalanceID)
	require.NoError(t, err, "Failed to get updated destination balance")

	// Calculate expected balance changes
	expectedDebit := big.NewInt(int64(-500) * 100) // Amount * precision
	expectedCredit := big.NewInt(int64(500) * 100)

	// Verify balance changes
	require.Equal(t, 0, updatedSource.InflightBalance.Cmp(expectedDebit),
		"Source balance should be immediately reduced by transaction amount")
	require.Equal(t, 0, updatedDest.InflightBalance.Cmp(expectedCredit),
		"Destination balance should be immediately increased by transaction amount")

	// Verify no queued entry exists
	queuedEntry, err := ds.GetTransactionByRef(ctx, txnRef)
	require.NoError(t, err, "Failed to get queued transaction entry")
	require.Equal(t, StatusInflight, queuedEntry.Status, "Should have an INFLIGHT transaction entry")

	// Commit the transaction
	_, err = blnk.CommitInflightTransaction(ctx, txn.TransactionID, big.NewInt(0))
	require.NoError(t, err, "Failed to commit transaction")

	// Verify balances were updated immediately
	updatedSourceAfterCommit, err := ds.GetBalanceByIDLite(source.BalanceID)
	require.NoError(t, err, "Failed to get updated source balance")

	updatedDestAfterCommit, err := ds.GetBalanceByIDLite(dest.BalanceID)
	require.NoError(t, err, "Failed to get updated destination balance")

	// Verify balance changes
	require.Equal(t, 0, updatedSourceAfterCommit.Balance.Cmp(expectedDebit),
		"Source balance should be immediately reduced by transaction amount")
	require.Equal(t, 0, updatedDestAfterCommit.Balance.Cmp(expectedCredit),
		"Destination balance should be immediately increased by transaction amount")
}

func TestInflightTransactionFlowWithSkipQueueThenPartialCommit(t *testing.T) {
	// Skip in short mode
	if testing.Short() {
		t.Skip("Skipping queue flow test in short mode")
	}

	ctx := context.Background()
	cnf := &config.Configuration{
		Redis: config.RedisConfig{
			Dns: "localhost:6379",
		},
		DataSource: config.DataSourceConfig{
			Dns: "postgres://postgres:password@localhost:5432/blnk?sslmode=disable",
		},
		Queue: config.QueueConfig{
			WebhookQueue:     "webhook_queue_test",
			IndexQueue:       "index_queue_test",
			TransactionQueue: "transaction_queue_test",
			NumberOfQueues:   1,
		},
		Server: config.ServerConfig{
			SecretKey: "test-secret",
		},
		Transaction: config.TransactionConfig{
			BatchSize:        100,
			MaxQueueSize:     1000,
			LockDuration:     time.Second * 30,
			IndexQueuePrefix: "test_index",
		},
	}
	config.ConfigStore.Store(cnf)

	ds, err := database.NewDataSource(cnf)
	require.NoError(t, err, "Failed to create datasource")

	blnk, err := NewBlnk(ds)
	require.NoError(t, err, "Failed to create Blnk instance")

	txnRef := "txn_" + model.GenerateUUIDWithSuffix("test")

	// Create test balances
	sourceBalance := &model.Balance{
		Currency: "USD",
		LedgerID: "general_ledger_id",
	}
	destBalance := &model.Balance{
		Currency: "USD",
		LedgerID: "general_ledger_id",
	}

	source, err := ds.CreateBalance(*sourceBalance)
	require.NoError(t, err, "Failed to create source balance")

	dest, err := ds.CreateBalance(*destBalance)
	require.NoError(t, err, "Failed to create destination balance")

	// Create transaction with skip_queue set to true
	originalAmount := 500.0
	txn := &model.Transaction{
		Reference:      txnRef,
		Source:         source.BalanceID,
		Destination:    dest.BalanceID,
		Amount:         originalAmount,
		Inflight:       true,
		Currency:       "USD",
		AllowOverdraft: true,
		Precision:      100,
		MetaData:       map[string]interface{}{"test": true},
		SkipQueue:      true, // Enable skip queue
	}

	// Queue the transaction
	queuedTxn, err := blnk.QueueTransaction(ctx, txn)
	require.NoError(t, err, "Failed to queue transaction")

	// Verify that the transaction was processed immediately
	require.Equal(t, StatusInflight, queuedTxn.Status, "Transaction should be INFLIGHT immediately when skip_queue is true")

	// Verify balances were updated immediately
	updatedSource, err := ds.GetBalanceByIDLite(source.BalanceID)
	require.NoError(t, err, "Failed to get updated source balance")

	updatedDest, err := ds.GetBalanceByIDLite(dest.BalanceID)
	require.NoError(t, err, "Failed to get updated destination balance")

	// Calculate expected balance changes for the full amount
	expectedFullDebit := big.NewInt(int64(-originalAmount) * 100) // Amount * precision
	expectedFullCredit := big.NewInt(int64(originalAmount) * 100)

	// Verify balance changes
	require.Equal(t, 0, updatedSource.InflightBalance.Cmp(expectedFullDebit),
		"Source balance should be immediately reduced by transaction amount")
	require.Equal(t, 0, updatedDest.InflightBalance.Cmp(expectedFullCredit),
		"Destination balance should be immediately increased by transaction amount")

	// Verify inflight entry exists
	inflightEntry, err := ds.GetTransactionByRef(ctx, txnRef)
	require.NoError(t, err, "Failed to get inflight transaction entry")
	require.Equal(t, StatusInflight, inflightEntry.Status, "Should have an INFLIGHT transaction entry")

	// Partially commit the transaction (commit half of the original amount)
	partialAmount := originalAmount / 2 // 250.0
	partialPreciseAmount := new(big.Int).SetInt64(int64(partialAmount * txn.Precision))
	partialCommitTxn, err := blnk.CommitInflightTransaction(ctx, inflightEntry.TransactionID, partialPreciseAmount)
	require.NoError(t, err, "Failed to partially commit transaction")
	require.Equal(t, StatusApplied, partialCommitTxn.Status, "Partial commit transaction should have APPLIED status")
	require.Equal(t, partialAmount, partialCommitTxn.Amount, "Partial commit amount should match specified amount")

	// Verify balances were updated after partial commit
	updatedSourceAfterPartialCommit, err := ds.GetBalanceByIDLite(source.BalanceID)
	require.NoError(t, err, "Failed to get updated source balance")

	updatedDestAfterPartialCommit, err := ds.GetBalanceByIDLite(dest.BalanceID)
	require.NoError(t, err, "Failed to get updated destination balance")

	// Calculate expected balance changes for the partial commit
	expectedPartialDebit := big.NewInt(int64(-partialAmount) * 100) // PartialAmount * precision
	expectedPartialCredit := big.NewInt(int64(partialAmount) * 100)

	// Inflight balance should be reduced by the committed amount
	expectedRemainingInflightDebit := big.NewInt(0).Sub(expectedFullDebit, expectedPartialDebit)
	expectedRemainingInflightCredit := big.NewInt(0).Sub(expectedFullCredit, expectedPartialCredit)

	// Verify balance changes
	require.Equal(t, 0, updatedSourceAfterPartialCommit.Balance.Cmp(expectedPartialDebit),
		"Source balance should reflect the partial commit amount")
	require.Equal(t, 0, updatedDestAfterPartialCommit.Balance.Cmp(expectedPartialCredit),
		"Destination balance should reflect the partial commit amount")

	require.Equal(t, 0, updatedSourceAfterPartialCommit.InflightBalance.Cmp(expectedRemainingInflightDebit),
		"Source inflight balance should be reduced by the committed amount")
	require.Equal(t, 0, updatedDestAfterPartialCommit.InflightBalance.Cmp(expectedRemainingInflightCredit),
		"Destination inflight balance should be reduced by the committed amount")

	// Commit the remaining amount (should succeed)
	remainingCommitTxn, err := blnk.CommitInflightTransaction(ctx, inflightEntry.TransactionID, big.NewInt(0)) // 0 means commit remaining amount
	require.NoError(t, err, "Failed to commit remaining transaction amount")
	require.Equal(t, StatusApplied, remainingCommitTxn.Status, "Remaining commit transaction should have APPLIED status")

	// Verify the remaining commit amount
	remainingAmount := originalAmount - partialAmount
	require.InDelta(t, remainingAmount, remainingCommitTxn.Amount, 0.01, "Remaining commit amount should match expected remaining amount")

	// Verify final balances
	updatedSourceAfterFullCommit, err := ds.GetBalanceByIDLite(source.BalanceID)
	require.NoError(t, err, "Failed to get final source balance")

	updatedDestAfterFullCommit, err := ds.GetBalanceByIDLite(dest.BalanceID)
	require.NoError(t, err, "Failed to get final destination balance")

	// Verify final balance changes
	require.Equal(t, 0, updatedSourceAfterFullCommit.Balance.Cmp(expectedFullDebit),
		"Source balance should ultimately reflect the full amount")
	require.Equal(t, 0, updatedDestAfterFullCommit.Balance.Cmp(expectedFullCredit),
		"Destination balance should ultimately reflect the full amount")

	require.Equal(t, 0, updatedSourceAfterFullCommit.InflightBalance.Cmp(big.NewInt(0)),
		"Source inflight balance should be zero after full commit")
	require.Equal(t, 0, updatedDestAfterFullCommit.InflightBalance.Cmp(big.NewInt(0)),
		"Destination inflight balance should be zero after full commit")

	// Attempt another commit (should fail)
	_, err = blnk.CommitInflightTransaction(ctx, inflightEntry.TransactionID, big.NewInt(0))
	require.Error(t, err, "Committing a fully committed transaction should fail")
	require.Contains(t, err.Error(), "cannot commit. Transaction already committed",
		"Error message should indicate transaction is already committed")

	// Verify commit history
	commitHistory, err := ds.GetTransactionsByParent(ctx, inflightEntry.TransactionID, 2, 0)
	require.NoError(t, err, "Failed to get commit history")
	require.Equal(t, 2, len(commitHistory), "Should have exactly two commit transactions")

	// Just verify both transactions have the correct parent and status
	for i := range commitHistory {
		require.Equal(t, StatusApplied, commitHistory[i].Status,
			"Commit transaction should have APPLIED status")
		require.Equal(t, inflightEntry.TransactionID, commitHistory[i].ParentTransaction,
			"Commit should reference original transaction")
	}

	// Verify the total committed amount equals the original amount
	// Sort transactions by created time to identify first and second commit
	sort.Slice(commitHistory, func(i, j int) bool {
		return commitHistory[i].CreatedAt.Before(commitHistory[j].CreatedAt)
	})

	firstCommit := commitHistory[0]
	secondCommit := commitHistory[1]

	// Verify both commits exist
	require.NotNil(t, firstCommit, "Should have found the first commit transaction")
	require.NotNil(t, secondCommit, "Should have found the second commit transaction")

	// The sum of both commit amounts should equal the original amount
	totalCommittedAmount := firstCommit.Amount + secondCommit.Amount
	require.InDelta(t, originalAmount, totalCommittedAmount, 0.01,
		"Total committed amount should equal original amount")
}

func TestInflightTransactionFlowWithSkipQueueThenPartialCommitThenVoid(t *testing.T) {
	// Skip in short mode
	if testing.Short() {
		t.Skip("Skipping queue flow test in short mode")
	}

	ctx := context.Background()
	cnf := &config.Configuration{
		Redis: config.RedisConfig{
			Dns: "localhost:6379",
		},
		DataSource: config.DataSourceConfig{
			Dns: "postgres://postgres:password@localhost:5432/blnk?sslmode=disable",
		},
		Queue: config.QueueConfig{
			WebhookQueue:     "webhook_queue_test",
			IndexQueue:       "index_queue_test",
			TransactionQueue: "transaction_queue_test",
			NumberOfQueues:   1,
		},
		Server: config.ServerConfig{
			SecretKey: "test-secret",
		},
		Transaction: config.TransactionConfig{
			BatchSize:        100,
			MaxQueueSize:     1000,
			LockDuration:     time.Second * 30,
			IndexQueuePrefix: "test_index",
		},
	}
	config.ConfigStore.Store(cnf)

	ds, err := database.NewDataSource(cnf)
	require.NoError(t, err, "Failed to create datasource")

	blnk, err := NewBlnk(ds)
	require.NoError(t, err, "Failed to create Blnk instance")

	txnRef := "txn_" + model.GenerateUUIDWithSuffix("test")

	// Create test balances
	sourceBalance := &model.Balance{
		Currency: "USD",
		LedgerID: "general_ledger_id",
	}
	destBalance := &model.Balance{
		Currency: "USD",
		LedgerID: "general_ledger_id",
	}

	source, err := ds.CreateBalance(*sourceBalance)
	require.NoError(t, err, "Failed to create source balance")

	dest, err := ds.CreateBalance(*destBalance)
	require.NoError(t, err, "Failed to create destination balance")

	// Create transaction with skip_queue set to true
	originalAmount := 500.0
	txn := &model.Transaction{
		Reference:      txnRef,
		Source:         source.BalanceID,
		Destination:    dest.BalanceID,
		Amount:         originalAmount,
		Inflight:       true,
		Currency:       "USD",
		AllowOverdraft: true,
		Precision:      100,
		MetaData:       map[string]interface{}{"test": true},
		SkipQueue:      true, // Enable skip queue
	}

	// Queue the transaction
	queuedTxn, err := blnk.QueueTransaction(ctx, txn)
	require.NoError(t, err, "Failed to queue transaction")

	// Verify that the transaction was processed immediately
	require.Equal(t, StatusInflight, queuedTxn.Status, "Transaction should be INFLIGHT immediately when skip_queue is true")

	// Verify balances were updated immediately
	updatedSource, err := ds.GetBalanceByIDLite(source.BalanceID)
	require.NoError(t, err, "Failed to get updated source balance")

	updatedDest, err := ds.GetBalanceByIDLite(dest.BalanceID)
	require.NoError(t, err, "Failed to get updated destination balance")

	// Calculate expected balance changes for the full amount
	expectedFullDebit := big.NewInt(int64(-originalAmount) * 100) // Amount * precision
	expectedFullCredit := big.NewInt(int64(originalAmount) * 100)

	// Verify balance changes
	require.Equal(t, 0, updatedSource.InflightBalance.Cmp(expectedFullDebit),
		"Source balance should be immediately reduced by transaction amount")
	require.Equal(t, 0, updatedDest.InflightBalance.Cmp(expectedFullCredit),
		"Destination balance should be immediately increased by transaction amount")

	// Verify inflight entry exists
	inflightEntry, err := ds.GetTransactionByRef(ctx, txnRef)
	require.NoError(t, err, "Failed to get inflight transaction entry")
	require.Equal(t, StatusInflight, inflightEntry.Status, "Should have an INFLIGHT transaction entry")

	// Partially commit the transaction (commit half of the original amount)
	partialAmount := originalAmount / 2 // 250.0
	partialPreciseAmount := new(big.Int).SetInt64(int64(partialAmount * txn.Precision))
	partialCommitTxn, err := blnk.CommitInflightTransaction(ctx, inflightEntry.TransactionID, partialPreciseAmount)
	require.NoError(t, err, "Failed to partially commit transaction")
	require.Equal(t, StatusApplied, partialCommitTxn.Status, "Partial commit transaction should have APPLIED status")
	require.Equal(t, partialAmount, partialCommitTxn.Amount, "Partial commit amount should match specified amount")

	// Verify balances were updated after partial commit
	updatedSourceAfterPartialCommit, err := ds.GetBalanceByIDLite(source.BalanceID)
	require.NoError(t, err, "Failed to get updated source balance")

	updatedDestAfterPartialCommit, err := ds.GetBalanceByIDLite(dest.BalanceID)
	require.NoError(t, err, "Failed to get updated destination balance")

	// Calculate expected balance changes for the partial commit
	expectedPartialDebit := big.NewInt(int64(-partialAmount) * 100) // PartialAmount * precision
	expectedPartialCredit := big.NewInt(int64(partialAmount) * 100)

	// Inflight balance should be reduced by the committed amount
	expectedRemainingInflightDebit := big.NewInt(0).Sub(expectedFullDebit, expectedPartialDebit)
	expectedRemainingInflightCredit := big.NewInt(0).Sub(expectedFullCredit, expectedPartialCredit)

	// Verify balance changes
	require.Equal(t, 0, updatedSourceAfterPartialCommit.Balance.Cmp(expectedPartialDebit),
		"Source balance should reflect the partial commit amount")
	require.Equal(t, 0, updatedDestAfterPartialCommit.Balance.Cmp(expectedPartialCredit),
		"Destination balance should reflect the partial commit amount")

	require.Equal(t, 0, updatedSourceAfterPartialCommit.InflightBalance.Cmp(expectedRemainingInflightDebit),
		"Source inflight balance should be reduced by the committed amount")
	require.Equal(t, 0, updatedDestAfterPartialCommit.InflightBalance.Cmp(expectedRemainingInflightCredit),
		"Destination inflight balance should be reduced by the committed amount")

	// Void the remaining amount
	voidTxn, err := blnk.VoidInflightTransaction(ctx, inflightEntry.TransactionID)
	require.NoError(t, err, "Failed to void remaining transaction amount")
	require.Equal(t, StatusVoid, voidTxn.Status, "Void transaction should have VOID status")

	// Verify balances after void
	updatedSourceAfterVoid, err := ds.GetBalanceByIDLite(source.BalanceID)
	require.NoError(t, err, "Failed to get source balance after void")

	updatedDestAfterVoid, err := ds.GetBalanceByIDLite(dest.BalanceID)
	require.NoError(t, err, "Failed to get destination balance after void")

	// Verify that permanent balances remain at the partially committed amount
	require.Equal(t, 0, updatedSourceAfterVoid.Balance.Cmp(expectedPartialDebit),
		"Source balance should still reflect the partial commit amount after void")
	require.Equal(t, 0, updatedDestAfterVoid.Balance.Cmp(expectedPartialCredit),
		"Destination balance should still reflect the partial commit amount after void")

	// Verify that inflight balances are now zero after void
	require.Equal(t, 0, updatedSourceAfterVoid.InflightBalance.Cmp(big.NewInt(0)),
		"Source inflight balance should be zero after void")
	require.Equal(t, 0, updatedDestAfterVoid.InflightBalance.Cmp(big.NewInt(0)),
		"Destination inflight balance should be zero after void")

	// Attempt another commit (should fail)
	_, err = blnk.CommitInflightTransaction(ctx, inflightEntry.TransactionID, big.NewInt(0))
	require.Error(t, err, "Committing a voided transaction should fail")
	require.Contains(t, err.Error(), "transaction has already been voided",
		"Error message should indicate transaction is already voided")

	// Attempt another void (should fail)
	_, err = blnk.VoidInflightTransaction(ctx, inflightEntry.TransactionID)
	require.Error(t, err, "Voiding an already voided transaction should fail")
	require.Contains(t, err.Error(), "transaction has already been voided",
		"Error message should indicate transaction is already voided")

	// Verify transaction history
	txnHistory, err := ds.GetTransactionsByParent(ctx, inflightEntry.TransactionID, 10, 0)
	require.NoError(t, err, "Failed to get transaction history")
	require.Equal(t, 2, len(txnHistory), "Should have exactly two transactions: one commit and one void")

	// Check that we have one APPLIED transaction and one VOID transaction
	hasApplied := false
	hasVoid := false

	for _, historyTxn := range txnHistory {
		if historyTxn.Status == StatusApplied {
			hasApplied = true
			require.Equal(t, partialAmount, historyTxn.Amount, "Applied transaction should have partial amount")
		} else if historyTxn.Status == StatusVoid {
			hasVoid = true
			remainingAmount := originalAmount - partialAmount
			require.InDelta(t, remainingAmount, historyTxn.Amount, 0.01,
				"Void transaction should have remaining amount")
		}
	}

	require.True(t, hasApplied, "Transaction history should include an APPLIED transaction")
	require.True(t, hasVoid, "Transaction history should include a VOID transaction")
}

func TestTwoInflightTransactionsCommitOneVoidCommitted(t *testing.T) {
	// Skip in short mode
	if testing.Short() {
		t.Skip("Skipping queue flow test in short mode")
	}

	ctx := context.Background()
	cnf := &config.Configuration{
		Redis: config.RedisConfig{
			Dns: "localhost:6379",
		},
		DataSource: config.DataSourceConfig{
			Dns: "postgres://postgres:password@localhost:5432/blnk?sslmode=disable",
		},
		Queue: config.QueueConfig{
			WebhookQueue:     "webhook_queue_test",
			IndexQueue:       "index_queue_test",
			TransactionQueue: "transaction_queue_test",
			NumberOfQueues:   1,
		},
		Server: config.ServerConfig{
			SecretKey: "test-secret",
		},
		Transaction: config.TransactionConfig{
			BatchSize:        100,
			MaxQueueSize:     1000,
			LockDuration:     time.Second * 30,
			IndexQueuePrefix: "test_index",
		},
	}
	config.ConfigStore.Store(cnf)

	ds, err := database.NewDataSource(cnf)
	require.NoError(t, err, "Failed to create datasource")

	blnk, err := NewBlnk(ds)
	require.NoError(t, err, "Failed to create Blnk instance")

	// Create test balances
	sourceBalance := &model.Balance{
		Currency: "USD",
		LedgerID: "general_ledger_id",
	}
	destBalance := &model.Balance{
		Currency: "USD",
		LedgerID: "general_ledger_id",
	}

	source, err := ds.CreateBalance(*sourceBalance)
	require.NoError(t, err, "Failed to create source balance")

	dest, err := ds.CreateBalance(*destBalance)
	require.NoError(t, err, "Failed to create destination balance")

	// Create first inflight transaction with HIGHER amount
	txnRef1 := "txn_" + model.GenerateUUIDWithSuffix("test1")
	amount1 := 800.0 // Higher amount
	txn1 := &model.Transaction{
		Reference:      txnRef1,
		Source:         source.BalanceID,
		Destination:    dest.BalanceID,
		Amount:         amount1,
		Inflight:       true,
		Currency:       "USD",
		AllowOverdraft: true,
		Precision:      100,
		MetaData:       map[string]interface{}{"test": "transaction1_higher"},
		SkipQueue:      true,
	}

	// Create second inflight transaction with LOWER amount
	txnRef2 := "txn_" + model.GenerateUUIDWithSuffix("test2")
	amount2 := 300.0 // Lower amount
	txn2 := &model.Transaction{
		Reference:      txnRef2,
		Source:         source.BalanceID,
		Destination:    dest.BalanceID,
		Amount:         amount2,
		Inflight:       true,
		Currency:       "USD",
		AllowOverdraft: true,
		Precision:      100,
		MetaData:       map[string]interface{}{"test": "transaction2_lower"},
		SkipQueue:      true,
	}

	// Queue both transactions
	queuedTxn1, err := blnk.QueueTransaction(ctx, txn1)
	require.NoError(t, err, "Failed to queue first (higher) transaction")
	require.Equal(t, StatusInflight, queuedTxn1.Status, "First transaction should be INFLIGHT")

	queuedTxn2, err := blnk.QueueTransaction(ctx, txn2)
	require.NoError(t, err, "Failed to queue second (lower) transaction")
	require.Equal(t, StatusInflight, queuedTxn2.Status, "Second transaction should be INFLIGHT")

	// Get the inflight entries
	inflightEntry1, err := ds.GetTransactionByRef(ctx, txnRef1)
	require.NoError(t, err, "Failed to get first inflight transaction")
	require.Equal(t, StatusInflight, inflightEntry1.Status, "First transaction should be INFLIGHT")

	inflightEntry2, err := ds.GetTransactionByRef(ctx, txnRef2)
	require.NoError(t, err, "Failed to get second inflight transaction")
	require.Equal(t, StatusInflight, inflightEntry2.Status, "Second transaction should be INFLIGHT")

	// Check balances after both inflight transactions
	afterInflightSource, err := ds.GetBalanceByIDLite(source.BalanceID)
	require.NoError(t, err, "Failed to get source balance")

	afterInflightDest, err := ds.GetBalanceByIDLite(dest.BalanceID)
	require.NoError(t, err, "Failed to get destination balance")

	// Calculate expected inflight balances (both transactions are inflight)
	totalInflightAmount := amount1 + amount2
	expectedInflightDebit := big.NewInt(int64(-totalInflightAmount) * 100)
	expectedInflightCredit := big.NewInt(int64(totalInflightAmount) * 100)

	// Verify inflight balances
	require.Equal(t, 0, afterInflightSource.InflightBalance.Cmp(expectedInflightDebit),
		"Source inflight balance should reflect both transactions")
	require.Equal(t, 0, afterInflightDest.InflightBalance.Cmp(expectedInflightCredit),
		"Destination inflight balance should reflect both transactions")

	// Commit the second transaction (lower amount)
	amount2Precise := new(big.Int).SetInt64(int64(amount2 * txn2.Precision))
	commitTxn, err := blnk.CommitInflightTransaction(ctx, inflightEntry2.TransactionID, amount2Precise)
	require.NoError(t, err, "Failed to commit second transaction")
	require.Equal(t, StatusApplied, commitTxn.Status, "Second transaction should have APPLIED status")

	// Check balances after committing second transaction
	afterCommitSource, err := ds.GetBalanceByIDLite(source.BalanceID)
	require.NoError(t, err, "Failed to get source balance after commit")

	afterCommitDest, err := ds.GetBalanceByIDLite(dest.BalanceID)
	require.NoError(t, err, "Failed to get destination balance after commit")

	// Calculate expected balances
	// Permanent balance should reflect only the committed transaction
	expectedCommittedDebit := big.NewInt(int64(-amount2) * 100)
	expectedCommittedCredit := big.NewInt(int64(amount2) * 100)

	// Inflight balance should reflect only the first transaction
	expectedRemainingInflightDebit := big.NewInt(int64(-amount1) * 100)
	expectedRemainingInflightCredit := big.NewInt(int64(amount1) * 100)

	// Verify permanent balances
	require.Equal(t, 0, afterCommitSource.Balance.Cmp(expectedCommittedDebit),
		"Source permanent balance should reflect only the committed transaction")
	require.Equal(t, 0, afterCommitDest.Balance.Cmp(expectedCommittedCredit),
		"Destination permanent balance should reflect only the committed transaction")

	// Verify inflight balances
	require.Equal(t, 0, afterCommitSource.InflightBalance.Cmp(expectedRemainingInflightDebit),
		"Source inflight balance should reflect only the first transaction")
	require.Equal(t, 0, afterCommitDest.InflightBalance.Cmp(expectedRemainingInflightCredit),
		"Destination inflight balance should reflect only the first transaction")

	// Save the balances for later comparison
	balanceBeforeVoidAttemptSource := afterCommitSource
	balanceBeforeVoidAttemptDest := afterCommitDest

	// Attempt to void the already committed transaction (should fail)
	_, err = blnk.VoidInflightTransaction(ctx, inflightEntry2.TransactionID)
	require.Error(t, err, "Voiding an already committed transaction should fail")
	require.Contains(t, err.Error(), "Transaction already committed",
		"Error message should indicate transaction is already committed")

	// Check balances again to ensure they weren't affected by the failed void attempt
	afterVoidAttemptSource, err := ds.GetBalanceByIDLite(source.BalanceID)
	require.NoError(t, err, "Failed to get source balance after void attempt")

	afterVoidAttemptDest, err := ds.GetBalanceByIDLite(dest.BalanceID)
	require.NoError(t, err, "Failed to get destination balance after void attempt")

	// Verify permanent balances remain unchanged
	require.Equal(t, 0, balanceBeforeVoidAttemptSource.Balance.Cmp(afterVoidAttemptSource.Balance),
		"Source permanent balance should be unchanged after failed void attempt")
	require.Equal(t, 0, balanceBeforeVoidAttemptDest.Balance.Cmp(afterVoidAttemptDest.Balance),
		"Destination permanent balance should be unchanged after failed void attempt")

	// Verify inflight balances remain unchanged
	require.Equal(t, 0, balanceBeforeVoidAttemptSource.InflightBalance.Cmp(afterVoidAttemptSource.InflightBalance),
		"Source inflight balance should be unchanged after failed void attempt")
	require.Equal(t, 0, balanceBeforeVoidAttemptDest.InflightBalance.Cmp(afterVoidAttemptDest.InflightBalance),
		"Destination inflight balance should be unchanged after failed void attempt")

	// Verify we can still void the first transaction (still inflight)
	voidTxn, err := blnk.VoidInflightTransaction(ctx, inflightEntry1.TransactionID)
	require.NoError(t, err, "Failed to void first transaction")
	require.Equal(t, StatusVoid, voidTxn.Status, "First transaction should have VOID status")

	// Check balances after voiding first transaction
	afterVoidFirstSource, err := ds.GetBalanceByIDLite(source.BalanceID)
	require.NoError(t, err, "Failed to get source balance after voiding first transaction")

	afterVoidFirstDest, err := ds.GetBalanceByIDLite(dest.BalanceID)
	require.NoError(t, err, "Failed to get destination balance after voiding first transaction")

	// Permanent balances should still reflect only the second committed transaction
	require.Equal(t, 0, afterVoidFirstSource.Balance.Cmp(expectedCommittedDebit),
		"Source permanent balance should still reflect only the second committed transaction")
	require.Equal(t, 0, afterVoidFirstDest.Balance.Cmp(expectedCommittedCredit),
		"Destination permanent balance should still reflect only the second committed transaction")

	// Inflight balances should now be zero
	require.Equal(t, 0, afterVoidFirstSource.InflightBalance.Cmp(big.NewInt(0)),
		"Source inflight balance should be zero after voiding first transaction")
	require.Equal(t, 0, afterVoidFirstDest.InflightBalance.Cmp(big.NewInt(0)),
		"Destination inflight balance should be zero after voiding first transaction")

	// Verify transaction history for both transactions
	txnHistory1, err := ds.GetTransactionsByParent(ctx, inflightEntry1.TransactionID, 10, 0)
	require.NoError(t, err, "Failed to get first transaction history")
	require.Equal(t, 1, len(txnHistory1), "Should have exactly one transaction for the first parent: the void")
	require.Equal(t, StatusVoid, txnHistory1[0].Status, "First transaction history should include a VOID transaction")

	txnHistory2, err := ds.GetTransactionsByParent(ctx, inflightEntry2.TransactionID, 10, 0)
	require.NoError(t, err, "Failed to get second transaction history")
	require.Equal(t, 1, len(txnHistory2), "Should have exactly one transaction for the second parent: the commit")
	require.Equal(t, StatusApplied, txnHistory2[0].Status, "Second transaction history should include an APPLIED transaction")
}

func TestInflightTransactionWithOvercommitValidation(t *testing.T) {
	// Skip in short mode
	if testing.Short() {
		t.Skip("Skipping queue flow test in short mode")
	}

	ctx := context.Background()
	cnf := &config.Configuration{
		Redis: config.RedisConfig{
			Dns: "localhost:6379",
		},
		DataSource: config.DataSourceConfig{
			Dns: "postgres://postgres:password@localhost:5432/blnk?sslmode=disable",
		},
		Queue: config.QueueConfig{
			WebhookQueue:     "webhook_queue_test",
			IndexQueue:       "index_queue_test",
			TransactionQueue: "transaction_queue_test",
			NumberOfQueues:   1,
		},
		Server: config.ServerConfig{
			SecretKey: "test-secret",
		},
		Transaction: config.TransactionConfig{
			BatchSize:        100,
			MaxQueueSize:     1000,
			LockDuration:     time.Second * 30,
			IndexQueuePrefix: "test_index",
		},
	}
	config.ConfigStore.Store(cnf)

	ds, err := database.NewDataSource(cnf)
	require.NoError(t, err, "Failed to create datasource")

	blnk, err := NewBlnk(ds)
	require.NoError(t, err, "Failed to create Blnk instance")

	// Create source and destination balances
	sourceBalance := &model.Balance{
		Currency: "USD",
		LedgerID: "general_ledger_id",
	}
	destBalance := &model.Balance{
		Currency: "USD",
		LedgerID: "general_ledger_id",
	}

	source, err := ds.CreateBalance(*sourceBalance)
	require.NoError(t, err, "Failed to create source balance")

	dest, err := ds.CreateBalance(*destBalance)
	require.NoError(t, err, "Failed to create destination balance")

	// Create an initial transaction to transfer from source to destination
	initialBalance := 1000.0
	depositTxnRef := "transfer_" + model.GenerateUUIDWithSuffix("test")
	depositTxn := &model.Transaction{
		Reference:      depositTxnRef,
		Source:         source.BalanceID,
		Destination:    dest.BalanceID,
		Amount:         initialBalance,
		Currency:       "USD",
		Precision:      100,
		AllowOverdraft: true,
		MetaData:       map[string]interface{}{"test": "initial_transfer"},
		SkipQueue:      true,
	}

	// Process the initial transaction
	depositResult, err := blnk.QueueTransaction(ctx, depositTxn)
	require.NoError(t, err, "Failed to process initial transaction")
	require.Equal(t, StatusApplied, depositResult.Status, "Initial transaction should be APPLIED")

	// Verify balances after initial transfer
	sourceAfterTransfer, err := ds.GetBalanceByIDLite(source.BalanceID)
	require.NoError(t, err, "Failed to get source balance after transfer")

	destAfterTransfer, err := ds.GetBalanceByIDLite(dest.BalanceID)
	require.NoError(t, err, "Failed to get destination balance after transfer")

	// Source should be -1000, destination should be 1000
	expectedSourceBalance := big.NewInt(int64(-initialBalance * 100))
	expectedDestBalance := big.NewInt(int64(initialBalance * 100))

	require.Equal(t, 0, sourceAfterTransfer.Balance.Cmp(expectedSourceBalance),
		"Source balance should be -1000 after initial transfer")
	require.Equal(t, 0, destAfterTransfer.Balance.Cmp(expectedDestBalance),
		"Destination balance should be 1000 after initial transfer")

	// Step 1: Create an inflight transaction
	inflightAmount := 100.0
	txnRef := "txn_" + model.GenerateUUIDWithSuffix("test")
	txn := &model.Transaction{
		Reference:      txnRef,
		Source:         source.BalanceID,
		Destination:    dest.BalanceID,
		Amount:         inflightAmount,
		Inflight:       true,
		Currency:       "USD",
		AllowOverdraft: true,
		Precision:      100,
		MetaData:       map[string]interface{}{"test": "overcommit_test"},
		SkipQueue:      true,
	}

	// Queue the transaction
	queuedTxn, err := blnk.QueueTransaction(ctx, txn)
	require.NoError(t, err, "Failed to queue transaction")
	require.Equal(t, StatusInflight, queuedTxn.Status, "Transaction should be INFLIGHT")

	// Verify inflight balance is updated correctly
	afterInflightSource, err := ds.GetBalanceByIDLite(source.BalanceID)
	require.NoError(t, err, "Failed to get source balance after inflight")

	expectedInflightBalance := big.NewInt(int64(-inflightAmount * 100))
	require.Equal(t, 0, afterInflightSource.InflightBalance.Cmp(expectedInflightBalance),
		"Source inflight balance should be -100 after inflight transaction")

	// Get the inflight entry
	inflightEntry, err := ds.GetTransactionByRef(ctx, txnRef)
	require.NoError(t, err, "Failed to get inflight transaction")
	require.Equal(t, StatusInflight, inflightEntry.Status, "Transaction should be INFLIGHT")

	// Step 2: Attempt to overcommit the transaction (should fail)
	overCommitAmount := 150.0
	overCommitPreciseAmount := new(big.Int).SetInt64(int64(overCommitAmount * txn.Precision))
	_, err = blnk.CommitInflightTransaction(ctx, inflightEntry.TransactionID, overCommitPreciseAmount)

	// Now verify overcommit is prevented
	require.Error(t, err, "Overcommitting should fail with an error")
	require.Contains(t, err.Error(), "cannot commit more than the original transaction amount",
		"Error should mention that overcommit is not allowed")

	// Verify balances remain unchanged after failed overcommit
	afterFailedOvercommitSource, err := ds.GetBalanceByIDLite(source.BalanceID)
	require.NoError(t, err, "Failed to get source balance after failed overcommit")

	afterFailedOvercommitDest, err := ds.GetBalanceByIDLite(dest.BalanceID)
	require.NoError(t, err, "Failed to get destination balance after failed overcommit")

	// Source balance should still be -1000
	require.Equal(t, 0, afterFailedOvercommitSource.Balance.Cmp(expectedSourceBalance),
		"Source balance should remain -1000 after failed overcommit")

	// Destination balance should still be 1000
	require.Equal(t, 0, afterFailedOvercommitDest.Balance.Cmp(expectedDestBalance),
		"Destination balance should remain 1000 after failed overcommit")

	// Inflight balance should still be -100 for source and +100 for destination
	require.Equal(t, 0, afterFailedOvercommitSource.InflightBalance.Cmp(expectedInflightBalance),
		"Source inflight balance should remain -100 after failed overcommit")

	expectedDestInflightBalance := big.NewInt(int64(inflightAmount * 100))
	require.Equal(t, 0, afterFailedOvercommitDest.InflightBalance.Cmp(expectedDestInflightBalance),
		"Destination inflight balance should remain 100 after failed overcommit")

	// Step 3: Partial commit (50 of the 100)
	partialAmount := 50.0
	partialPreciseAmount := new(big.Int).SetInt64(int64(partialAmount * txn.Precision))
	partialCommitTxn, err := blnk.CommitInflightTransaction(ctx, inflightEntry.TransactionID, partialPreciseAmount)
	require.NoError(t, err, "Partial commit should succeed")
	require.Equal(t, StatusApplied, partialCommitTxn.Status, "Partial commit transaction should have APPLIED status")

	// Verify balances after partial commit
	afterPartialCommitSource, err := ds.GetBalanceByIDLite(source.BalanceID)
	require.NoError(t, err, "Failed to get source balance after partial commit")

	afterPartialCommitDest, err := ds.GetBalanceByIDLite(dest.BalanceID)
	require.NoError(t, err, "Failed to get destination balance after partial commit")

	// Source balance should be -1000 - 50 = -1050
	expectedSourceBalanceAfterPartial := big.NewInt(int64((-initialBalance - partialAmount) * 100))
	require.Equal(t, 0, afterPartialCommitSource.Balance.Cmp(expectedSourceBalanceAfterPartial),
		"Source balance should be -1050 after partial commit")

	// Destination balance should be 1000 + 50 = 1050
	expectedDestBalanceAfterPartial := big.NewInt(int64((initialBalance + partialAmount) * 100))
	require.Equal(t, 0, afterPartialCommitDest.Balance.Cmp(expectedDestBalanceAfterPartial),
		"Destination balance should be 1050 after partial commit")

	// Source inflight balance should be -100 + 50 = -50
	expectedSourceInflightAfterPartial := big.NewInt(int64(-(inflightAmount - partialAmount) * 100))
	require.Equal(t, 0, afterPartialCommitSource.InflightBalance.Cmp(expectedSourceInflightAfterPartial),
		"Source inflight balance should be -50 after partial commit")

	// Destination inflight balance should be 100 - 50 = 50
	expectedDestInflightAfterPartial := big.NewInt(int64((inflightAmount - partialAmount) * 100))
	require.Equal(t, 0, afterPartialCommitDest.InflightBalance.Cmp(expectedDestInflightAfterPartial),
		"Destination inflight balance should be 50 after partial commit")

	// Step 4: Full commit (remaining 50)
	fullCommitTxn, err := blnk.CommitInflightTransaction(ctx, inflightEntry.TransactionID, big.NewInt(0))
	require.NoError(t, err, "Full commit should succeed")
	require.Equal(t, StatusApplied, fullCommitTxn.Status, "Full commit transaction should have APPLIED status")

	// Verify balances after full commit
	afterFullCommitSource, err := ds.GetBalanceByIDLite(source.BalanceID)
	require.NoError(t, err, "Failed to get source balance after full commit")

	afterFullCommitDest, err := ds.GetBalanceByIDLite(dest.BalanceID)
	require.NoError(t, err, "Failed to get destination balance after full commit")

	// Source balance should be -1050 - 50 = -1100
	expectedSourceBalanceAfterFull := big.NewInt(int64((-initialBalance - inflightAmount) * 100))
	require.Equal(t, 0, afterFullCommitSource.Balance.Cmp(expectedSourceBalanceAfterFull),
		"Source balance should be -1100 after full commit")

	// Destination balance should be 1050 + 50 = 1100
	expectedDestBalanceAfterFull := big.NewInt(int64((initialBalance + inflightAmount) * 100))
	require.Equal(t, 0, afterFullCommitDest.Balance.Cmp(expectedDestBalanceAfterFull),
		"Destination balance should be 1100 after full commit")

	// Both inflight balances should be 0
	require.Equal(t, 0, afterFullCommitSource.InflightBalance.Cmp(big.NewInt(0)),
		"Source inflight balance should be zero after full commit")
	require.Equal(t, 0, afterFullCommitDest.InflightBalance.Cmp(big.NewInt(0)),
		"Destination inflight balance should be zero after full commit")

	// Step 5: Attempt another commit (should fail)
	_, err = blnk.CommitInflightTransaction(ctx, inflightEntry.TransactionID, big.NewInt(0))
	require.Error(t, err, "Second full commit should fail")
	require.Contains(t, err.Error(), "cannot commit. Transaction already committed",
		"Error should mention that transaction is already committed")

	// Verify balances remain unchanged after failed second commit
	afterFailedSecondCommitSource, err := ds.GetBalanceByIDLite(source.BalanceID)
	require.NoError(t, err, "Failed to get source balance after failed second commit")

	afterFailedSecondCommitDest, err := ds.GetBalanceByIDLite(dest.BalanceID)
	require.NoError(t, err, "Failed to get destination balance after failed second commit")

	require.Equal(t, 0, afterFailedSecondCommitSource.Balance.Cmp(expectedSourceBalanceAfterFull),
		"Source balance should remain -1100 after failed second commit")
	require.Equal(t, 0, afterFailedSecondCommitDest.Balance.Cmp(expectedDestBalanceAfterFull),
		"Destination balance should remain 1100 after failed second commit")
	require.Equal(t, 0, afterFailedSecondCommitSource.InflightBalance.Cmp(big.NewInt(0)),
		"Source inflight balance should remain zero after failed second commit")
	require.Equal(t, 0, afterFailedSecondCommitDest.InflightBalance.Cmp(big.NewInt(0)),
		"Destination inflight balance should remain zero after failed second commit")

	// Step 6: Attempt to void a fully committed transaction (should fail)
	_, err = blnk.VoidInflightTransaction(ctx, inflightEntry.TransactionID)
	require.Error(t, err, "Voiding a fully committed transaction should fail")
	require.Contains(t, err.Error(), "cannot void. Transaction already committed",
		"Error should mention that transaction is already committed")

	// Verify balances remain unchanged after failed void
	afterFailedVoidSource, err := ds.GetBalanceByIDLite(source.BalanceID)
	require.NoError(t, err, "Failed to get source balance after failed void")

	afterFailedVoidDest, err := ds.GetBalanceByIDLite(dest.BalanceID)
	require.NoError(t, err, "Failed to get destination balance after failed void")

	require.Equal(t, 0, afterFailedVoidSource.Balance.Cmp(expectedSourceBalanceAfterFull),
		"Source balance should remain -1100 after failed void")
	require.Equal(t, 0, afterFailedVoidDest.Balance.Cmp(expectedDestBalanceAfterFull),
		"Destination balance should remain 1100 after failed void")
	require.Equal(t, 0, afterFailedVoidSource.InflightBalance.Cmp(big.NewInt(0)),
		"Source inflight balance should remain zero after failed void")
	require.Equal(t, 0, afterFailedVoidDest.InflightBalance.Cmp(big.NewInt(0)),
		"Destination inflight balance should remain zero after failed void")

	// Verify transaction history
	txnHistory, err := ds.GetTransactionsByParent(ctx, inflightEntry.TransactionID, 10, 0)
	require.NoError(t, err, "Failed to get transaction history")
	require.Equal(t, 2, len(txnHistory), "Should have exactly two transactions: partial commit and full commit")

	// Check transaction history details
	for _, historyTxn := range txnHistory {
		require.Equal(t, StatusApplied, historyTxn.Status, "All history transactions should be APPLIED")
		require.Equal(t, "USD", historyTxn.Currency, "All history transactions should be in USD")
	}
}

func TestInflightTransactionWithPartialOvercommitValidation(t *testing.T) {
	// Skip in short mode
	if testing.Short() {
		t.Skip("Skipping queue flow test in short mode")
	}

	ctx := context.Background()
	cnf := &config.Configuration{
		Redis: config.RedisConfig{
			Dns: "localhost:6379",
		},
		DataSource: config.DataSourceConfig{
			Dns: "postgres://postgres:password@localhost:5432/blnk?sslmode=disable",
		},
		Queue: config.QueueConfig{
			WebhookQueue:     "webhook_queue_test",
			IndexQueue:       "index_queue_test",
			TransactionQueue: "transaction_queue_test",
			NumberOfQueues:   1,
		},
		Server: config.ServerConfig{
			SecretKey: "test-secret",
		},
		Transaction: config.TransactionConfig{
			BatchSize:        100,
			MaxQueueSize:     1000,
			LockDuration:     time.Second * 30,
			IndexQueuePrefix: "test_index",
		},
	}
	config.ConfigStore.Store(cnf)

	ds, err := database.NewDataSource(cnf)
	require.NoError(t, err, "Failed to create datasource")

	blnk, err := NewBlnk(ds)
	require.NoError(t, err, "Failed to create Blnk instance")

	// Create source and destination balances
	sourceBalance := &model.Balance{
		Currency: "USD",
		LedgerID: "general_ledger_id",
	}
	destBalance := &model.Balance{
		Currency: "USD",
		LedgerID: "general_ledger_id",
	}

	source, err := ds.CreateBalance(*sourceBalance)
	require.NoError(t, err, "Failed to create source balance")

	dest, err := ds.CreateBalance(*destBalance)
	require.NoError(t, err, "Failed to create destination balance")

	// Create an initial transaction to transfer from source to destination
	initialBalance := 1000.0
	depositTxnRef := "transfer_" + model.GenerateUUIDWithSuffix("test")
	depositTxn := &model.Transaction{
		Reference:      depositTxnRef,
		Source:         source.BalanceID,
		Destination:    dest.BalanceID,
		Amount:         initialBalance,
		Currency:       "USD",
		Precision:      100,
		AllowOverdraft: true,
		MetaData:       map[string]interface{}{"test": "initial_transfer"},
		SkipQueue:      true,
	}

	// Process the initial transaction
	depositResult, err := blnk.QueueTransaction(ctx, depositTxn)
	require.NoError(t, err, "Failed to process initial transaction")
	require.Equal(t, StatusApplied, depositResult.Status, "Initial transaction should be APPLIED")

	// Verify balances after initial transfer
	sourceAfterTransfer, err := ds.GetBalanceByIDLite(source.BalanceID)
	require.NoError(t, err, "Failed to get source balance after transfer")

	destAfterTransfer, err := ds.GetBalanceByIDLite(dest.BalanceID)
	require.NoError(t, err, "Failed to get destination balance after transfer")

	// Source should be -1000, destination should be 1000
	expectedSourceBalance := big.NewInt(int64(-initialBalance * 100))
	expectedDestBalance := big.NewInt(int64(initialBalance * 100))

	require.Equal(t, 0, sourceAfterTransfer.Balance.Cmp(expectedSourceBalance),
		"Source balance should be -1000 after initial transfer")
	require.Equal(t, 0, destAfterTransfer.Balance.Cmp(expectedDestBalance),
		"Destination balance should be 1000 after initial transfer")

	// Step 1: Create an inflight transaction
	inflightAmount := 100.0
	txnRef := "txn_" + model.GenerateUUIDWithSuffix("test")
	txn := &model.Transaction{
		Reference:      txnRef,
		Source:         source.BalanceID,
		Destination:    dest.BalanceID,
		Amount:         inflightAmount,
		Inflight:       true,
		Currency:       "USD",
		AllowOverdraft: true,
		Precision:      100,
		MetaData:       map[string]interface{}{"test": "partial_overcommit_test"},
		SkipQueue:      true,
	}

	// Queue the transaction
	queuedTxn, err := blnk.QueueTransaction(ctx, txn)
	require.NoError(t, err, "Failed to queue transaction")
	require.Equal(t, StatusInflight, queuedTxn.Status, "Transaction should be INFLIGHT")

	// Verify inflight balance is updated correctly
	afterInflightSource, err := ds.GetBalanceByIDLite(source.BalanceID)
	require.NoError(t, err, "Failed to get source balance after inflight")

	expectedInflightBalance := big.NewInt(int64(-inflightAmount * 100))
	require.Equal(t, 0, afterInflightSource.InflightBalance.Cmp(expectedInflightBalance),
		"Source inflight balance should be -100 after inflight transaction")

	// Get the inflight entry
	inflightEntry, err := ds.GetTransactionByRef(ctx, txnRef)
	require.NoError(t, err, "Failed to get inflight transaction")
	require.Equal(t, StatusInflight, inflightEntry.Status, "Transaction should be INFLIGHT")

	// Step 2: Do a partial commit (50 of the 100)
	partialAmount := 50.0
	partialPreciseAmount := new(big.Int).SetInt64(int64(partialAmount * txn.Precision))
	partialCommitTxn, err := blnk.CommitInflightTransaction(ctx, inflightEntry.TransactionID, partialPreciseAmount)
	require.NoError(t, err, "Partial commit should succeed")
	require.Equal(t, StatusApplied, partialCommitTxn.Status, "Partial commit transaction should have APPLIED status")

	// Verify balances after partial commit
	afterPartialCommitSource, err := ds.GetBalanceByIDLite(source.BalanceID)
	require.NoError(t, err, "Failed to get source balance after partial commit")

	afterPartialCommitDest, err := ds.GetBalanceByIDLite(dest.BalanceID)
	require.NoError(t, err, "Failed to get destination balance after partial commit")

	// Source balance should be -1000 - 50 = -1050
	expectedSourceBalanceAfterPartial := big.NewInt(int64((-initialBalance - partialAmount) * 100))
	require.Equal(t, 0, afterPartialCommitSource.Balance.Cmp(expectedSourceBalanceAfterPartial),
		"Source balance should be -1050 after partial commit")

	// Destination balance should be 1000 + 50 = 1050
	expectedDestBalanceAfterPartial := big.NewInt(int64((initialBalance + partialAmount) * 100))
	require.Equal(t, 0, afterPartialCommitDest.Balance.Cmp(expectedDestBalanceAfterPartial),
		"Destination balance should be 1050 after partial commit")

	// Source inflight balance should be -100 + 50 = -50
	expectedSourceInflightAfterPartial := big.NewInt(int64(-(inflightAmount - partialAmount) * 100))
	require.Equal(t, 0, afterPartialCommitSource.InflightBalance.Cmp(expectedSourceInflightAfterPartial),
		"Source inflight balance should be -50 after partial commit")

	// Destination inflight balance should be 100 - 50 = 50
	expectedDestInflightAfterPartial := big.NewInt(int64((inflightAmount - partialAmount) * 100))
	require.Equal(t, 0, afterPartialCommitDest.InflightBalance.Cmp(expectedDestInflightAfterPartial),
		"Destination inflight balance should be 50 after partial commit")

	// Step 3: Attempt to overcommit the REMAINING balance (should fail)
	// Try to commit 60 when only 50 remains
	overCommitAmount := 60.0
	overCommitPreciseAmount := new(big.Int).SetInt64(int64(overCommitAmount * txn.Precision))
	_, err = blnk.CommitInflightTransaction(ctx, inflightEntry.TransactionID, overCommitPreciseAmount)

	// Now verify partial overcommit is prevented
	require.Error(t, err, "Partial overcommit should fail with an error")
	require.Contains(t, err.Error(), "cannot commit more than the remaining amount",
		"Error should mention that overcommit of the remaining amount is not allowed")

	// Verify balances remain unchanged after failed partial overcommit
	afterFailedOvercommitSource, err := ds.GetBalanceByIDLite(source.BalanceID)
	require.NoError(t, err, "Failed to get source balance after failed partial overcommit")

	afterFailedOvercommitDest, err := ds.GetBalanceByIDLite(dest.BalanceID)
	require.NoError(t, err, "Failed to get destination balance after failed partial overcommit")

	// Source balance should still be -1050
	require.Equal(t, 0, afterFailedOvercommitSource.Balance.Cmp(expectedSourceBalanceAfterPartial),
		"Source balance should remain -1050 after failed partial overcommit")

	// Destination balance should still be 1050
	require.Equal(t, 0, afterFailedOvercommitDest.Balance.Cmp(expectedDestBalanceAfterPartial),
		"Destination balance should remain 1050 after failed partial overcommit")

	// Inflight balance should remain unchanged at -50 for source
	require.Equal(t, 0, afterFailedOvercommitSource.InflightBalance.Cmp(expectedSourceInflightAfterPartial),
		"Source inflight balance should remain -50 after failed partial overcommit")

	// Inflight balance should remain unchanged at 50 for destination
	require.Equal(t, 0, afterFailedOvercommitDest.InflightBalance.Cmp(expectedDestInflightAfterPartial),
		"Destination inflight balance should remain 50 after failed partial overcommit")

	// Step 4: Full commit (remaining 50)
	fullCommitTxn, err := blnk.CommitInflightTransaction(ctx, inflightEntry.TransactionID, big.NewInt(0))
	require.NoError(t, err, "Full commit should succeed")
	require.Equal(t, StatusApplied, fullCommitTxn.Status, "Full commit transaction should have APPLIED status")

	// Verify balances after full commit
	afterFullCommitSource, err := ds.GetBalanceByIDLite(source.BalanceID)
	require.NoError(t, err, "Failed to get source balance after full commit")

	afterFullCommitDest, err := ds.GetBalanceByIDLite(dest.BalanceID)
	require.NoError(t, err, "Failed to get destination balance after full commit")

	// Source balance should be -1050 - 50 = -1100
	expectedSourceBalanceAfterFull := big.NewInt(int64((-initialBalance - inflightAmount) * 100))
	require.Equal(t, 0, afterFullCommitSource.Balance.Cmp(expectedSourceBalanceAfterFull),
		"Source balance should be -1100 after full commit")

	// Destination balance should be 1050 + 50 = 1100
	expectedDestBalanceAfterFull := big.NewInt(int64((initialBalance + inflightAmount) * 100))
	require.Equal(t, 0, afterFullCommitDest.Balance.Cmp(expectedDestBalanceAfterFull),
		"Destination balance should be 1100 after full commit")

	// Both inflight balances should be 0
	require.Equal(t, 0, afterFullCommitSource.InflightBalance.Cmp(big.NewInt(0)),
		"Source inflight balance should be zero after full commit")
	require.Equal(t, 0, afterFullCommitDest.InflightBalance.Cmp(big.NewInt(0)),
		"Destination inflight balance should be zero after full commit")

	// Step 5: Attempt another commit (should fail)
	_, err = blnk.CommitInflightTransaction(ctx, inflightEntry.TransactionID, big.NewInt(0))
	require.Error(t, err, "Second full commit should fail")
	require.Contains(t, err.Error(), "cannot commit. Transaction already committed",
		"Error should mention that transaction is already committed")

	// Verify balances remain unchanged after failed second commit
	afterFailedSecondCommitSource, err := ds.GetBalanceByIDLite(source.BalanceID)
	require.NoError(t, err, "Failed to get source balance after failed second commit")

	afterFailedSecondCommitDest, err := ds.GetBalanceByIDLite(dest.BalanceID)
	require.NoError(t, err, "Failed to get destination balance after failed second commit")

	require.Equal(t, 0, afterFailedSecondCommitSource.Balance.Cmp(expectedSourceBalanceAfterFull),
		"Source balance should remain -1100 after failed second commit")
	require.Equal(t, 0, afterFailedSecondCommitDest.Balance.Cmp(expectedDestBalanceAfterFull),
		"Destination balance should remain 1100 after failed second commit")
	require.Equal(t, 0, afterFailedSecondCommitSource.InflightBalance.Cmp(big.NewInt(0)),
		"Source inflight balance should remain zero after failed second commit")
	require.Equal(t, 0, afterFailedSecondCommitDest.InflightBalance.Cmp(big.NewInt(0)),
		"Destination inflight balance should remain zero after failed second commit")

	// Step 6: Attempt to void a fully committed transaction (should fail)
	_, err = blnk.VoidInflightTransaction(ctx, inflightEntry.TransactionID)
	require.Error(t, err, "Voiding a fully committed transaction should fail")
	require.Contains(t, err.Error(), "cannot void. Transaction already committed",
		"Error should mention that transaction is already committed")

	// Verify balances remain unchanged after failed void
	afterFailedVoidSource, err := ds.GetBalanceByIDLite(source.BalanceID)
	require.NoError(t, err, "Failed to get source balance after failed void")

	afterFailedVoidDest, err := ds.GetBalanceByIDLite(dest.BalanceID)
	require.NoError(t, err, "Failed to get destination balance after failed void")

	require.Equal(t, 0, afterFailedVoidSource.Balance.Cmp(expectedSourceBalanceAfterFull),
		"Source balance should remain -1100 after failed void")
	require.Equal(t, 0, afterFailedVoidDest.Balance.Cmp(expectedDestBalanceAfterFull),
		"Destination balance should remain 1100 after failed void")
	require.Equal(t, 0, afterFailedVoidSource.InflightBalance.Cmp(big.NewInt(0)),
		"Source inflight balance should remain zero after failed void")
	require.Equal(t, 0, afterFailedVoidDest.InflightBalance.Cmp(big.NewInt(0)),
		"Destination inflight balance should remain zero after failed void")

	// Verify transaction history
	txnHistory, err := ds.GetTransactionsByParent(ctx, inflightEntry.TransactionID, 10, 0)
	require.NoError(t, err, "Failed to get transaction history")
	require.Equal(t, 2, len(txnHistory), "Should have exactly two transactions: partial commit and full commit")

	// Check transaction history details
	for _, historyTxn := range txnHistory {
		require.Equal(t, StatusApplied, historyTxn.Status, "All history transactions should be APPLIED")
		require.Equal(t, "USD", historyTxn.Currency, "All history transactions should be in USD")
	}
}

func TestInflightTransactionFlowWithSkipQueueThenVoid(t *testing.T) {
	// Skip in short mode
	if testing.Short() {
		t.Skip("Skipping queue flow test in short mode")
	}

	ctx := context.Background()
	cnf := &config.Configuration{
		Redis: config.RedisConfig{
			Dns: "localhost:6379",
		},
		DataSource: config.DataSourceConfig{
			Dns: "postgres://postgres:password@localhost:5432/blnk?sslmode=disable",
		},
		Queue: config.QueueConfig{
			WebhookQueue:     "webhook_queue_test",
			IndexQueue:       "index_queue_test",
			TransactionQueue: "transaction_queue_test",
			NumberOfQueues:   1,
		},
		Server: config.ServerConfig{
			SecretKey: "test-secret",
		},
		Transaction: config.TransactionConfig{
			BatchSize:        100,
			MaxQueueSize:     1000,
			LockDuration:     time.Second * 30,
			IndexQueuePrefix: "test_index",
		},
	}
	config.ConfigStore.Store(cnf)

	ds, err := database.NewDataSource(cnf)
	require.NoError(t, err, "Failed to create datasource")

	blnk, err := NewBlnk(ds)
	require.NoError(t, err, "Failed to create Blnk instance")

	txnRef := "txn_" + model.GenerateUUIDWithSuffix("test")

	// Create test balances
	sourceBalance := &model.Balance{
		Currency: "USD",
		LedgerID: "general_ledger_id",
	}
	destBalance := &model.Balance{
		Currency: "USD",
		LedgerID: "general_ledger_id",
	}

	source, err := ds.CreateBalance(*sourceBalance)
	require.NoError(t, err, "Failed to create source balance")

	dest, err := ds.CreateBalance(*destBalance)
	require.NoError(t, err, "Failed to create destination balance")

	// Create transaction with skip_queue set to true
	txn := &model.Transaction{
		Reference:      txnRef,
		Source:         source.BalanceID,
		Destination:    dest.BalanceID,
		Amount:         500,
		Inflight:       true,
		Currency:       "USD",
		AllowOverdraft: true,
		Precision:      100,
		MetaData:       map[string]interface{}{"test": true},
		SkipQueue:      true, // Enable skip queue
	}

	// Queue the transaction
	queuedTxn, err := blnk.QueueTransaction(ctx, txn)
	require.NoError(t, err, "Failed to queue transaction")

	// Verify that the transaction was processed immediately
	require.Equal(t, StatusInflight, queuedTxn.Status, "Transaction should be INFLIGHT immediately when skip_queue is true")

	// Verify balances were updated immediately
	updatedSource, err := ds.GetBalanceByIDLite(source.BalanceID)
	require.NoError(t, err, "Failed to get updated source balance")

	updatedDest, err := ds.GetBalanceByIDLite(dest.BalanceID)
	require.NoError(t, err, "Failed to get updated destination balance")

	// Calculate expected balance changes
	expectedDebit := big.NewInt(int64(-500) * 100) // Amount * precision
	expectedCredit := big.NewInt(int64(500) * 100)

	// Verify balance changes
	require.Equal(t, 0, updatedSource.InflightBalance.Cmp(expectedDebit),
		"Source balance should be immediately reduced by transaction amount")
	require.Equal(t, 0, updatedDest.InflightBalance.Cmp(expectedCredit),
		"Destination balance should be immediately increased by transaction amount")

	// Verify inflight entry exists
	queuedEntry, err := ds.GetTransactionByRef(ctx, txnRef)
	require.NoError(t, err, "Failed to get queued transaction entry")
	require.Equal(t, StatusInflight, queuedEntry.Status, "Should have an INFLIGHT transaction entry")

	// Now void the transaction
	_, err = blnk.VoidInflightTransaction(ctx, queuedEntry.TransactionID)
	require.NoError(t, err, "Failed to void transaction")

	// Verify balances were reset after void
	updatedSourceAfterVoid, err := ds.GetBalanceByIDLite(source.BalanceID)
	require.NoError(t, err, "Failed to get updated source balance after void")

	updatedDestAfterVoid, err := ds.GetBalanceByIDLite(dest.BalanceID)
	require.NoError(t, err, "Failed to get updated destination balance after void")

	// Verify balance changes were reversed
	require.Equal(t, 0, updatedSourceAfterVoid.InflightBalance.Cmp(big.NewInt(0)),
		"Source inflight balance should be reset to zero after void")
	require.Equal(t, 0, updatedDestAfterVoid.InflightBalance.Cmp(big.NewInt(0)),
		"Destination inflight balance should be reset to zero after void")

	// Verify normal balance wasn't affected
	require.Equal(t, 0, updatedSourceAfterVoid.Balance.Cmp(big.NewInt(0)),
		"Source normal balance should remain zero")
	require.Equal(t, 0, updatedDestAfterVoid.Balance.Cmp(big.NewInt(0)),
		"Destination normal balance should remain zero")

	// Attempt to void again - should fail
	_, err = blnk.VoidInflightTransaction(ctx, queuedEntry.TransactionID)
	require.Error(t, err, "Voiding an already voided transaction should fail")
	require.Contains(t, err.Error(), "transaction has already been voided", "Error message should indicate transaction is already voided")
}

func TestMultipleSourcesTransactionFlowWithSkipQueue(t *testing.T) {
	// Skip in short mode
	if testing.Short() {
		t.Skip("Skipping queue flow test in short mode")
	}

	ctx := context.Background()
	cnf := &config.Configuration{
		Redis: config.RedisConfig{
			Dns: "localhost:6379",
		},
		DataSource: config.DataSourceConfig{
			Dns: "postgres://postgres:password@localhost:5432/blnk?sslmode=disable",
		},
		Queue: config.QueueConfig{
			WebhookQueue:     "webhook_queue_test",
			IndexQueue:       "index_queue_test",
			TransactionQueue: "transaction_queue_test",
			NumberOfQueues:   1,
		},
		Server: config.ServerConfig{
			SecretKey: "test-secret",
		},
		Transaction: config.TransactionConfig{
			BatchSize:        100,
			MaxQueueSize:     1000,
			LockDuration:     time.Second * 30,
			IndexQueuePrefix: "test_index",
		},
	}
	config.ConfigStore.Store(cnf)

	ds, err := database.NewDataSource(cnf)
	require.NoError(t, err, "Failed to create datasource")

	blnk, err := NewBlnk(ds)
	require.NoError(t, err, "Failed to create Blnk instance")

	txnRef := "txn_" + model.GenerateUUIDWithSuffix("test")

	// Create test balances
	sourceBalanceOne := &model.Balance{
		Currency: "USD",
		LedgerID: "general_ledger_id",
	}
	sourceBalanceTwo := &model.Balance{
		Currency: "USD",
		LedgerID: "general_ledger_id",
	}
	destBalance := &model.Balance{
		Currency: "USD",
		LedgerID: "general_ledger_id",
	}

	sourceOne, err := ds.CreateBalance(*sourceBalanceOne)
	require.NoError(t, err, "Failed to create source balance one")

	sourceTwo, err := ds.CreateBalance(*sourceBalanceTwo)
	require.NoError(t, err, "Failed to create source balance two")

	dest, err := ds.CreateBalance(*destBalance)
	require.NoError(t, err, "Failed to create destination balance")

	// Create transaction with skip_queue set to true
	txn := &model.Transaction{
		Reference: txnRef,
		Sources: []model.Distribution{
			{Identifier: sourceOne.BalanceID, Distribution: "50%"},
			{Identifier: sourceTwo.BalanceID, Distribution: "50%"},
		},
		Destination:    dest.BalanceID,
		Amount:         500,
		Inflight:       false,
		Currency:       "USD",
		AllowOverdraft: true,
		Precision:      100,
		MetaData:       map[string]interface{}{"test": true},
		SkipQueue:      true, // Enable skip queue
	}

	// Queue the transaction
	queuedTxn, err := blnk.QueueTransaction(ctx, txn)
	require.NoError(t, err, "Failed to queue transaction")

	// Verify that the transaction was processed immediately
	require.Equal(t, StatusApplied, queuedTxn.Status, "Transaction should be APPLIED immediately when skip_queue is true")

	// Verify balances were updated immediately
	updatedSourceOne, err := ds.GetBalanceByIDLite(sourceOne.BalanceID)
	require.NoError(t, err, "Failed to get updated source balance")

	updatedSourceTwo, err := ds.GetBalanceByIDLite(sourceTwo.BalanceID)
	require.NoError(t, err, "Failed to get updated source balance")

	updatedDest, err := ds.GetBalanceByIDLite(dest.BalanceID)
	require.NoError(t, err, "Failed to get updated destination balance")

	// Calculate expected balance changes
	expectedDebit := big.NewInt(int64(-250) * 100) // Amount * precision
	expectedCredit := big.NewInt(int64(500) * 100)

	// Verify balance changes
	require.Equal(t, 0, updatedSourceOne.Balance.Cmp(expectedDebit),
		"Source balance should be immediately reduced by transaction amount")

	require.Equal(t, 0, updatedSourceTwo.Balance.Cmp(expectedDebit),
		"Source balance should be immediately reduced by transaction amount")

	require.Equal(t, 0, updatedDest.Balance.Cmp(expectedCredit),
		"Destination balance should be immediately increased by transaction amount")

	// Verify no queued entry exists
	queuedEntryOne, err := ds.GetTransactionByRef(ctx, fmt.Sprintf("%s-1", txnRef))
	require.NoError(t, err, "Failed to get queued transaction entry")
	require.Equal(t, StatusApplied, queuedEntryOne.Status, "Should have an APPLIED transaction entry")

	queuedEntryTwo, err := ds.GetTransactionByRef(ctx, fmt.Sprintf("%s-2", txnRef))
	require.NoError(t, err, "Failed to get queued transaction entry")
	require.Equal(t, StatusApplied, queuedEntryTwo.Status, "Should have an APPLIED transaction entry")
}

func TestMultipleSourcesInflightTransactionFlowWithSkipQueueAndCommit(t *testing.T) {
	// Skip in short mode
	if testing.Short() {
		t.Skip("Skipping queue flow test in short mode")
	}

	ctx := context.Background()
	cnf := &config.Configuration{
		Redis: config.RedisConfig{
			Dns: "localhost:6379",
		},
		DataSource: config.DataSourceConfig{
			Dns: "postgres://postgres:password@localhost:5432/blnk?sslmode=disable",
		},
		Queue: config.QueueConfig{
			WebhookQueue:     "webhook_queue_test",
			IndexQueue:       "index_queue_test",
			TransactionQueue: "transaction_queue_test",
			NumberOfQueues:   1,
		},
		Server: config.ServerConfig{
			SecretKey: "test-secret",
		},
		Transaction: config.TransactionConfig{
			BatchSize:        100,
			MaxQueueSize:     1000,
			LockDuration:     time.Second * 30,
			IndexQueuePrefix: "test_index",
		},
	}
	config.ConfigStore.Store(cnf)

	ds, err := database.NewDataSource(cnf)
	require.NoError(t, err, "Failed to create datasource")

	blnk, err := NewBlnk(ds)
	require.NoError(t, err, "Failed to create Blnk instance")

	txnRef := "txn_" + model.GenerateUUIDWithSuffix("test")

	// Create test balances
	sourceBalanceOne := &model.Balance{
		Currency: "USD",
		LedgerID: "general_ledger_id",
	}
	sourceBalanceTwo := &model.Balance{
		Currency: "USD",
		LedgerID: "general_ledger_id",
	}
	destBalance := &model.Balance{
		Currency: "USD",
		LedgerID: "general_ledger_id",
	}

	sourceOne, err := ds.CreateBalance(*sourceBalanceOne)
	require.NoError(t, err, "Failed to create source balance one")

	sourceTwo, err := ds.CreateBalance(*sourceBalanceTwo)
	require.NoError(t, err, "Failed to create source balance two")

	dest, err := ds.CreateBalance(*destBalance)
	require.NoError(t, err, "Failed to create destination balance")

	// Create transaction with skip_queue set to true
	originalAmount := 500.0
	txn := &model.Transaction{
		Reference: txnRef,
		Sources: []model.Distribution{
			{Identifier: sourceOne.BalanceID, Distribution: "50%"},
			{Identifier: sourceTwo.BalanceID, Distribution: "50%"},
		},
		Destination:    dest.BalanceID,
		Amount:         originalAmount,
		Inflight:       true, // Set as inflight transaction
		Currency:       "USD",
		AllowOverdraft: true,
		Precision:      100,
		MetaData:       map[string]interface{}{"test": true},
		SkipQueue:      true, // Enable skip queue
	}

	// Queue the transaction
	queuedTxn, err := blnk.QueueTransaction(ctx, txn)
	require.NoError(t, err, "Failed to queue transaction")

	// Verify that the transaction was processed immediately
	require.Equal(t, StatusInflight, queuedTxn.Status, "Transaction should be INFLIGHT immediately when skip_queue is true")

	// Verify balances were updated immediately
	updatedSourceOne, err := ds.GetBalanceByIDLite(sourceOne.BalanceID)
	require.NoError(t, err, "Failed to get updated source balance one")

	updatedSourceTwo, err := ds.GetBalanceByIDLite(sourceTwo.BalanceID)
	require.NoError(t, err, "Failed to get updated source balance two")

	updatedDest, err := ds.GetBalanceByIDLite(dest.BalanceID)
	require.NoError(t, err, "Failed to get updated destination balance")

	// Calculate expected balance changes
	sourceOneShare := originalAmount * 0.5       // 50% of 500 = 250
	sourceTwoShare := originalAmount * 0.5       // 50% of 500 = 250
	sourceOnePartialAmount := sourceOneShare / 2 // Half of source one's share = 125
	sourceTwoPartialAmount := sourceTwoShare / 2 // Half of source two's share = 125

	// Calculate expected inflight balances
	expectedSourceOneInflightDebit := big.NewInt(int64(-sourceOneShare) * 100) // Amount * precision
	expectedSourceTwoInflightDebit := big.NewInt(int64(-sourceTwoShare) * 100) // Amount * precision
	expectedDestInflightCredit := big.NewInt(int64(originalAmount) * 100)      // Full amount to destination

	// Verify inflight balance changes
	require.Equal(t, 0, updatedSourceOne.InflightBalance.Cmp(expectedSourceOneInflightDebit),
		"Source one inflight balance should be immediately reduced by its share of transaction amount")

	require.Equal(t, 0, updatedSourceTwo.InflightBalance.Cmp(expectedSourceTwoInflightDebit),
		"Source two inflight balance should be immediately reduced by its share of transaction amount")

	require.Equal(t, 0, updatedDest.InflightBalance.Cmp(expectedDestInflightCredit),
		"Destination inflight balance should be immediately increased by full transaction amount")

	// Verify actual balances are not affected yet
	require.Equal(t, 0, updatedSourceOne.Balance.Cmp(big.NewInt(0)),
		"Source one actual balance should not be affected yet")

	require.Equal(t, 0, updatedSourceTwo.Balance.Cmp(big.NewInt(0)),
		"Source two actual balance should not be affected yet")

	require.Equal(t, 0, updatedDest.Balance.Cmp(big.NewInt(0)),
		"Destination actual balance should not be affected yet")

	// For multi-source transactions, there are separate entries for each source
	// Get the inflight transaction entries for each source
	inflightEntryOne, err := ds.GetTransactionByRef(ctx, fmt.Sprintf("%s-1", txnRef))
	require.NoError(t, err, "Failed to get inflight transaction entry for source one")
	require.Equal(t, StatusInflight, inflightEntryOne.Status, "Should have an INFLIGHT transaction entry for source one")

	inflightEntryTwo, err := ds.GetTransactionByRef(ctx, fmt.Sprintf("%s-2", txnRef))
	require.NoError(t, err, "Failed to get inflight transaction entry for source two")
	require.Equal(t, StatusInflight, inflightEntryTwo.Status, "Should have an INFLIGHT transaction entry for source two")

	// Partially commit the transaction (commit half of the total amount)
	// Since the original distribution is 50/50, each source has 250 of the original 500
	// When committing half of the total, that's 250, which is 125 from each source
	partialAmount := originalAmount / 2 // 250.0 total (125 from each source)

	sourceOnePartialPreciseAmount := new(big.Int).SetInt64(int64(sourceOnePartialAmount * txn.Precision))
	partialCommitTxnOne, err := blnk.CommitInflightTransaction(ctx, inflightEntryOne.TransactionID, sourceOnePartialPreciseAmount)
	require.NoError(t, err, "Failed to partially commit transaction for source one")
	require.Equal(t, StatusApplied, partialCommitTxnOne.Status, "Partial commit transaction should have APPLIED status")
	require.Equal(t, sourceOnePartialAmount, partialCommitTxnOne.Amount, "Partial commit amount should match specified amount")

	sourceTwoPartialPreciseAmount := new(big.Int).SetInt64(int64(sourceTwoPartialAmount * txn.Precision))
	partialCommitTxnTwo, err := blnk.CommitInflightTransaction(ctx, inflightEntryTwo.TransactionID, sourceTwoPartialPreciseAmount)
	require.NoError(t, err, "Failed to partially commit transaction for source two")
	require.Equal(t, StatusApplied, partialCommitTxnTwo.Status, "Partial commit transaction should have APPLIED status")
	require.Equal(t, sourceTwoPartialAmount, partialCommitTxnTwo.Amount, "Partial commit amount should match specified amount")

	// Verify balances were updated after partial commit
	updatedSourceOneAfterPartialCommit, err := ds.GetBalanceByIDLite(sourceOne.BalanceID)
	require.NoError(t, err, "Failed to get updated source one balance")

	updatedSourceTwoAfterPartialCommit, err := ds.GetBalanceByIDLite(sourceTwo.BalanceID)
	require.NoError(t, err, "Failed to get updated source two balance")

	updatedDestAfterPartialCommit, err := ds.GetBalanceByIDLite(dest.BalanceID)
	require.NoError(t, err, "Failed to get updated destination balance")

	expectedSourceOnePartialDebit := big.NewInt(int64(-sourceOnePartialAmount) * 100)
	expectedSourceTwoPartialDebit := big.NewInt(int64(-sourceTwoPartialAmount) * 100)
	expectedDestPartialCredit := big.NewInt(int64(partialAmount) * 100)

	// Inflight balance should be reduced by the committed amount
	expectedSourceOneRemainingInflightDebit := big.NewInt(0).Sub(expectedSourceOneInflightDebit, expectedSourceOnePartialDebit)
	expectedSourceTwoRemainingInflightDebit := big.NewInt(0).Sub(expectedSourceTwoInflightDebit, expectedSourceTwoPartialDebit)
	expectedDestRemainingInflightCredit := big.NewInt(0).Sub(expectedDestInflightCredit, expectedDestPartialCredit)

	// Verify actual balance changes
	require.Equal(t, 0, updatedSourceOneAfterPartialCommit.Balance.Cmp(expectedSourceOnePartialDebit),
		"Source one balance should reflect its share of the partial commit amount")
	require.Equal(t, 0, updatedSourceTwoAfterPartialCommit.Balance.Cmp(expectedSourceTwoPartialDebit),
		"Source two balance should reflect its share of the partial commit amount")
	require.Equal(t, 0, updatedDestAfterPartialCommit.Balance.Cmp(expectedDestPartialCredit),
		"Destination balance should reflect the partial commit amount")

	// Verify inflight balance changes
	require.Equal(t, 0, updatedSourceOneAfterPartialCommit.InflightBalance.Cmp(expectedSourceOneRemainingInflightDebit),
		"Source one inflight balance should be reduced by its share of the committed amount")
	require.Equal(t, 0, updatedSourceTwoAfterPartialCommit.InflightBalance.Cmp(expectedSourceTwoRemainingInflightDebit),
		"Source two inflight balance should be reduced by its share of the committed amount")
	require.Equal(t, 0, updatedDestAfterPartialCommit.InflightBalance.Cmp(expectedDestRemainingInflightCredit),
		"Destination inflight balance should be reduced by the committed amount")

	// Commit the remaining amount for both sources
	remainingCommitTxnOne, err := blnk.CommitInflightTransaction(ctx, inflightEntryOne.TransactionID, big.NewInt(0)) // 0 means commit remaining amount
	require.NoError(t, err, "Failed to commit remaining transaction amount for source one")
	require.Equal(t, StatusApplied, remainingCommitTxnOne.Status, "Remaining commit transaction should have APPLIED status")

	remainingCommitTxnTwo, err := blnk.CommitInflightTransaction(ctx, inflightEntryTwo.TransactionID, big.NewInt(0)) // 0 means commit remaining amount
	require.NoError(t, err, "Failed to commit remaining transaction amount for source two")
	require.Equal(t, StatusApplied, remainingCommitTxnTwo.Status, "Remaining commit transaction should have APPLIED status")

	// Verify the remaining commit amount for each source
	sourceOneRemainingAmount := sourceOneShare - sourceOnePartialAmount
	sourceTwoRemainingAmount := sourceTwoShare - sourceTwoPartialAmount

	require.InDelta(t, sourceOneRemainingAmount, remainingCommitTxnOne.Amount, 0.01,
		"Remaining commit amount for source one should match expected remaining amount")
	require.InDelta(t, sourceTwoRemainingAmount, remainingCommitTxnTwo.Amount, 0.01,
		"Remaining commit amount for source two should match expected remaining amount")

	// Verify final balances
	updatedSourceOneAfterFullCommit, err := ds.GetBalanceByIDLite(sourceOne.BalanceID)
	require.NoError(t, err, "Failed to get final source one balance")

	updatedSourceTwoAfterFullCommit, err := ds.GetBalanceByIDLite(sourceTwo.BalanceID)
	require.NoError(t, err, "Failed to get final source two balance")

	updatedDestAfterFullCommit, err := ds.GetBalanceByIDLite(dest.BalanceID)
	require.NoError(t, err, "Failed to get final destination balance")

	// Calculate expected final balances
	expectedSourceOneFinalDebit := big.NewInt(int64(-sourceOneShare) * 100) // Full share for source one
	expectedSourceTwoFinalDebit := big.NewInt(int64(-sourceTwoShare) * 100) // Full share for source two
	expectedDestFinalCredit := big.NewInt(int64(originalAmount) * 100)      // Full amount to destination

	// Verify final balance changes
	require.Equal(t, 0, updatedSourceOneAfterFullCommit.Balance.Cmp(expectedSourceOneFinalDebit),
		"Source one balance should ultimately reflect its full share amount")
	require.Equal(t, 0, updatedSourceTwoAfterFullCommit.Balance.Cmp(expectedSourceTwoFinalDebit),
		"Source two balance should ultimately reflect its full share amount")
	require.Equal(t, 0, updatedDestAfterFullCommit.Balance.Cmp(expectedDestFinalCredit),
		"Destination balance should ultimately reflect the full amount")

	// Verify all inflight balances are zero after full commit
	require.Equal(t, 0, updatedSourceOneAfterFullCommit.InflightBalance.Cmp(big.NewInt(0)),
		"Source one inflight balance should be zero after full commit")
	require.Equal(t, 0, updatedSourceTwoAfterFullCommit.InflightBalance.Cmp(big.NewInt(0)),
		"Source two inflight balance should be zero after full commit")
	require.Equal(t, 0, updatedDestAfterFullCommit.InflightBalance.Cmp(big.NewInt(0)),
		"Destination inflight balance should be zero after full commit")

	// Attempt another commit on each source (should fail)
	_, err = blnk.CommitInflightTransaction(ctx, inflightEntryOne.TransactionID, big.NewInt(0))
	require.Error(t, err, "Committing a fully committed transaction should fail")
	require.Contains(t, err.Error(), "cannot commit. Transaction already committed",
		"Error message should indicate transaction is already committed")

	_, err = blnk.CommitInflightTransaction(ctx, inflightEntryTwo.TransactionID, big.NewInt(0))
	require.Error(t, err, "Committing a fully committed transaction should fail")
	require.Contains(t, err.Error(), "cannot commit. Transaction already committed",
		"Error message should indicate transaction is already committed")

	// Verify commit history for both sources
	commitHistoryOne, err := ds.GetTransactionsByParent(ctx, inflightEntryOne.TransactionID, 5, 0)
	require.NoError(t, err, "Failed to get commit history for source one")
	require.Equal(t, 2, len(commitHistoryOne), "Should have exactly two commit transactions for source one")

	commitHistoryTwo, err := ds.GetTransactionsByParent(ctx, inflightEntryTwo.TransactionID, 5, 0)
	require.NoError(t, err, "Failed to get commit history for source two")
	require.Equal(t, 2, len(commitHistoryTwo), "Should have exactly two commit transactions for source two")

	// Verify transactions exist with correct statuses
	// Each source transaction will have a partial commit and a final commit

	// Since we already have the commit histories for both sources, we can use them to verify
	// the partial commits rather than trying to fetch them directly

	// Get all commit transactions for each source
	for i := range commitHistoryOne {
		require.Equal(t, StatusApplied, commitHistoryOne[i].Status,
			"Commit transaction for source one should have APPLIED status")
		require.Equal(t, inflightEntryOne.TransactionID, commitHistoryOne[i].ParentTransaction,
			"Commit should reference original source one transaction")
	}

	for i := range commitHistoryTwo {
		require.Equal(t, StatusApplied, commitHistoryTwo[i].Status,
			"Commit transaction for source two should have APPLIED status")
		require.Equal(t, inflightEntryTwo.TransactionID, commitHistoryTwo[i].ParentTransaction,
			"Commit should reference original source two transaction")
	}

	// Sort transactions by created time to identify first and second commit for each source
	sort.Slice(commitHistoryOne, func(i, j int) bool {
		return commitHistoryOne[i].CreatedAt.Before(commitHistoryOne[j].CreatedAt)
	})

	sort.Slice(commitHistoryTwo, func(i, j int) bool {
		return commitHistoryTwo[i].CreatedAt.Before(commitHistoryTwo[j].CreatedAt)
	})

	// Verify both commits exist for each source
	sourceOneFirstCommit := commitHistoryOne[0]
	sourceOneSecondCommit := commitHistoryOne[1]
	require.NotNil(t, sourceOneFirstCommit, "Should have found the first commit transaction for source one")
	require.NotNil(t, sourceOneSecondCommit, "Should have found the second commit transaction for source one")

	sourceTwoFirstCommit := commitHistoryTwo[0]
	sourceTwoSecondCommit := commitHistoryTwo[1]
	require.NotNil(t, sourceTwoFirstCommit, "Should have found the first commit transaction for source two")
	require.NotNil(t, sourceTwoSecondCommit, "Should have found the second commit transaction for source two")

	// The sum of both commit amounts should equal each source's share
	sourceOneTotalCommittedAmount := sourceOneFirstCommit.Amount + sourceOneSecondCommit.Amount
	sourceTwoTotalCommittedAmount := sourceTwoFirstCommit.Amount + sourceTwoSecondCommit.Amount

	require.InDelta(t, sourceOneShare, sourceOneTotalCommittedAmount, 0.01,
		"Total committed amount for source one should equal its share")
	require.InDelta(t, sourceTwoShare, sourceTwoTotalCommittedAmount, 0.01,
		"Total committed amount for source two should equal its share")
}

func TestMultipleDestinationTransactionFlowWithSkipQueue(t *testing.T) {
	// Skip in short mode
	if testing.Short() {
		t.Skip("Skipping queue flow test in short mode")
	}

	ctx := context.Background()
	cnf := &config.Configuration{
		Redis: config.RedisConfig{
			Dns: "localhost:6379",
		},
		DataSource: config.DataSourceConfig{
			Dns: "postgres://postgres:password@localhost:5432/blnk?sslmode=disable",
		},
		Queue: config.QueueConfig{
			WebhookQueue:     "webhook_queue_test",
			IndexQueue:       "index_queue_test",
			TransactionQueue: "transaction_queue_test",
			NumberOfQueues:   1,
		},
		Server: config.ServerConfig{
			SecretKey: "test-secret",
		},
		Transaction: config.TransactionConfig{
			BatchSize:        100,
			MaxQueueSize:     1000,
			LockDuration:     time.Second * 30,
			IndexQueuePrefix: "test_index",
		},
	}
	config.ConfigStore.Store(cnf)

	ds, err := database.NewDataSource(cnf)
	require.NoError(t, err, "Failed to create datasource")

	blnk, err := NewBlnk(ds)
	require.NoError(t, err, "Failed to create Blnk instance")

	txnRef := "txn_" + model.GenerateUUIDWithSuffix("test")

	// Create test balances
	destinationBalanceOne := &model.Balance{
		Currency: "USD",
		LedgerID: "general_ledger_id",
	}
	destinationBalanceTwo := &model.Balance{
		Currency: "USD",
		LedgerID: "general_ledger_id",
	}
	destBalance := &model.Balance{
		Currency: "USD",
		LedgerID: "general_ledger_id",
	}

	destinationOne, err := ds.CreateBalance(*destinationBalanceOne)
	require.NoError(t, err, "Failed to create destination balance one")

	destinationTwo, err := ds.CreateBalance(*destinationBalanceTwo)
	require.NoError(t, err, "Failed to create destination balance two")

	source, err := ds.CreateBalance(*destBalance)
	require.NoError(t, err, "Failed to create destination balance")

	// Create transaction with skip_queue set to true
	txn := &model.Transaction{
		Reference: txnRef,
		Destinations: []model.Distribution{
			{Identifier: destinationOne.BalanceID, Distribution: "50%"},
			{Identifier: destinationTwo.BalanceID, Distribution: "50%"},
		},
		Source:         source.BalanceID,
		Amount:         500,
		Inflight:       false,
		Currency:       "USD",
		AllowOverdraft: true,
		Precision:      100,
		MetaData:       map[string]interface{}{"test": true},
		SkipQueue:      true, // Enable skip queue
	}

	// Queue the transaction
	queuedTxn, err := blnk.QueueTransaction(ctx, txn)
	require.NoError(t, err, "Failed to queue transaction")

	// Verify that the transaction was processed immediately
	require.Equal(t, StatusApplied, queuedTxn.Status, "Transaction should be APPLIED immediately when skip_queue is true")

	// Verify balances were updated immediately
	updatedDestinationOne, err := ds.GetBalanceByIDLite(destinationOne.BalanceID)
	require.NoError(t, err, "Failed to get updated destination balance")

	updatedDestinationTwo, err := ds.GetBalanceByIDLite(destinationTwo.BalanceID)
	require.NoError(t, err, "Failed to get updated destination balance")

	updatedSource, err := ds.GetBalanceByIDLite(source.BalanceID)
	require.NoError(t, err, "Failed to get updated source balance")

	// Calculate expected balance changes
	expectedCredit := big.NewInt(int64(250) * 100) // Amount * precision
	expectedDebit := big.NewInt(int64(-500) * 100)

	// Verify balance changes
	require.Equal(t, 0, updatedDestinationOne.Balance.Cmp(expectedCredit),
		"destination balance should be immediately reduced by transaction amount")

	require.Equal(t, 0, updatedDestinationTwo.Balance.Cmp(expectedCredit),
		"destination balance should be immediately reduced by transaction amount")

	require.Equal(t, 0, updatedSource.Balance.Cmp(expectedDebit),
		"Source balance should be immediately increased by transaction amount")

	// Verify no queued entry exists
	queuedEntryOne, err := ds.GetTransactionByRef(ctx, fmt.Sprintf("%s-1", txnRef))
	require.NoError(t, err, "Failed to get queued transaction entry")
	require.Equal(t, StatusApplied, queuedEntryOne.Status, "Should have an APPLIED transaction entry")

	queuedEntryTwo, err := ds.GetTransactionByRef(ctx, fmt.Sprintf("%s-2", txnRef))
	require.NoError(t, err, "Failed to get queued transaction entry")
	require.Equal(t, StatusApplied, queuedEntryTwo.Status, "Should have an APPLIED transaction entry")
}

func TestMultipleDestinationTransactionFlowWithTwoDistributions(t *testing.T) {
	// Skip in short mode
	if testing.Short() {
		t.Skip("Skipping queue flow test in short mode")
	}

	ctx := context.Background()
	cnf := &config.Configuration{
		Redis: config.RedisConfig{
			Dns: "localhost:6379",
		},
		DataSource: config.DataSourceConfig{
			Dns: "postgres://postgres:password@localhost:5432/blnk?sslmode=disable",
		},
		Queue: config.QueueConfig{
			WebhookQueue:     "webhook_queue_test",
			IndexQueue:       "index_queue_test",
			TransactionQueue: "transaction_queue_test",
			NumberOfQueues:   1,
		},
		Server: config.ServerConfig{
			SecretKey: "test-secret",
		},
		Transaction: config.TransactionConfig{
			BatchSize:        100,
			MaxQueueSize:     1000,
			LockDuration:     time.Second * 30,
			IndexQueuePrefix: "test_index",
		},
	}
	config.ConfigStore.Store(cnf)

	ds, err := database.NewDataSource(cnf)
	require.NoError(t, err, "Failed to create datasource")

	blnk, err := NewBlnk(ds)
	require.NoError(t, err, "Failed to create Blnk instance")

	txnRef := "txn_" + model.GenerateUUIDWithSuffix("test")

	// Create test balances for two destinations
	destinationBalanceOne := &model.Balance{
		Currency: "USD",
		LedgerID: "general_ledger_id",
	}
	destinationBalanceTwo := &model.Balance{
		Currency: "USD",
		LedgerID: "general_ledger_id",
	}
	sourceBalance := &model.Balance{
		Currency: "USD",
		LedgerID: "general_ledger_id",
	}

	destinationOne, err := ds.CreateBalance(*destinationBalanceOne)
	require.NoError(t, err, "Failed to create destination balance one")

	destinationTwo, err := ds.CreateBalance(*destinationBalanceTwo)
	require.NoError(t, err, "Failed to create destination balance two")

	source, err := ds.CreateBalance(*sourceBalance)
	require.NoError(t, err, "Failed to create source balance")

	// Transaction amount
	amount := 3.0

	// Create transaction with two destinations
	txn := &model.Transaction{
		Reference: txnRef,
		Destinations: []model.Distribution{
			{Identifier: destinationOne.BalanceID, Distribution: "4.300000%"}, // First gets 4.3%
			{Identifier: destinationTwo.BalanceID, Distribution: "left"},      // Second gets remaining
		},
		Source:         source.BalanceID,
		Amount:         amount,
		Inflight:       false,
		Currency:       "USD",
		AllowOverdraft: true,
		Precision:      10000000000,
		MetaData:       map[string]interface{}{"test": true},
		SkipQueue:      true, // Enable skip queue for immediate processing
	}

	// Queue the transaction
	queuedTxn, err := blnk.QueueTransaction(ctx, txn)
	require.NoError(t, err, "Failed to queue transaction")

	// Verify that the transaction was processed immediately
	require.Equal(t, StatusApplied, queuedTxn.Status, "Transaction should be APPLIED immediately when skip_queue is true")

	// Verify balances were updated immediately
	updatedDestinationOne, err := ds.GetBalanceByIDLite(destinationOne.BalanceID)
	require.NoError(t, err, "Failed to get updated destination balance one")

	updatedDestinationTwo, err := ds.GetBalanceByIDLite(destinationTwo.BalanceID)
	require.NoError(t, err, "Failed to get updated destination balance two")

	updatedSource, err := ds.GetBalanceByIDLite(source.BalanceID)
	require.NoError(t, err, "Failed to get updated source balance")

	// Calculate expected balance changes
	precision := int64(10000000000)

	// First destination: 4.3% of 3 = 0.129
	firstDistributionAmount := int64(float64(amount) * 0.043 * float64(precision)) // Amount * percentage * precision
	expectedFirstCredit := big.NewInt(firstDistributionAmount)

	// Second destination: remaining amount (100% - 4.3%) = 95.7% of 3 = 2.871
	secondDistributionAmount := int64(amount)*precision - firstDistributionAmount
	expectedSecondCredit := big.NewInt(secondDistributionAmount)

	// Source: debit of total amount
	expectedDebit := big.NewInt(int64(-amount) * precision)

	// Verify balance changes
	require.Equal(t, 0, updatedDestinationOne.Balance.Cmp(expectedFirstCredit),
		"First destination balance should be immediately credited with 4.3% of transaction amount")

	require.Equal(t, 0, updatedDestinationTwo.Balance.Cmp(expectedSecondCredit),
		"Second destination balance should be immediately credited with remaining amount (95.7%)")

	require.Equal(t, 0, updatedSource.Balance.Cmp(expectedDebit),
		"Source balance should be immediately debited by full transaction amount")

	// Verify transaction entries exist and are applied
	queuedEntryOne, err := ds.GetTransactionByRef(ctx, fmt.Sprintf("%s-1", txnRef))
	require.NoError(t, err, "Failed to get queued transaction entry one")
	require.Equal(t, StatusApplied, queuedEntryOne.Status, "Should have an APPLIED transaction entry for first destination")

	queuedEntryTwo, err := ds.GetTransactionByRef(ctx, fmt.Sprintf("%s-2", txnRef))
	require.NoError(t, err, "Failed to get queued transaction entry two")
	require.Equal(t, StatusApplied, queuedEntryTwo.Status, "Should have an APPLIED transaction entry for second destination")
}

func TestQueueTransactionFlowWithSkipQueueAndPreciseAmount(t *testing.T) {
	// Skip in short mode
	if testing.Short() {
		t.Skip("Skipping queue flow test in short mode")
	}

	ctx := context.Background()
	cnf := &config.Configuration{
		Redis: config.RedisConfig{
			Dns: "localhost:6379",
		},
		DataSource: config.DataSourceConfig{
			Dns: "postgres://postgres:password@localhost:5432/blnk?sslmode=disable",
		},
		Queue: config.QueueConfig{
			WebhookQueue:     "webhook_queue_test",
			IndexQueue:       "index_queue_test",
			TransactionQueue: "transaction_queue_test",
			NumberOfQueues:   1,
		},
		Server: config.ServerConfig{
			SecretKey: "test-secret",
		},
		Transaction: config.TransactionConfig{
			BatchSize:        100,
			MaxQueueSize:     1000,
			LockDuration:     time.Second * 30,
			IndexQueuePrefix: "test_index",
		},
	}
	config.ConfigStore.Store(cnf)

	ds, err := database.NewDataSource(cnf)
	require.NoError(t, err, "Failed to create datasource")

	blnk, err := NewBlnk(ds)
	require.NoError(t, err, "Failed to create Blnk instance")

	txnRef := "txn_" + model.GenerateUUIDWithSuffix("test")

	// Create test balances
	sourceBalance := &model.Balance{
		Currency: "USD",
		LedgerID: "general_ledger_id",
	}
	destBalance := &model.Balance{
		Currency: "USD",
		LedgerID: "general_ledger_id",
	}

	source, err := ds.CreateBalance(*sourceBalance)
	require.NoError(t, err, "Failed to create source balance")

	dest, err := ds.CreateBalance(*destBalance)
	require.NoError(t, err, "Failed to create destination balance")

	// Create transaction with skip_queue set to true
	txn := &model.Transaction{
		Reference:      txnRef,
		Source:         source.BalanceID,
		Destination:    dest.BalanceID,
		PreciseAmount:  big.NewInt(50000),
		Currency:       "USD",
		AllowOverdraft: true,
		Precision:      100,
		MetaData:       map[string]interface{}{"test": true},
		SkipQueue:      true, // Enable skip queue
	}

	// Queue the transaction
	queuedTxn, err := blnk.QueueTransaction(ctx, txn)
	require.NoError(t, err, "Failed to queue transaction")

	// Verify that the transaction was processed immediately
	require.Equal(t, StatusApplied, queuedTxn.Status, "Transaction should be APPLIED immediately when skip_queue is true")

	// Verify balances were updated immediately
	updatedSource, err := ds.GetBalanceByIDLite(source.BalanceID)
	require.NoError(t, err, "Failed to get updated source balance")

	updatedDest, err := ds.GetBalanceByIDLite(dest.BalanceID)
	require.NoError(t, err, "Failed to get updated destination balance")

	// Calculate expected balance changes
	expectedDebit := big.NewInt(int64(-500) * 100) // Amount * precision
	expectedCredit := big.NewInt(int64(500) * 100)

	// Verify balance changes
	require.Equal(t, 0, updatedSource.Balance.Cmp(expectedDebit),
		"Source balance should be immediately reduced by transaction amount")
	require.Equal(t, 0, updatedDest.Balance.Cmp(expectedCredit),
		"Destination balance should be immediately increased by transaction amount")

	// Verify no queued entry exists
	queuedEntry, err := ds.GetTransactionByRef(ctx, txnRef)
	require.NoError(t, err, "Failed to get queued transaction entry")
	require.Equal(t, StatusApplied, queuedEntry.Status, "Should have an APPLIED transaction entry")
}

func TestMultipleSourcesTransactionFlowWithSkipQueueAndPreciseAmount(t *testing.T) {
	// Skip in short mode
	if testing.Short() {
		t.Skip("Skipping queue flow test in short mode")
	}

	ctx := context.Background()
	cnf := &config.Configuration{
		Redis: config.RedisConfig{
			Dns: "localhost:6379",
		},
		DataSource: config.DataSourceConfig{
			Dns: "postgres://postgres:password@localhost:5432/blnk?sslmode=disable",
		},
		Queue: config.QueueConfig{
			WebhookQueue:     "webhook_queue_test",
			IndexQueue:       "index_queue_test",
			TransactionQueue: "transaction_queue_test",
			NumberOfQueues:   1,
		},
		Server: config.ServerConfig{
			SecretKey: "test-secret",
		},
		Transaction: config.TransactionConfig{
			BatchSize:        100,
			MaxQueueSize:     1000,
			LockDuration:     time.Second * 30,
			IndexQueuePrefix: "test_index",
		},
	}
	config.ConfigStore.Store(cnf)

	ds, err := database.NewDataSource(cnf)
	require.NoError(t, err, "Failed to create datasource")

	blnk, err := NewBlnk(ds)
	require.NoError(t, err, "Failed to create Blnk instance")

	txnRef := "txn_" + model.GenerateUUIDWithSuffix("test")

	// Create test balances
	sourceBalanceOne := &model.Balance{
		Currency: "USD",
		LedgerID: "general_ledger_id",
	}
	sourceBalanceTwo := &model.Balance{
		Currency: "USD",
		LedgerID: "general_ledger_id",
	}
	destBalance := &model.Balance{
		Currency: "USD",
		LedgerID: "general_ledger_id",
	}

	sourceOne, err := ds.CreateBalance(*sourceBalanceOne)
	require.NoError(t, err, "Failed to create source balance one")

	sourceTwo, err := ds.CreateBalance(*sourceBalanceTwo)
	require.NoError(t, err, "Failed to create source balance two")

	dest, err := ds.CreateBalance(*destBalance)
	require.NoError(t, err, "Failed to create destination balance")

	// Create transaction with skip_queue set to true
	txn := &model.Transaction{
		Reference: txnRef,
		Sources: []model.Distribution{
			{Identifier: sourceOne.BalanceID, Distribution: "50%"},
			{Identifier: sourceTwo.BalanceID, Distribution: "50%"},
		},
		Destination:    dest.BalanceID,
		PreciseAmount:  big.NewInt(50000),
		Inflight:       false,
		Currency:       "USD",
		AllowOverdraft: true,
		Precision:      100,
		MetaData:       map[string]interface{}{"test": true},
		SkipQueue:      true, // Enable skip queue
	}

	// Queue the transaction
	queuedTxn, err := blnk.QueueTransaction(ctx, txn)
	require.NoError(t, err, "Failed to queue transaction")

	// Verify that the transaction was processed immediately
	require.Equal(t, StatusApplied, queuedTxn.Status, "Transaction should be APPLIED immediately when skip_queue is true")

	// Verify balances were updated immediately
	updatedSourceOne, err := ds.GetBalanceByIDLite(sourceOne.BalanceID)
	require.NoError(t, err, "Failed to get updated source balance")

	updatedSourceTwo, err := ds.GetBalanceByIDLite(sourceTwo.BalanceID)
	require.NoError(t, err, "Failed to get updated source balance")

	updatedDest, err := ds.GetBalanceByIDLite(dest.BalanceID)
	require.NoError(t, err, "Failed to get updated destination balance")

	// Calculate expected balance changes
	expectedDebit := big.NewInt(int64(-250) * 100) // Amount * precision
	expectedCredit := big.NewInt(int64(500) * 100)

	// Verify balance changes
	require.Equal(t, 0, updatedSourceOne.Balance.Cmp(expectedDebit),
		"Source balance should be immediately reduced by transaction amount")

	require.Equal(t, 0, updatedSourceTwo.Balance.Cmp(expectedDebit),
		"Source balance should be immediately reduced by transaction amount")

	require.Equal(t, 0, updatedDest.Balance.Cmp(expectedCredit),
		"Destination balance should be immediately increased by transaction amount")

	// Verify no queued entry exists
	queuedEntryOne, err := ds.GetTransactionByRef(ctx, fmt.Sprintf("%s-1", txnRef))
	require.NoError(t, err, "Failed to get queued transaction entry")
	require.Equal(t, StatusApplied, queuedEntryOne.Status, "Should have an APPLIED transaction entry")

	queuedEntryTwo, err := ds.GetTransactionByRef(ctx, fmt.Sprintf("%s-2", txnRef))
	require.NoError(t, err, "Failed to get queued transaction entry")
	require.Equal(t, StatusApplied, queuedEntryTwo.Status, "Should have an APPLIED transaction entry")
}

func TestMultipleDestinationTransactionFlowWithSkipQueueWithPreciseAmount(t *testing.T) {
	// Skip in short mode
	if testing.Short() {
		t.Skip("Skipping queue flow test in short mode")
	}

	ctx := context.Background()
	cnf := &config.Configuration{
		Redis: config.RedisConfig{
			Dns: "localhost:6379",
		},
		DataSource: config.DataSourceConfig{
			Dns: "postgres://postgres:password@localhost:5432/blnk?sslmode=disable",
		},
		Queue: config.QueueConfig{
			WebhookQueue:     "webhook_queue_test",
			IndexQueue:       "index_queue_test",
			TransactionQueue: "transaction_queue_test",
			NumberOfQueues:   1,
		},
		Server: config.ServerConfig{
			SecretKey: "test-secret",
		},
		Transaction: config.TransactionConfig{
			BatchSize:        100,
			MaxQueueSize:     1000,
			LockDuration:     time.Second * 30,
			IndexQueuePrefix: "test_index",
		},
	}
	config.ConfigStore.Store(cnf)

	ds, err := database.NewDataSource(cnf)
	require.NoError(t, err, "Failed to create datasource")

	blnk, err := NewBlnk(ds)
	require.NoError(t, err, "Failed to create Blnk instance")

	txnRef := "txn_" + model.GenerateUUIDWithSuffix("test")

	// Create test balances
	destinationBalanceOne := &model.Balance{
		Currency: "USD",
		LedgerID: "general_ledger_id",
	}
	destinationBalanceTwo := &model.Balance{
		Currency: "USD",
		LedgerID: "general_ledger_id",
	}
	destBalance := &model.Balance{
		Currency: "USD",
		LedgerID: "general_ledger_id",
	}

	destinationOne, err := ds.CreateBalance(*destinationBalanceOne)
	require.NoError(t, err, "Failed to create destination balance one")

	destinationTwo, err := ds.CreateBalance(*destinationBalanceTwo)
	require.NoError(t, err, "Failed to create destination balance two")

	source, err := ds.CreateBalance(*destBalance)
	require.NoError(t, err, "Failed to create destination balance")

	// Create transaction with skip_queue set to true
	txn := &model.Transaction{
		Reference: txnRef,
		Destinations: []model.Distribution{
			{Identifier: destinationOne.BalanceID, Distribution: "50%"},
			{Identifier: destinationTwo.BalanceID, Distribution: "50%"},
		},
		Source:         source.BalanceID,
		PreciseAmount:  big.NewInt(50000),
		Inflight:       false,
		Currency:       "USD",
		AllowOverdraft: true,
		Precision:      100,
		MetaData:       map[string]interface{}{"test": true},
		SkipQueue:      true, // Enable skip queue
	}

	// Queue the transaction
	queuedTxn, err := blnk.QueueTransaction(ctx, txn)
	require.NoError(t, err, "Failed to queue transaction")

	// Verify that the transaction was processed immediately
	require.Equal(t, StatusApplied, queuedTxn.Status, "Transaction should be APPLIED immediately when skip_queue is true")

	// Verify balances were updated immediately
	updatedDestinationOne, err := ds.GetBalanceByIDLite(destinationOne.BalanceID)
	require.NoError(t, err, "Failed to get updated destination balance")

	updatedDestinationTwo, err := ds.GetBalanceByIDLite(destinationTwo.BalanceID)
	require.NoError(t, err, "Failed to get updated destination balance")

	updatedSource, err := ds.GetBalanceByIDLite(source.BalanceID)
	require.NoError(t, err, "Failed to get updated source balance")

	// Calculate expected balance changes
	expectedCredit := big.NewInt(int64(250) * 100) // Amount * precision
	expectedDebit := big.NewInt(int64(-500) * 100)

	// Verify balance changes
	require.Equal(t, 0, updatedDestinationOne.Balance.Cmp(expectedCredit),
		"destination balance should be immediately reduced by transaction amount")

	require.Equal(t, 0, updatedDestinationTwo.Balance.Cmp(expectedCredit),
		"destination balance should be immediately reduced by transaction amount")

	require.Equal(t, 0, updatedSource.Balance.Cmp(expectedDebit),
		"Source balance should be immediately increased by transaction amount")

	// Verify no queued entry exists
	queuedEntryOne, err := ds.GetTransactionByRef(ctx, fmt.Sprintf("%s-1", txnRef))
	require.NoError(t, err, "Failed to get queued transaction entry")
	require.Equal(t, StatusApplied, queuedEntryOne.Status, "Should have an APPLIED transaction entry")

	queuedEntryTwo, err := ds.GetTransactionByRef(ctx, fmt.Sprintf("%s-2", txnRef))
	require.NoError(t, err, "Failed to get queued transaction entry")
	require.Equal(t, StatusApplied, queuedEntryTwo.Status, "Should have an APPLIED transaction entry")
}

func TestQueueTransactionStatus(t *testing.T) {
	// Setup basic configuration
	cnf := &config.Configuration{
		Redis: config.RedisConfig{
			Dns: "localhost:6379",
		},
		DataSource: config.DataSourceConfig{
			Dns: "postgres://postgres:password@localhost:5432/blnk?sslmode=disable",
		},
		Queue: config.QueueConfig{
			WebhookQueue:     "webhook_queue_test",
			IndexQueue:       "index_queue_test",
			TransactionQueue: "transaction_queue_test",
			NumberOfQueues:   1,
		},
		Server: config.ServerConfig{
			SecretKey: "test-secret",
		},
	}
	config.ConfigStore.Store(cnf)

	// Create datasource and blnk instance
	ds, err := database.NewDataSource(cnf)
	require.NoError(t, err)
	blnk, err := NewBlnk(ds)
	require.NoError(t, err)

	// Create test balances
	sourceBalance := &model.Balance{
		Currency: "USD",
		LedgerID: "general_ledger_id",
	}
	destBalance := &model.Balance{
		Currency: "USD",
		LedgerID: "general_ledger_id",
	}

	source, err := ds.CreateBalance(*sourceBalance)
	require.NoError(t, err)
	dest, err := ds.CreateBalance(*destBalance)
	require.NoError(t, err)

	tests := []struct {
		name       string
		skipQueue  bool
		inflight   bool
		wantStatus string
	}{
		{
			name:       "Skip Queue True - Should be APPLIED",
			skipQueue:  true,
			inflight:   false,
			wantStatus: StatusApplied,
		},
		{
			name:       "Skip Queue False - Should be QUEUED",
			skipQueue:  false,
			inflight:   false,
			wantStatus: StatusQueued,
		},
		{
			name:       "Skip Queue True with Inflight - Should be INFLIGHT",
			skipQueue:  true,
			inflight:   true,
			wantStatus: StatusInflight,
		},
		{
			name:       "Skip Queue False with Inflight - Should be QUEUED",
			skipQueue:  false,
			inflight:   true,
			wantStatus: StatusQueued,
		},
	}

	for _, tt := range tests {
		t.Run(tt.name, func(t *testing.T) {
			txn := &model.Transaction{
				Reference:      "txn_" + model.GenerateUUIDWithSuffix("test"),
				Source:         source.BalanceID,
				Destination:    dest.BalanceID,
				Amount:         100,
				Currency:       "USD",
				Precision:      100,
				AllowOverdraft: true,
				SkipQueue:      tt.skipQueue,
				Inflight:       tt.inflight,
			}

			result, err := blnk.QueueTransaction(context.Background(), txn)
			require.NoError(t, err)
			assert.Equal(t, tt.wantStatus, result.Status,
				"Status mismatch for skipQueue=%v, inflight=%v", tt.skipQueue, tt.inflight)
		})
	}
}

func TestStandardTransactionRefund(t *testing.T) {
	// Skip in short mode
	if testing.Short() {
		t.Skip("Skipping queue flow test in short mode")
	}

	ctx := context.Background()
	cnf := &config.Configuration{
		Redis: config.RedisConfig{
			Dns: "localhost:6379",
		},
		DataSource: config.DataSourceConfig{
			Dns: "postgres://postgres:password@localhost:5432/blnk?sslmode=disable",
		},
		Queue: config.QueueConfig{
			WebhookQueue:     "webhook_queue_test",
			IndexQueue:       "index_queue_test",
			TransactionQueue: "transaction_queue_test",
			NumberOfQueues:   1,
		},
		Server: config.ServerConfig{
			SecretKey: "test-secret",
		},
		Transaction: config.TransactionConfig{
			BatchSize:        100,
			MaxQueueSize:     1000,
			LockDuration:     time.Second * 30,
			IndexQueuePrefix: "test_index",
		},
	}
	config.ConfigStore.Store(cnf)

	ds, err := database.NewDataSource(cnf)
	require.NoError(t, err, "Failed to create datasource")

	blnk, err := NewBlnk(ds)
	require.NoError(t, err, "Failed to create Blnk instance")

	txnRef := "txn_" + model.GenerateUUIDWithSuffix("test")

	// Create test balances with initial amounts
	sourceBalance := &model.Balance{
		Currency: "USD",
		LedgerID: "general_ledger_id",
	}
	destBalance := &model.Balance{
		Currency: "USD",
		LedgerID: "general_ledger_id",
	}

	source, err := ds.CreateBalance(*sourceBalance)
	require.NoError(t, err, "Failed to create source balance")

	dest, err := ds.CreateBalance(*destBalance)
	require.NoError(t, err, "Failed to create destination balance")

	// Create a standard transaction with skip_queue set to true
	originalAmount := 500.0
	txn := &model.Transaction{
		Reference:      txnRef,
		Source:         source.BalanceID,
		Destination:    dest.BalanceID,
		Amount:         originalAmount,
		Currency:       "USD",
		AllowOverdraft: true,
		Precision:      100,
		MetaData:       map[string]interface{}{"test": true},
		SkipQueue:      true, // Enable skip queue
		// No Inflight flag - this is a standard transaction
	}

	// Process the transaction
	originalTxn, err := blnk.QueueTransaction(ctx, txn)
	require.NoError(t, err, "Failed to process transaction")
	require.Equal(t, StatusApplied, originalTxn.Status, "Transaction should be APPLIED immediately with skip_queue enabled")

	// Verify balances after initial transaction
	updatedSource, err := ds.GetBalanceByIDLite(source.BalanceID)
	require.NoError(t, err, "Failed to get updated source balance")

	updatedDest, err := ds.GetBalanceByIDLite(dest.BalanceID)
	require.NoError(t, err, "Failed to get updated destination balance")

	// Calculate expected balance changes
	expectedDebit := big.NewInt(int64(-originalAmount) * 100) // Amount * precision
	expectedCredit := big.NewInt(int64(originalAmount) * 100)

	// Verify balance changes
	require.Equal(t, 0, updatedSource.Balance.Cmp(expectedDebit),
		"Source balance should be immediately reduced by transaction amount")
	require.Equal(t, 0, updatedDest.Balance.Cmp(expectedCredit),
		"Destination balance should be immediately increased by transaction amount")
	require.Equal(t, 0, updatedSource.InflightBalance.Cmp(big.NewInt(0)),
		"Source inflight balance should be zero for standard transaction")
	require.Equal(t, 0, updatedDest.InflightBalance.Cmp(big.NewInt(0)),
		"Destination inflight balance should be zero for standard transaction")

	// Get the actual transaction ID for the refund operation
	txnEntry, err := ds.GetTransactionByRef(ctx, txnRef)
	require.NoError(t, err, "Failed to get transaction entry")

	// Refund the transaction
	refundTxn, err := blnk.RefundTransaction(ctx, txnEntry.TransactionID, true)
	require.NoError(t, err, "Failed to refund transaction")
	require.Equal(t, StatusApplied, refundTxn.Status, "Refund transaction should be APPLIED immediately")
	require.Equal(t, txnEntry.TransactionID, refundTxn.ParentTransaction, "Refund should reference original transaction as parent")
	require.Equal(t, originalAmount, refundTxn.Amount, "Refund amount should match original amount")

	// Most importantly: verify that source and destination are swapped
	require.Equal(t, txnEntry.Destination, refundTxn.Source, "Refund source should be original destination")
	require.Equal(t, txnEntry.Source, refundTxn.Destination, "Refund destination should be original source")

	// Verify balances after refund
	sourceAfterRefund, err := ds.GetBalanceByIDLite(source.BalanceID)
	require.NoError(t, err, "Failed to get source balance after refund")

	destAfterRefund, err := ds.GetBalanceByIDLite(dest.BalanceID)
	require.NoError(t, err, "Failed to get destination balance after refund")

	// Balances should be back to where they started (zero)
	require.Equal(t, 0, sourceAfterRefund.Balance.Cmp(big.NewInt(0)),
		"Source balance should be reset to zero after refund")
	require.Equal(t, 0, destAfterRefund.Balance.Cmp(big.NewInt(0)),
		"Destination balance should be reset to zero after refund")
	require.Equal(t, 0, sourceAfterRefund.InflightBalance.Cmp(big.NewInt(0)),
		"Source inflight balance should remain zero after refund")
	require.Equal(t, 0, destAfterRefund.InflightBalance.Cmp(big.NewInt(0)),
		"Destination inflight balance should remain zero after refund")

	// Verify refund transaction in transaction history
	transactions, err := ds.GetTransactionsByParent(ctx, txnEntry.TransactionID, 10, 0)
	require.NoError(t, err, "Failed to get transaction history")
	require.Equal(t, 1, len(transactions), "Should have exactly one child transaction (the refund)")
	require.Equal(t, StatusApplied, transactions[0].Status, "Refund transaction should have APPLIED status")
	require.Equal(t, originalAmount, transactions[0].Amount, "Refund amount should match original")

	// Attempt to refund the refund transaction (should succeed)
	refundOfRefundTxn, err := blnk.RefundTransaction(ctx, refundTxn.TransactionID, true)
	require.NoError(t, err, "Failed to refund the refund transaction")
	require.Equal(t, StatusApplied, refundOfRefundTxn.Status, "Refund of refund should be APPLIED")
	require.Equal(t, refundTxn.TransactionID, refundOfRefundTxn.ParentTransaction, "Refund of refund should reference refund as parent")

	// Verify balances after refund of refund (should be back to the state after original transaction)
	sourceAfterRefundOfRefund, err := ds.GetBalanceByIDLite(source.BalanceID)
	require.NoError(t, err, "Failed to get source balance after refund of refund")

	destAfterRefundOfRefund, err := ds.GetBalanceByIDLite(dest.BalanceID)
	require.NoError(t, err, "Failed to get destination balance after refund of refund")

	// Verify the balances are back to the post-original transaction state
	require.Equal(t, 0, sourceAfterRefundOfRefund.Balance.Cmp(expectedDebit),
		"Source balance should be back to original debit after refund of refund")
	require.Equal(t, 0, destAfterRefundOfRefund.Balance.Cmp(expectedCredit),
		"Destination balance should be back to original credit after refund of refund")
}

func TestCategorizeRejectionReason(t *testing.T) {
	tests := []struct {
		name     string
		reason   string
		expected string
	}{
		// Actual error strings from the codebase:
		{
			name:     "insufficient funds from model.go",
			reason:   "insufficient funds in source balance",
			expected: "insufficient_funds",
		},
		{
			name:     "overdraft limit from model.go",
			reason:   "transaction exceeds overdraft limit",
			expected: "overdraft_limit",
		},
		{
			name:     "lock already held",
			reason:   "failed to acquire lock: resource already held",
			expected: "lock_contention",
		},
		{
			name:     "failed to acquire lock",
			reason:   "failed to acquire lock",
			expected: "lock_contention",
		},
		{
			name:     "failed to acquire batch lock",
			reason:   "failed to acquire batch lock for balance xyz",
			expected: "lock_contention",
		},
		{
			name:     "max queued recovery attempts from queue_recovery.go",
			reason:   "exceeded max queued recovery attempts",
			expected: "max_retries",
		},
		{
			name:     "unknown error maps to other",
			reason:   "some unexpected database error",
			expected: "other",
		},
		{
			name:     "case insensitive matching",
			reason:   "INSUFFICIENT FUNDS in source balance",
			expected: "insufficient_funds",
		},
	}

	for _, tt := range tests {
		t.Run(tt.name, func(t *testing.T) {
			result := categorizeRejectionReason(tt.reason)
			assert.Equal(t, tt.expected, result)
		})
	}
}

func TestRejectTransaction(t *testing.T) {
	// Skip in short mode
	if testing.Short() {
		t.Skip("Skipping transaction rejection test in short mode")
	}

	ctx := context.Background()
	cnf := &config.Configuration{
		Redis: config.RedisConfig{
			Dns: "localhost:6379",
		},
		DataSource: config.DataSourceConfig{
			Dns: "postgres://postgres:password@localhost:5432/blnk?sslmode=disable",
		},
		Queue: config.QueueConfig{
			WebhookQueue:     "webhook_queue_test",
			IndexQueue:       "index_queue_test",
			TransactionQueue: "transaction_queue_test",
			NumberOfQueues:   1,
		},
		Server: config.ServerConfig{
			SecretKey: "test-secret",
		},
		Transaction: config.TransactionConfig{
			BatchSize:        100,
			MaxQueueSize:     1000,
			LockDuration:     time.Second * 30,
			IndexQueuePrefix: "test_index",
		},
	}
	config.ConfigStore.Store(cnf)

	ds, err := database.NewDataSource(cnf)
	require.NoError(t, err, "Failed to create datasource")

	blnk, err := NewBlnk(ds)
	require.NoError(t, err, "Failed to create Blnk instance")

	// Create a unique transaction reference
	txnRef := "txn_" + model.GenerateUUIDWithSuffix("reject_test")

	// Create test balances
	sourceBalance := &model.Balance{
		Currency: "USD",
		LedgerID: "general_ledger_id",
	}
	destBalance := &model.Balance{
		Currency: "USD",
		LedgerID: "general_ledger_id",
	}

	source, err := ds.CreateBalance(*sourceBalance)
	require.NoError(t, err, "Failed to create source balance")

	dest, err := ds.CreateBalance(*destBalance)
	require.NoError(t, err, "Failed to create destination balance")

	// Create transaction to reject
	txn := &model.Transaction{
		TransactionID:  model.GenerateUUIDWithSuffix("txn"), // Ensure unique ID
		Reference:      txnRef,
		Source:         source.BalanceID,
		Destination:    dest.BalanceID,
		Amount:         500,
		AmountString:   "5.00",
		PreciseAmount:  big.NewInt(50000),
		Currency:       "USD",
		AllowOverdraft: true,
		Precision:      100,
		MetaData:       map[string]interface{}{"test": true},
		Status:         StatusQueued, // Initialize as queued
	}

	// Set a rejection reason
	rejectionReason := "Insufficient funds"

	// Reject the transaction
	rejectedTxn, err := blnk.RejectTransaction(ctx, txn, rejectionReason)
	require.NoError(t, err, "Failed to reject transaction")

	// Verify rejection details
	require.Equal(t, StatusRejected, rejectedTxn.Status, "Transaction status should be REJECTED")
	require.Contains(t, rejectedTxn.MetaData, "blnk_rejection_reason", "Metadata should contain rejection reason")
	require.Equal(t, rejectionReason, rejectedTxn.MetaData["blnk_rejection_reason"], "Rejection reason should match")

	// Verify transaction was persisted
	persistedTxn, err := ds.GetTransactionByRef(ctx, txnRef)
	require.NoError(t, err, "Failed to get persisted transaction")
	require.Equal(t, StatusRejected, persistedTxn.Status, "Persisted transaction should have REJECTED status")
	require.Contains(t, persistedTxn.MetaData, "blnk_rejection_reason", "Persisted metadata should contain rejection reason")
	require.Equal(t, rejectionReason, persistedTxn.MetaData["blnk_rejection_reason"], "Persisted rejection reason should match")

	// Verify balances should be unaffected by rejected transaction
	updatedSource, err := ds.GetBalanceByIDLite(source.BalanceID)
	require.NoError(t, err, "Failed to get updated source balance")

	updatedDest, err := ds.GetBalanceByIDLite(dest.BalanceID)
	require.NoError(t, err, "Failed to get updated destination balance")

	// Balances should be zero
	require.Equal(t, 0, updatedSource.Balance.Cmp(big.NewInt(0)),
		"Source balance should be zero for rejected transaction")
	require.Equal(t, 0, updatedDest.Balance.Cmp(big.NewInt(0)),
		"Destination balance should be zero for rejected transaction")
	require.Equal(t, 0, updatedSource.InflightBalance.Cmp(big.NewInt(0)),
		"Source inflight balance should be zero for rejected transaction")
	require.Equal(t, 0, updatedDest.InflightBalance.Cmp(big.NewInt(0)),
		"Destination inflight balance should be zero for rejected transaction")

	// Attempt to refund the rejected transaction - should fail
	_, err = blnk.RefundTransaction(ctx, rejectedTxn.TransactionID, true)
	require.Error(t, err, "Refunding a rejected transaction should fail")
	// Check for the specific error substring, ignoring the transaction ID
	require.Contains(t, err.Error(), "is not in a state that can be refunded (status: REJECTED)",
		"Error message should indicate transaction cannot be refunded due to its state")
}

func TestCommitWorkerFullFlow(t *testing.T) {
	// Skip in short mode
	if testing.Short() {
		t.Skip("Skipping commit worker test in short mode")
	}

	ctx := context.Background()
	cnf := &config.Configuration{
		Redis: config.RedisConfig{
			Dns: "localhost:6379",
		},
		DataSource: config.DataSourceConfig{
			Dns: "postgres://postgres:password@localhost:5432/blnk?sslmode=disable",
		},
		Queue: config.QueueConfig{
			WebhookQueue:     "webhook_queue_test",
			IndexQueue:       "index_queue_test",
			TransactionQueue: "transaction_queue_test",
			NumberOfQueues:   1,
		},
		Server: config.ServerConfig{
			SecretKey: "test-secret",
		},
		Transaction: config.TransactionConfig{
			BatchSize:        100,
			MaxQueueSize:     1000,
			LockDuration:     time.Second * 30,
			IndexQueuePrefix: "test_index",
		},
	}
	config.ConfigStore.Store(cnf)

	ds, err := database.NewDataSource(cnf)
	require.NoError(t, err, "Failed to create datasource")

	blnk, err := NewBlnk(ds)
	require.NoError(t, err, "Failed to create Blnk instance")

	queueName := fmt.Sprintf("%s_%d", cnf.Queue.TransactionQueue, 1)
	cleanupWorker := startTestAsynqWorker(t, cnf, blnk, queueName)
	defer cleanupWorker()

	// Create test balances with initial amounts of zero
	sourceBalance := &model.Balance{
		Currency: "USD",
		LedgerID: "general_ledger_id",
	}
	destBalance := &model.Balance{
		Currency: "USD",
		LedgerID: "general_ledger_id",
	}

	source, err := ds.CreateBalance(*sourceBalance)
	require.NoError(t, err, "Failed to create source balance")

	dest, err := ds.CreateBalance(*destBalance)
	require.NoError(t, err, "Failed to create destination balance")

	// Step 1: Create an inflight transaction with skip_queue set to true
	txnRef := "txn_" + model.GenerateUUIDWithSuffix("commit_flow_test")
	txn := &model.Transaction{
		Reference:      txnRef,
		Source:         source.BalanceID,
		Destination:    dest.BalanceID,
		Amount:         500,
		Currency:       "USD",
		AllowOverdraft: true,
		Precision:      100,
		MetaData:       map[string]interface{}{"test": true},
		Inflight:       true, // Make this an inflight transaction
		SkipQueue:      true, // Process immediately
	}

	// Step 2: Queue/process the transaction (which will apply it immediately due to skip_queue)
	queuedTxn, err := blnk.QueueTransaction(ctx, txn)
	require.NoError(t, err, "Failed to queue transaction")
	require.Equal(t, StatusInflight, queuedTxn.Status, "Transaction should be INFLIGHT immediately when skip_queue is true")

	// Step 3: Verify balances were updated immediately with inflight amounts
	updatedSource, err := ds.GetBalanceByIDLite(source.BalanceID)
	require.NoError(t, err, "Failed to get updated source balance")

	updatedDest, err := ds.GetBalanceByIDLite(dest.BalanceID)
	require.NoError(t, err, "Failed to get updated destination balance")

	// Verify inflight balance changes
	expectedInflightDebit := "-50000" // -500 * 100
	expectedInflightCredit := "50000" // 500 * 100

	require.Equal(t, expectedInflightDebit, updatedSource.InflightBalance.String(),
		"Source inflight balance should show debit immediately")
	require.Equal(t, expectedInflightCredit, updatedDest.InflightBalance.String(),
		"Destination inflight balance should show credit immediately")
	require.Equal(t, "0", updatedSource.Balance.String(),
		"Source main balance should still be zero")
	require.Equal(t, "0", updatedDest.Balance.String(),
		"Destination main balance should still be zero")

	// Step 4: Get the transaction by reference to get its ID
	inflightEntry, err := ds.GetTransactionByRef(ctx, txnRef)
	require.NoError(t, err, "Failed to get transaction by reference")
	require.Equal(t, StatusInflight, inflightEntry.Status, "Transaction should have INFLIGHT status")

	// Step 5: Setup commit worker
	var wg sync.WaitGroup
	wg.Add(1)

	jobs := make(chan *model.Transaction, 1)
	results := make(chan BatchJobResult, 1)

	// Get the transaction object by ID for the worker
	txnObj, err := ds.GetTransaction(ctx, inflightEntry.TransactionID)
	require.NoError(t, err, "Failed to get transaction by ID")

	// Push to the jobs channel
	jobs <- txnObj
	close(jobs)

	// Step 6: Run the commit worker
	go blnk.CommitWorker(ctx, jobs, results, &wg, big.NewInt(0)) // 0 = commit full amount

	// Wait for the worker to finish
	wg.Wait()
	close(results)

	// Collect the result
	var result BatchJobResult
	select {
	case result = <-results:
		// Got a result
	default:
		t.Fatal("No results received from commit worker")
	}

	// Step 7: Verify the commit worker result
	require.NoError(t, result.Error, "Commit worker should not return an error")
	require.NotNil(t, result.Txn, "Commit worker should return a transaction")
	require.Equal(t, StatusApplied, result.Txn.Status, "Committed transaction should have APPLIED status")
	require.Equal(t, inflightEntry.TransactionID, result.Txn.ParentTransaction,
		"Committed transaction should reference the original as parent")

	polledTxn, err := pollForTransactionStatus(ctx, ds, result.Txn.Reference, "APPLIED", 1*time.Second, 10*time.Second)
	require.NoError(t, err)
	require.Equal(t, StatusApplied, polledTxn.Status, "Committed transaction should have APPLIED status")

	// Step 8: Verify final balances
	finalSource, err := ds.GetBalanceByIDLite(source.BalanceID)
	require.NoError(t, err, "Failed to get final source balance")

	finalDest, err := ds.GetBalanceByIDLite(dest.BalanceID)
	require.NoError(t, err, "Failed to get final destination balance")

	// Verify balance changes - amounts should move from inflight to main balances
	expectedDebit := "-50000" // -500 * 100
	expectedCredit := "50000" // 500 * 100

	require.Equal(t, expectedDebit, finalSource.Balance.String(),
		"Source balance should now reflect the committed amount")
	require.Equal(t, expectedCredit, finalDest.Balance.String(),
		"Destination balance should now reflect the committed amount")
	require.Equal(t, "0", finalSource.InflightBalance.String(),
		"Source inflight balance should be zero after commit")
	require.Equal(t, "0", finalDest.InflightBalance.String(),
		"Destination inflight balance should be zero after commit")
}

func TestVoidWorkerFullFlow(t *testing.T) {
	// Skip in short mode
	if testing.Short() {
		t.Skip("Skipping void worker test in short mode")
	}

	ctx := context.Background()
	cnf := &config.Configuration{
		Redis: config.RedisConfig{
			Dns: "localhost:6379",
		},
		DataSource: config.DataSourceConfig{
			Dns: "postgres://postgres:password@localhost:5432/blnk?sslmode=disable",
		},
		Queue: config.QueueConfig{
			WebhookQueue:     "webhook_queue_test",
			IndexQueue:       "index_queue_test",
			TransactionQueue: "transaction_queue_test",
			NumberOfQueues:   1,
		},
		Server: config.ServerConfig{
			SecretKey: "test-secret",
		},
		Transaction: config.TransactionConfig{
			BatchSize:        100,
			MaxQueueSize:     1000,
			LockDuration:     time.Second * 30,
			IndexQueuePrefix: "test_index",
		},
	}
	config.ConfigStore.Store(cnf)

	ds, err := database.NewDataSource(cnf)
	require.NoError(t, err, "Failed to create datasource")

	blnk, err := NewBlnk(ds)
	require.NoError(t, err, "Failed to create Blnk instance")

	// Create test balances with initial amounts of zero
	sourceBalance := &model.Balance{
		Currency: "USD",
		LedgerID: "general_ledger_id",
	}
	destBalance := &model.Balance{
		Currency: "USD",
		LedgerID: "general_ledger_id",
	}

	source, err := ds.CreateBalance(*sourceBalance)
	require.NoError(t, err, "Failed to create source balance")

	dest, err := ds.CreateBalance(*destBalance)
	require.NoError(t, err, "Failed to create destination balance")

	// Step 1: Create an inflight transaction with skip_queue set to true
	txnRef := "txn_" + model.GenerateUUIDWithSuffix("void_flow_test")
	txn := &model.Transaction{
		Reference:      txnRef,
		Source:         source.BalanceID,
		Destination:    dest.BalanceID,
		Amount:         500,
		Currency:       "USD",
		AllowOverdraft: true,
		Precision:      100,
		MetaData:       map[string]interface{}{"test": true},
		Inflight:       true, // Make this an inflight transaction
		SkipQueue:      true, // Process immediately
	}

	// Step 2: Queue/process the transaction (which will apply it immediately due to skip_queue)
	queuedTxn, err := blnk.QueueTransaction(ctx, txn)
	require.NoError(t, err, "Failed to queue transaction")
	require.Equal(t, StatusInflight, queuedTxn.Status, "Transaction should be INFLIGHT immediately when skip_queue is true")

	// Step 3: Verify balances were updated immediately with inflight amounts
	updatedSource, err := ds.GetBalanceByIDLite(source.BalanceID)
	require.NoError(t, err, "Failed to get updated source balance")

	updatedDest, err := ds.GetBalanceByIDLite(dest.BalanceID)
	require.NoError(t, err, "Failed to get updated destination balance")

	// Verify inflight balance changes
	expectedInflightDebit := "-50000" // -500 * 100
	expectedInflightCredit := "50000" // 500 * 100

	require.Equal(t, expectedInflightDebit, updatedSource.InflightBalance.String(),
		"Source inflight balance should show debit immediately")
	require.Equal(t, expectedInflightCredit, updatedDest.InflightBalance.String(),
		"Destination inflight balance should show credit immediately")
	require.Equal(t, "0", updatedSource.Balance.String(),
		"Source main balance should still be zero")
	require.Equal(t, "0", updatedDest.Balance.String(),
		"Destination main balance should still be zero")

	// Step 4: Get the transaction by reference to get its ID
	inflightEntry, err := ds.GetTransactionByRef(ctx, txnRef)
	require.NoError(t, err, "Failed to get transaction by reference")
	require.Equal(t, StatusInflight, inflightEntry.Status, "Transaction should have INFLIGHT status")

	// Step 5: Setup void worker
	var wg sync.WaitGroup
	wg.Add(1)

	jobs := make(chan *model.Transaction, 1)
	results := make(chan BatchJobResult, 1)

	// Get the transaction object by ID for the worker
	txnObj, err := ds.GetTransaction(ctx, inflightEntry.TransactionID)
	require.NoError(t, err, "Failed to get transaction by ID")

	// Push to the jobs channel
	jobs <- txnObj
	close(jobs)

	// Step 6: Run the void worker
	go blnk.VoidWorker(ctx, jobs, results, &wg, big.NewInt(0)) // Amount parameter is not used in VoidWorker

	// Wait for the worker to finish
	wg.Wait()
	close(results)

	// Collect the result
	var result BatchJobResult
	select {
	case result = <-results:
		// Got a result
	default:
		t.Fatal("No results received from void worker")
	}

	// Step 7: Verify the void worker result
	require.NoError(t, result.Error, "Void worker should not return an error")
	require.NotNil(t, result.Txn, "Void worker should return a transaction")
	require.Equal(t, StatusVoid, result.Txn.Status, "Voided transaction should have VOID status")
	require.Equal(t, inflightEntry.TransactionID, result.Txn.ParentTransaction,
		"Voided transaction should reference the original as parent")

	// Step 8: Verify final balances
	finalSource, err := ds.GetBalanceByIDLite(source.BalanceID)
	require.NoError(t, err, "Failed to get final source balance")

	finalDest, err := ds.GetBalanceByIDLite(dest.BalanceID)
	require.NoError(t, err, "Failed to get final destination balance")

	// Verify balance changes - inflight balances should be zeroed and main balances should remain unchanged
	require.Equal(t, "0", finalSource.Balance.String(),
		"Source balance should remain unchanged at zero")
	require.Equal(t, "0", finalDest.Balance.String(),
		"Destination balance should remain unchanged at zero")
	require.Equal(t, "0", finalSource.InflightBalance.String(),
		"Source inflight balance should be zero after void")
	require.Equal(t, "0", finalDest.InflightBalance.String(),
		"Destination inflight balance should be zero after void")

	// Step 9: Verify that attempting to void again fails
	var wg2 sync.WaitGroup
	wg2.Add(1)

	jobs2 := make(chan *model.Transaction, 1)
	results2 := make(chan BatchJobResult, 1)

	jobs2 <- txnObj
	close(jobs2)

	go blnk.VoidWorker(ctx, jobs2, results2, &wg2, big.NewInt(0))
	wg2.Wait()
	close(results2)

	var result2 BatchJobResult
	select {
	case result2 = <-results2:
		// Got a result
	default:
		t.Fatal("No results received from second void worker")
	}

	require.Error(t, result2.Error, "Voiding an already voided transaction should fail")
	require.Contains(t, result2.Error.Error(), "transaction has already been voided",
		"Error should indicate transaction is already voided")
}

func TestRefundWorkerFullFlow(t *testing.T) {
	// Skip in short mode
	if testing.Short() {
		t.Skip("Skipping refund worker test in short mode")
	}

	ctx := context.Background()
	cnf := &config.Configuration{
		Redis: config.RedisConfig{
			Dns: "localhost:6379",
		},
		DataSource: config.DataSourceConfig{
			Dns: "postgres://postgres:password@localhost:5432/blnk?sslmode=disable",
		},
		Queue: config.QueueConfig{
			WebhookQueue:     "webhook_queue_test",
			IndexQueue:       "index_queue_test",
			TransactionQueue: "transaction_queue_test",
			NumberOfQueues:   1,
		},
		Server: config.ServerConfig{
			SecretKey: "test-secret",
		},
		Transaction: config.TransactionConfig{
			BatchSize:        100,
			MaxQueueSize:     1000,
			LockDuration:     time.Second * 30,
			IndexQueuePrefix: "test_index",
		},
	}
	config.ConfigStore.Store(cnf)

	ds, err := database.NewDataSource(cnf)
	require.NoError(t, err, "Failed to create datasource")

	blnk, err := NewBlnk(ds)
	require.NoError(t, err, "Failed to create Blnk instance")

	// Create test balances with initial amounts of zero
	sourceBalance := &model.Balance{
		Currency: "USD",
		LedgerID: "general_ledger_id",
	}
	destBalance := &model.Balance{
		Currency: "USD",
		LedgerID: "general_ledger_id",
	}

	source, err := ds.CreateBalance(*sourceBalance)
	require.NoError(t, err, "Failed to create source balance")

	dest, err := ds.CreateBalance(*destBalance)
	require.NoError(t, err, "Failed to create destination balance")

	// Step 1: Create a standard (non-inflight) transaction with skip_queue set to true
	txnRef := "txn_" + model.GenerateUUIDWithSuffix("refund_flow_test")
	txn := &model.Transaction{
		Reference:      txnRef,
		Source:         source.BalanceID,
		Destination:    dest.BalanceID,
		Amount:         500,
		Currency:       "USD",
		AllowOverdraft: true,
		Precision:      100,
		MetaData:       map[string]interface{}{"test": true},
		SkipQueue:      true, // Process immediately
		// No Inflight flag - this is a standard transaction that will be immediately applied
	}

	// Step 2: Queue/process the transaction (which will apply it immediately due to skip_queue)
	queuedTxn, err := blnk.QueueTransaction(ctx, txn)
	require.NoError(t, err, "Failed to queue transaction")
	require.Equal(t, StatusApplied, queuedTxn.Status, "Transaction should be APPLIED immediately when skip_queue is true")

	// Step 3: Verify balances were updated immediately with the transaction amounts
	updatedSource, err := ds.GetBalanceByIDLite(source.BalanceID)
	require.NoError(t, err, "Failed to get updated source balance")

	updatedDest, err := ds.GetBalanceByIDLite(dest.BalanceID)
	require.NoError(t, err, "Failed to get updated destination balance")

	// Verify main balance changes
	expectedDebit := "-50000" // -500 * 100
	expectedCredit := "50000" // 500 * 100

	require.Equal(t, expectedDebit, updatedSource.Balance.String(),
		"Source balance should show debit immediately")
	require.Equal(t, expectedCredit, updatedDest.Balance.String(),
		"Destination balance should show credit immediately")
	require.Equal(t, "0", updatedSource.InflightBalance.String(),
		"Source inflight balance should remain zero")
	require.Equal(t, "0", updatedDest.InflightBalance.String(),
		"Destination inflight balance should remain zero")

	// Step 4: Get the transaction by reference to get its ID
	appliedEntry, err := ds.GetTransactionByRef(ctx, txnRef)
	require.NoError(t, err, "Failed to get transaction by reference")
	require.Equal(t, StatusApplied, appliedEntry.Status, "Transaction should have APPLIED status")

	// Step 5: Setup refund worker
	var wg sync.WaitGroup
	wg.Add(1)

	jobs := make(chan *model.Transaction, 1)
	results := make(chan BatchJobResult, 1)

	// Get the transaction object by ID for the worker
	txnObj, err := ds.GetTransaction(ctx, appliedEntry.TransactionID)
	require.NoError(t, err, "Failed to get transaction by ID")

	// apply skip queue to txn to process refund immediately
	txnObj.SkipQueue = true

	// Push to the jobs channel
	jobs <- txnObj
	close(jobs)

	// Step 6: Run the refund worker
	go blnk.RefundWorker(ctx, jobs, results, &wg, big.NewInt(0))

	// Wait for the worker to finish
	wg.Wait()
	close(results)

	// Collect the result
	var result BatchJobResult
	select {
	case result = <-results:
		// Got a result
	default:
		t.Fatal("No results received from refund worker")
	}

	// Step 7: Verify the refund worker result
	require.NoError(t, result.Error, "Refund worker should not return an error")
	require.NotNil(t, result.Txn, "Refund worker should return a transaction")
	require.Equal(t, StatusApplied, result.Txn.Status, "Refund transaction should have APPLIED status")
	require.Equal(t, appliedEntry.TransactionID, result.Txn.ParentTransaction,
		"Refund transaction should reference the original as parent")
	require.Equal(t, appliedEntry.Source, result.Txn.Destination,
		"Refund transaction should swap source and destination")
	require.Equal(t, appliedEntry.Destination, result.Txn.Source,
		"Refund transaction should swap source and destination")

	// Step 8: Verify final balances
	finalSource, err := ds.GetBalanceByIDLite(source.BalanceID)
	require.NoError(t, err, "Failed to get final source balance")

	finalDest, err := ds.GetBalanceByIDLite(dest.BalanceID)
	require.NoError(t, err, "Failed to get final destination balance")

	// Verify balances are now back to zero (original amount - refund = 0)
	require.Equal(t, "0", finalSource.Balance.String(),
		"Source balance should be zero after refund")
	require.Equal(t, "0", finalDest.Balance.String(),
		"Destination balance should be zero after refund")
	require.Equal(t, "0", finalSource.InflightBalance.String(),
		"Source inflight balance should remain zero")
	require.Equal(t, "0", finalDest.InflightBalance.String(),
		"Destination inflight balance should remain zero")

	// Step 9: Test refunding a rejected transaction (which should fail)
	// Create a transaction to be rejected
	rejectedTxnRef := "txn_" + model.GenerateUUIDWithSuffix("rejected_test")
	rejectedTxn := &model.Transaction{
		TransactionID:  model.GenerateUUIDWithSuffix("txn"),
		Reference:      rejectedTxnRef,
		Source:         source.BalanceID,
		Destination:    dest.BalanceID,
		Amount:         200,
		AmountString:   "2.00",
		PreciseAmount:  big.NewInt(20000),
		Currency:       "USD",
		AllowOverdraft: true,
		Precision:      100,
		Status:         StatusQueued,
	}

	// Reject the transaction
	rejectedTxn, err = blnk.RejectTransaction(ctx, rejectedTxn, "Insufficient funds")
	require.NoError(t, err, "Failed to reject transaction")
	require.Equal(t, StatusRejected, rejectedTxn.Status, "Transaction should have REJECTED status")

	// Try to refund the rejected transaction
	var wg2 sync.WaitGroup
	wg2.Add(1)

	jobs2 := make(chan *model.Transaction, 1)
	results2 := make(chan BatchJobResult, 1)

	jobs2 <- rejectedTxn
	close(jobs2)

	go blnk.RefundWorker(ctx, jobs2, results2, &wg2, big.NewInt(0))
	wg2.Wait()
	close(results2)

	var result2 BatchJobResult
	select {
	case result2 = <-results2:
		// Got a result
	default:
		t.Fatal("No results received from second refund worker")
	}

	require.Error(t, result2.Error, "Refunding a rejected transaction via worker should fail")
	// Check for the specific error substring, ignoring the transaction ID
	require.Contains(t, result2.Error.Error(), "is not in a state that can be refunded (status: REJECTED)",
		"Error should indicate transaction cannot be refunded")
}

func TestRefundAlreadyRefundedTransaction(t *testing.T) {
	// Skip in short mode
	if testing.Short() {
		t.Skip("Skipping queue flow test in short mode")
	}

	ctx := context.Background()
	cnf := &config.Configuration{
		Redis: config.RedisConfig{
			Dns: "localhost:6379",
		},
		DataSource: config.DataSourceConfig{
			Dns: "postgres://postgres:password@localhost:5432/blnk?sslmode=disable",
		},
		Queue: config.QueueConfig{
			WebhookQueue:     "webhook_queue_test",
			IndexQueue:       "index_queue_test",
			TransactionQueue: "transaction_queue_test",
			NumberOfQueues:   1,
		},
		Server: config.ServerConfig{
			SecretKey: "test-secret",
		},
		Transaction: config.TransactionConfig{
			BatchSize:        100,
			MaxQueueSize:     1000,
			LockDuration:     time.Second * 30,
			IndexQueuePrefix: "test_index",
		},
	}
	config.ConfigStore.Store(cnf)

	ds, err := database.NewDataSource(cnf)
	require.NoError(t, err, "Failed to create datasource")

	blnk, err := NewBlnk(ds)
	require.NoError(t, err, "Failed to create Blnk instance")

	txnRef := "txn_" + model.GenerateUUIDWithSuffix("test_already_refunded")

	// Create test balances
	sourceBalance := &model.Balance{
		Currency: "USD",
		LedgerID: "general_ledger_id",
	}
	destBalance := &model.Balance{
		Currency: "USD",
		LedgerID: "general_ledger_id",
	}

	source, err := ds.CreateBalance(*sourceBalance)
	require.NoError(t, err, "Failed to create source balance")

	dest, err := ds.CreateBalance(*destBalance)
	require.NoError(t, err, "Failed to create destination balance")

	// Create a standard transaction
	originalAmount := 500.0
	txn := &model.Transaction{
		Reference:      txnRef,
		Source:         source.BalanceID,
		Destination:    dest.BalanceID,
		Amount:         originalAmount,
		Currency:       "USD",
		AllowOverdraft: true,
		Precision:      100,
		MetaData:       map[string]interface{}{"test": true},
		SkipQueue:      true,
	}

	// Process the transaction
	originalTxn, err := blnk.QueueTransaction(ctx, txn)
	require.NoError(t, err, "Failed to process transaction")
	require.Equal(t, StatusApplied, originalTxn.Status, "Transaction should be APPLIED")

	// Verify balances after initial transaction
	updatedSource, err := ds.GetBalanceByIDLite(source.BalanceID)
	require.NoError(t, err, "Failed to get updated source balance")
	updatedDest, err := ds.GetBalanceByIDLite(dest.BalanceID)
	require.NoError(t, err, "Failed to get updated destination balance")
	expectedDebit := big.NewInt(int64(-originalAmount) * 100)
	expectedCredit := big.NewInt(int64(originalAmount) * 100)
	require.Equal(t, 0, updatedSource.Balance.Cmp(expectedDebit), "Source balance incorrect after initial transaction")
	require.Equal(t, 0, updatedDest.Balance.Cmp(expectedCredit), "Destination balance incorrect after initial transaction")

	// Get the actual transaction ID for the refund operation
	txnEntry, err := ds.GetTransactionByRef(ctx, txnRef)
	require.NoError(t, err, "Failed to get transaction entry")

	// Refund the transaction for the first time
	refundTxn, err := blnk.RefundTransaction(ctx, txnEntry.TransactionID, true)
	require.NoError(t, err, "Failed to refund transaction the first time")
	require.Equal(t, StatusApplied, refundTxn.Status, "First refund transaction should be APPLIED")

	// Verify balances are back to zero after the first refund
	sourceAfterFirstRefund, err := ds.GetBalanceByIDLite(source.BalanceID)
	require.NoError(t, err, "Failed to get source balance after first refund")
	destAfterFirstRefund, err := ds.GetBalanceByIDLite(dest.BalanceID)
	require.NoError(t, err, "Failed to get destination balance after first refund")
	require.Equal(t, 0, sourceAfterFirstRefund.Balance.Cmp(big.NewInt(0)), "Source balance should be zero after first refund")
	require.Equal(t, 0, destAfterFirstRefund.Balance.Cmp(big.NewInt(0)), "Destination balance should be zero after first refund")

	// Attempt to refund the original transaction again
	_, err = blnk.RefundTransaction(ctx, txnEntry.TransactionID, true)
	require.Error(t, err, "Refunding an already refunded transaction should fail")
	// Check for the specific error substring, ignoring the transaction ID
	require.Contains(t, err.Error(), "has already been refunded", "Error message should indicate the transaction was already refunded")

	// Verify balances remain unchanged after the failed second refund attempt
	sourceAfterSecondAttempt, err := ds.GetBalanceByIDLite(source.BalanceID)
	require.NoError(t, err, "Failed to get source balance after second refund attempt")
	destAfterSecondAttempt, err := ds.GetBalanceByIDLite(dest.BalanceID)
	require.NoError(t, err, "Failed to get destination balance after second refund attempt")
	require.Equal(t, 0, sourceAfterSecondAttempt.Balance.Cmp(big.NewInt(0)), "Source balance should still be zero after failed second refund")
	require.Equal(t, 0, destAfterSecondAttempt.Balance.Cmp(big.NewInt(0)), "Destination balance should still be zero after failed second refund")
}

func TestQueueTransactionFlowAsyncPolled(t *testing.T) {
	// Skip in short mode
	if testing.Short() {
		t.Skip("Skipping queue flow test in short mode")
	}

	ctx := context.Background()
	cnf := &config.Configuration{
		Redis: config.RedisConfig{
			Dns: "localhost:6379", // Ensure Redis is running for queue tests
		},
		DataSource: config.DataSourceConfig{
			Dns: "postgres://postgres:password@localhost:5432/blnk?sslmode=disable", // Ensure Postgres is running
		},
		Queue: config.QueueConfig{ // Ensure queue names are unique if tests run in parallel
			WebhookQueue:        "webhook_queue_test_async_polled",
			IndexQueue:          "index_queue_test_async_polled",
			TransactionQueue:    "transaction_queue_test_async_polled", // Base name for transaction queue
			NumberOfQueues:      1,                                     // Test with one queue for simplicity
			InflightExpiryQueue: "inflight_expiry_queue_test_async_polled",
		},
		Server: config.ServerConfig{
			SecretKey: "test-secret-async-polled",
		},
		Transaction: config.TransactionConfig{
			BatchSize:        100,
			MaxQueueSize:     1000,
			LockDuration:     time.Second * 30,
			IndexQueuePrefix: "test_index_async_polled",
		},
	}
	config.ConfigStore.Store(cnf)

	ds, err := database.NewDataSource(cnf)
	require.NoError(t, err, "Failed to create datasource")

	blnkInstance, err := NewBlnk(ds)
	require.NoError(t, err, "Failed to create Blnk instance")

	// Construct the specific transaction queue name
	transactionQueueName := fmt.Sprintf("%s_%d", cnf.Queue.TransactionQueue, 1)

	// Start the test Asynq worker
	cleanupWorker := startTestAsynqWorker(t, cnf, blnkInstance, transactionQueueName)
	defer cleanupWorker()

	txnRef := "txn_" + model.GenerateUUIDWithSuffix("test_async_polled")

	sourceBalance := &model.Balance{
		Currency: "USD",
		LedgerID: "general_ledger_id",
	}
	destBalance := &model.Balance{
		Currency: "USD",
		LedgerID: "general_ledger_id",
	}

	source, err := ds.CreateBalance(*sourceBalance)
	require.NoError(t, err, "Failed to create source balance")
	dest, err := ds.CreateBalance(*destBalance)
	require.NoError(t, err, "Failed to create destination balance")

	txn := &model.Transaction{
		Reference:      txnRef,
		Source:         source.BalanceID,
		Destination:    dest.BalanceID,
		Amount:         500,
		Currency:       "USD",
		AllowOverdraft: true,
		Precision:      100,
		MetaData:       map[string]interface{}{"test": true},
		SkipQueue:      false, // Ensure transaction is queued
	}

	queuedTxn, err := blnkInstance.QueueTransaction(ctx, txn)
	require.NoError(t, err, "Failed to queue transaction")
	require.Equal(t, StatusQueued, queuedTxn.Status, "Transaction should be QUEUED")
	t.Logf("Transaction %s (Ref: %s) queued with ID %s", queuedTxn.Reference, queuedTxn.TransactionID, queuedTxn.TransactionID)

	// The worker will pick up 'queuedTxn'.
	// RecordTransaction, when processing a queued transaction, will:
	// 1. Set queuedTxn.ID as ParentTransaction for the new record.
	// 2. Generate a new TransactionID for the processed record.
	// 3. Modify the Reference to be originalRef + "_q".
	// So we poll for this modified reference.
	processedTxnRef := fmt.Sprintf("%s_%s", txnRef, "q")
	pollInterval := 200 * time.Millisecond // Increased poll interval slightly
	timeoutDuration := 15 * time.Second    // Increased timeout slightly for real worker

	t.Logf("Polling for processed transaction ref: %s (Original Queued Txn ID: %s)", processedTxnRef, queuedTxn.TransactionID)

	appliedTxn, err := pollForTransactionStatus(ctx, ds, processedTxnRef, StatusApplied, pollInterval, timeoutDuration)
	// Check error from polling more carefully
	if err != nil {
		// If timeout, try to get the transaction directly to see its status for debugging
		currentTxn, getErr := ds.GetTransactionByRef(ctx, processedTxnRef)
		if getErr == nil {
			t.Logf("DEBUG: Polling timed out. Current status of txn ref %s is %s. Parent: %s", processedTxnRef, currentTxn.Status, currentTxn.ParentTransaction)
		} else {
			t.Logf("DEBUG: Polling timed out. Failed to get txn ref %s directly: %v", processedTxnRef, getErr)
		}
		// Also check original queued transaction status
		originalQueuedTxn, getOrigErr := ds.GetTransaction(ctx, queuedTxn.TransactionID)
		if getOrigErr == nil {
			t.Logf("DEBUG: Original queued transaction ID %s status is %s.", queuedTxn.TransactionID, originalQueuedTxn.Status)
		}
		require.NoError(t, err, fmt.Sprintf("Polling failed for applied transaction %s. Error: %v", processedTxnRef, err))
	}

	require.NotNil(t, appliedTxn, "Applied transaction should not be nil")
	require.Equal(t, StatusApplied, appliedTxn.Status, "Processed transaction status should be APPLIED")
	require.Equal(t, queuedTxn.TransactionID, appliedTxn.ParentTransaction, "Applied transaction's parent ID should match original queued transaction ID")

	updatedSource, err := ds.GetBalanceByIDLite(source.BalanceID)
	require.NoError(t, err, "Failed to get updated source balance")
	updatedDest, err := ds.GetBalanceByIDLite(dest.BalanceID)
	require.NoError(t, err, "Failed to get updated destination balance")

	expectedDebit := big.NewInt(int64(-500) * 100)
	expectedCredit := big.NewInt(int64(500) * 100)

	assert.Equal(t, 0, updatedSource.Balance.Cmp(expectedDebit), "Source balance incorrect")
	assert.Equal(t, 0, updatedDest.Balance.Cmp(expectedCredit), "Destination balance incorrect")
}

// Helper for Asynq logger to use t.Logf
type testLogger struct {
	t *testing.T
}

func newTestLogger(t *testing.T) *testLogger {
	return &testLogger{t: t}
}

func (tl *testLogger) Debug(args ...interface{}) {
	tl.t.Logf("ASYNQ_DEBUG: %s", fmt.Sprint(args...))
}

func (tl *testLogger) Info(args ...interface{}) {
	tl.t.Logf("ASYNQ_INFO: %s", fmt.Sprint(args...))
}

func (tl *testLogger) Warn(args ...interface{}) {
	tl.t.Logf("ASYNQ_WARN: %s", fmt.Sprint(args...))
}

func (tl *testLogger) Error(args ...interface{}) {
	tl.t.Logf("ASYNQ_ERROR: %s", fmt.Sprint(args...))
}

func (tl *testLogger) Fatal(args ...interface{}) {
	tl.t.Logf("ASYNQ_FATAL: %s", fmt.Sprint(args...)) // t.Fatalf would exit test
}

// startTestAsynqWorker sets up and starts an Asynq server for testing purposes.
// It takes the testing object, configuration, Blnk instance, and the specific transaction queue name.
// It returns a cleanup function that should be deferred by the caller to shut down the server.
func startTestAsynqWorker(t *testing.T, cnf *config.Configuration, blnkInstance *Blnk, transactionQueueName string) func() {
	redisOption, err := redis_db.ParseRedisURL(cnf.Redis.Dns, false)
	require.NoError(t, err, "Failed to parse Redis URL for Asynq")

	queues := make(map[string]int)
	queues[transactionQueueName] = 1 // Concurrency for the transaction queue

	srv := asynq.NewServer(
		asynq.RedisClientOpt{
			Addr:      redisOption.Addr,
			Password:  redisOption.Password,
			DB:        redisOption.DB,
			TLSConfig: redisOption.TLSConfig,
		},
		asynq.Config{
			Concurrency: 1, // Overall server concurrency
			Queues:      queues,
			Logger:      newTestLogger(t),
		},
	)

	mux := asynq.NewServeMux()

	// Define transaction processing handler
	processTransactionHandler := func(ctx context.Context, task *asynq.Task) error {
		var txn model.Transaction
		if err := json.Unmarshal(task.Payload(), &txn); err != nil {
			t.Logf("TEST_WORKER: Error unmarshalling transaction: %v", err)
			return fmt.Errorf("failed to unmarshal transaction: %w", err)
		}

		t.Logf("TEST_WORKER: Picked up transaction %s (Ref: %s) for processing.", txn.TransactionID, txn.Reference)
		processedTxn, err := blnkInstance.RecordTransaction(ctx, &txn) // Use blnkInstance from the outer scope
		if err != nil {
			t.Logf("TEST_WORKER: Error recording transaction %s (Ref: %s): %v", txn.TransactionID, txn.Reference, err)
			if strings.Contains(strings.ToLower(err.Error()), "insufficient funds") || strings.Contains(strings.ToLower(err.Error()), "transaction exceeds overdraft limit") {
				_, rejectErr := blnkInstance.RejectTransaction(ctx, &txn, err.Error())
				if rejectErr != nil {
					t.Logf("TEST_WORKER: Error rejecting transaction %s after processing error: %v", txn.TransactionID, rejectErr)
					return fmt.Errorf("processing error: %v, rejection error: %w", err, rejectErr)
				}
				t.Logf("TEST_WORKER: Rejected transaction %s (Ref: %s) due to: %v", txn.TransactionID, txn.Reference, err)
				return nil // Assuming rejection is a final state for this test handler.
			}
			return err // Allow Asynq to retry for other errors
		}
		t.Logf("TEST_WORKER: Successfully processed transaction %s (Ref: %s), new ID: %s, new Ref: %s, Status: %s", txn.TransactionID, txn.Reference, processedTxn.TransactionID, processedTxn.Reference, processedTxn.Status)
		return nil
	}

	mux.HandleFunc(transactionQueueName, processTransactionHandler)

	go func() {
		t.Logf("TEST_WORKER: Starting Asynq server, listening on queue: %s", transactionQueueName)
		if err := srv.Run(mux); err != nil {
			t.Errorf("TEST_WORKER: Asynq server Run() error: %v", err)
		}
	}()

	return func() {
		t.Log("TEST_WORKER: Shutting down Asynq server...")
		srv.Shutdown()
		t.Log("TEST_WORKER: Asynq server shut down.")
	}
}

func TestInflightTransactionFlowAsyncPolled(t *testing.T) {
	// Skip in short mode
	if testing.Short() {
		t.Skip("Skipping inflight async flow test in short mode")
	}

	ctx := context.Background()
	cnf := &config.Configuration{
		Redis: config.RedisConfig{
			Dns: "localhost:6379",
		},
		DataSource: config.DataSourceConfig{
			Dns: "postgres://postgres:password@localhost:5432/blnk?sslmode=disable",
		},
		Queue: config.QueueConfig{
			WebhookQueue:     "webhook_queue_test_inflight_async",
			IndexQueue:       "index_queue_test_inflight_async",
			TransactionQueue: "transaction_queue_test_inflight_async",
			NumberOfQueues:   1,
		},
		Server: config.ServerConfig{
			SecretKey: "test-secret-inflight-async",
		},
		Transaction: config.TransactionConfig{
			BatchSize:        100,
			MaxQueueSize:     1000,
			LockDuration:     time.Second * 30,
			IndexQueuePrefix: "test_index_inflight_async",
		},
	}
	config.ConfigStore.Store(cnf)

	// Construct the specific transaction queue name
	transactionQueueName := fmt.Sprintf("%s_%d", cnf.Queue.TransactionQueue, 1)

	ds, err := database.NewDataSource(cnf)
	require.NoError(t, err, "Failed to create datasource")

	blnkInstance, err := NewBlnk(ds)
	require.NoError(t, err, "Failed to create Blnk instance")

	// Start the test Asynq worker
	cleanupWorker := startTestAsynqWorker(t, cnf, blnkInstance, transactionQueueName)
	defer cleanupWorker()

	txnRef := "txn_" + model.GenerateUUIDWithSuffix("test_inflight_async")

	sourceBalance := &model.Balance{Currency: "USD", LedgerID: "general_ledger_id"}
	destBalance := &model.Balance{Currency: "USD", LedgerID: "general_ledger_id"}

	source, err := ds.CreateBalance(*sourceBalance)
	require.NoError(t, err)
	dest, err := ds.CreateBalance(*destBalance)
	require.NoError(t, err)

	txn := &model.Transaction{
		Reference:      txnRef,
		Source:         source.BalanceID,
		Destination:    dest.BalanceID,
		Amount:         500,
		Inflight:       true, // Key: This is an inflight transaction
		Currency:       "USD",
		AllowOverdraft: true,
		Precision:      100,
		MetaData:       map[string]interface{}{"test": true},
		SkipQueue:      false, // Key: It will be queued first
	}

	queuedTxn, err := blnkInstance.QueueTransaction(ctx, txn)
	require.NoError(t, err, "Failed to queue inflight transaction")
	require.Equal(t, StatusQueued, queuedTxn.Status, "Inflight transaction should initially be QUEUED")

	// Simulate worker picking up and processing the queued inflight transaction
	// The worker calls RecordTransaction, which should create the INFLIGHT record.
	queueCopy := createQueueCopy(queuedTxn, queuedTxn.Reference) // Preserves Inflight=true

	// Poll for the processed transaction (which should now be INFLIGHT)
	// The reference for the processed transaction by RecordTransaction will be queueCopy.Reference
	processedInflightTxnRef := queueCopy.Reference // This is typically originalRef + "_q"
	pollInterval := 200 * time.Millisecond
	timeoutDuration := 10 * time.Second

	fmt.Printf("Polling for INFLIGHT transaction ref: %s (Original Queued ID: %s)\n", processedInflightTxnRef, queuedTxn.TransactionID)

	inflightTxn, err := pollForTransactionStatus(ctx, ds, processedInflightTxnRef, StatusInflight, pollInterval, timeoutDuration)
	require.NoError(t, err, fmt.Sprintf("Polling failed for INFLIGHT transaction %s. Error: %v", processedInflightTxnRef, err))
	require.NotNil(t, inflightTxn, "INFLIGHT transaction should not be nil after polling")
	require.Equal(t, StatusInflight, inflightTxn.Status, "Processed transaction status should be INFLIGHT")
	// The inflightTxn created by RecordTransaction is a child of the *queueCopy* if queueCopy was saved first.
	// However, our createQueueCopy doesn't save itself, RecordTransaction processes its details.
	// The key is that an INFLIGHT transaction now exists with reference `processedInflightTxnRef`.
	// Let's verify its ParentTransaction refers to the *original* queued transaction.
	require.Equal(t, queuedTxn.TransactionID, inflightTxn.ParentTransaction, "INFLIGHT transaction's parent ID should match original queued transaction ID")

	// Verify inflight balances
	updatedSource, err := ds.GetBalanceByIDLite(source.BalanceID)
	require.NoError(t, err)
	updatedDest, err := ds.GetBalanceByIDLite(dest.BalanceID)
	require.NoError(t, err)

	expectedInflightDebit := big.NewInt(int64(-500) * 100)
	expectedInflightCredit := big.NewInt(int64(500) * 100)

	assert.Equal(t, 0, updatedSource.InflightBalance.Cmp(expectedInflightDebit), "Source inflight balance incorrect")
	assert.Equal(t, 0, updatedDest.InflightBalance.Cmp(expectedInflightCredit), "Destination inflight balance incorrect")
	assert.Equal(t, 0, updatedSource.Balance.Cmp(big.NewInt(0)), "Source main balance should be 0")
	assert.Equal(t, 0, updatedDest.Balance.Cmp(big.NewInt(0)), "Destination main balance should be 0")

	// --- Partial Commits ---
	originalAmountPrecise := inflightTxn.PreciseAmount // 50000
	partialAmount1Precise := big.NewInt(100 * 100)     // 10000
	partialAmount2Precise := big.NewInt(150 * 100)     // 15000
	// Remaining amount will be committed with big.NewInt(0) or calculated

	// First Partial Commit (100)
	t.Logf("Submitting first partial commit (amount: %s) for inflightTxn ID: %s, original queuedTxn ID: %s", partialAmount1Precise.String(), inflightTxn.TransactionID, queuedTxn.TransactionID)
	commitTxn1, err := blnkInstance.CommitInflightTransactionWithQueue(ctx, inflightTxn.TransactionID, partialAmount1Precise)
	require.NoError(t, err, "Failed to submit first partial commit")
	require.Equal(t, StatusApplied, commitTxn1.Status, "First partial commit submission should have status COMMIT")

	t.Logf("Polling for first APPLIED commit transaction (Ref: %s)", commitTxn1.Reference)
	appliedCommitTxn1, err := pollForTransactionStatus(ctx, ds, commitTxn1.Reference, StatusApplied, pollInterval, timeoutDuration)
	require.NoError(t, err, fmt.Sprintf("Polling failed for first applied commit (Ref: %s). Error: %v", commitTxn1.Reference, err))
	require.NotNil(t, appliedCommitTxn1, "First applied commit transaction should not be nil")
	require.Equal(t, StatusApplied, appliedCommitTxn1.Status, "First commit transaction status should be APPLIED")
	require.Equal(t, inflightTxn.TransactionID, appliedCommitTxn1.ParentTransaction, "First applied commit parent ID mismatch")

	// Verify balances after first partial commit
	sourceAfterPartial1, _ := ds.GetBalanceByIDLite(source.BalanceID)
	destAfterPartial1, _ := ds.GetBalanceByIDLite(dest.BalanceID)

	expectedSourceBalanceAfter1 := big.NewInt(0).Sub(big.NewInt(0), partialAmount1Precise)          // -10000
	expectedDestBalanceAfter1 := partialAmount1Precise                                              //  10000
	expectedSourceInflightAfter1 := big.NewInt(0).Add(expectedInflightDebit, partialAmount1Precise) // -50000 + 10000 = -40000
	expectedDestInflightAfter1 := big.NewInt(0).Sub(expectedInflightCredit, partialAmount1Precise)  //  50000 -   10000  =  40000

	assert.Equal(t, 0, sourceAfterPartial1.Balance.Cmp(expectedSourceBalanceAfter1), "Source balance incorrect after 1st partial commit")
	assert.Equal(t, 0, destAfterPartial1.Balance.Cmp(expectedDestBalanceAfter1), "Destination balance incorrect after 1st partial commit")
	assert.Equal(t, 0, sourceAfterPartial1.InflightBalance.Cmp(expectedSourceInflightAfter1), "Source inflight balance incorrect after 1st partial commit")
	assert.Equal(t, 0, destAfterPartial1.InflightBalance.Cmp(expectedDestInflightAfter1), "Destination inflight balance incorrect after 1st partial commit")
	t.Logf("Balances verified after first partial commit.")

	// Second Partial Commit (150)
	t.Logf("Submitting second partial commit (amount: %s) for inflightTxn ID: %s", partialAmount2Precise.String(), inflightTxn.TransactionID)
	commitTxn2, err := blnkInstance.CommitInflightTransactionWithQueue(ctx, inflightTxn.TransactionID, partialAmount2Precise)
	require.NoError(t, err, "Failed to submit second partial commit")
	require.Equal(t, StatusApplied, commitTxn2.Status, "Second partial commit submission should have status COMMIT")

	t.Logf("Polling for second APPLIED commit transaction (Ref: %s)", commitTxn2.Reference)
	appliedCommitTxn2, err := pollForTransactionStatus(ctx, ds, commitTxn2.Reference, StatusApplied, pollInterval, timeoutDuration)
	require.NoError(t, err, fmt.Sprintf("Polling failed for second applied commit (Ref: %s). Error: %v", commitTxn2.Reference, err))
	require.NotNil(t, appliedCommitTxn2, "Second applied commit transaction should not be nil")
	require.Equal(t, StatusApplied, appliedCommitTxn2.Status, "Second commit transaction status should be APPLIED")
	require.Equal(t, inflightTxn.TransactionID, appliedCommitTxn2.ParentTransaction, "Second applied commit parent ID mismatch")

	// Verify balances after second partial commit
	sourceAfterPartial2, _ := ds.GetBalanceByIDLite(source.BalanceID)
	destAfterPartial2, _ := ds.GetBalanceByIDLite(dest.BalanceID)

	totalCommittedAfter2 := big.NewInt(0).Add(partialAmount1Precise, partialAmount2Precise)        // 10000 + 15000 = 25000
	expectedSourceBalanceAfter2 := big.NewInt(0).Sub(big.NewInt(0), totalCommittedAfter2)          // -25000
	expectedDestBalanceAfter2 := totalCommittedAfter2                                              //  25000
	expectedSourceInflightAfter2 := big.NewInt(0).Add(expectedInflightDebit, totalCommittedAfter2) // -50000 + 25000 = -25000
	expectedDestInflightAfter2 := big.NewInt(0).Sub(expectedInflightCredit, totalCommittedAfter2)  //  50000 -   25000  =  25000

	assert.Equal(t, 0, sourceAfterPartial2.Balance.Cmp(expectedSourceBalanceAfter2), "Source balance incorrect after 2nd partial commit")
	assert.Equal(t, 0, destAfterPartial2.Balance.Cmp(expectedDestBalanceAfter2), "Destination balance incorrect after 2nd partial commit")
	assert.Equal(t, 0, sourceAfterPartial2.InflightBalance.Cmp(expectedSourceInflightAfter2), "Source inflight balance incorrect after 2nd partial commit")
	assert.Equal(t, 0, destAfterPartial2.InflightBalance.Cmp(expectedDestInflightAfter2), "Destination inflight balance incorrect after 2nd partial commit")
	t.Logf("Balances verified after second partial commit.")

	// Third Partial Commit (Remaining amount - 250)
	// Committing with 0 means commit remaining
	t.Logf("Submitting third partial commit (remaining amount) for inflightTxn ID: %s", inflightTxn.TransactionID)
	commitTxn3, err := blnkInstance.CommitInflightTransactionWithQueue(ctx, inflightTxn.TransactionID, big.NewInt(0))
	require.NoError(t, err, "Failed to submit third partial commit (remaining)")
	require.Equal(t, StatusApplied, commitTxn3.Status, "Third partial commit submission should have status COMMIT")

	t.Logf("Polling for third APPLIED commit transaction (Ref: %s)", commitTxn3.Reference)
	appliedCommitTxn3, err := pollForTransactionStatus(ctx, ds, commitTxn3.Reference, StatusApplied, pollInterval, timeoutDuration)
	require.NoError(t, err, fmt.Sprintf("Polling failed for third applied commit (Ref: %s). Error: %v", commitTxn3.Reference, err))
	require.NotNil(t, appliedCommitTxn3, "Third applied commit transaction should not be nil")
	require.Equal(t, StatusApplied, appliedCommitTxn3.Status, "Third commit transaction status should be APPLIED")
	require.Equal(t, inflightTxn.TransactionID, appliedCommitTxn3.ParentTransaction, "Third applied commit parent ID mismatch")

	// Verify final balances after all commits
	finalSource, _ := ds.GetBalanceByIDLite(source.BalanceID)
	finalDest, _ := ds.GetBalanceByIDLite(dest.BalanceID)

	// Balances should now reflect the full originalAmount, and inflight should be zero.
	// originalAmountPrecise = 50000
	expectedFinalSourceBalance := big.NewInt(0).Sub(big.NewInt(0), originalAmountPrecise) // -50000
	expectedFinalDestBalance := originalAmountPrecise                                     //  50000

	assert.Equal(t, 0, finalSource.Balance.Cmp(expectedFinalSourceBalance), "Source final balance incorrect after all commits")
	assert.Equal(t, 0, finalDest.Balance.Cmp(expectedFinalDestBalance), "Destination final balance incorrect after all commits")
	assert.Equal(t, 0, finalSource.InflightBalance.Cmp(big.NewInt(0)), "Source inflight balance should be 0 after all commits")
	assert.Equal(t, 0, finalDest.InflightBalance.Cmp(big.NewInt(0)), "Destination inflight balance should be 0 after all commits")
	t.Logf("Balances verified after third (final) partial commit.")

	// Verify the sum of amounts of appliedCommitTxn1, appliedCommitTxn2, appliedCommitTxn3 equals originalAmountPrecise
	totalCommittedPrecise := big.NewInt(0)
	totalCommittedPrecise.Add(totalCommittedPrecise, appliedCommitTxn1.PreciseAmount)
	totalCommittedPrecise.Add(totalCommittedPrecise, appliedCommitTxn2.PreciseAmount)
	totalCommittedPrecise.Add(totalCommittedPrecise, appliedCommitTxn3.PreciseAmount)
	require.Equal(t, 0, originalAmountPrecise.Cmp(totalCommittedPrecise), "Sum of partial commits does not equal original inflight amount")
	t.Logf("Sum of partial commit amounts verified.")
}

func TestMultipleSourcesTransactionFlowAsyncPolled(t *testing.T) {
	// Skip in short mode
	if testing.Short() {
		t.Skip("Skipping multi-source async flow test in short mode")
	}

	ctx := context.Background()
	cnf := &config.Configuration{
		Redis: config.RedisConfig{
			Dns: "localhost:6379",
		},
		DataSource: config.DataSourceConfig{
			Dns: "postgres://postgres:password@localhost:5432/blnk?sslmode=disable",
		},
		Queue: config.QueueConfig{
			WebhookQueue:     "webhook_queue_test_ms_async",
			IndexQueue:       "index_queue_test_ms_async",
			TransactionQueue: "transaction_queue_test_ms_async",
			NumberOfQueues:   1,
		},
		Server: config.ServerConfig{
			SecretKey: "test-secret-ms-async",
		},
		Transaction: config.TransactionConfig{
			BatchSize:        100,
			MaxQueueSize:     1000,
			LockDuration:     time.Second * 30,
			IndexQueuePrefix: "test_index_ms_async",
		},
	}
	config.ConfigStore.Store(cnf)

	ds, err := database.NewDataSource(cnf)
	require.NoError(t, err, "Failed to create datasource")

	blnkInstance, err := NewBlnk(ds)
	require.NoError(t, err, "Failed to create Blnk instance")

	// Construct the specific transaction queue name
	transactionQueueName := fmt.Sprintf("%s_%d", cnf.Queue.TransactionQueue, 1)

	// Start the test Asynq worker
	cleanupWorker := startTestAsynqWorker(t, cnf, blnkInstance, transactionQueueName)
	defer cleanupWorker()

	txnRef := "txn_" + model.GenerateUUIDWithSuffix("test_ms_async")

	sourceBalanceOne := &model.Balance{Currency: "USD", LedgerID: "general_ledger_id"}
	sourceBalanceTwo := &model.Balance{Currency: "USD", LedgerID: "general_ledger_id"}
	destBalance := &model.Balance{Currency: "USD", LedgerID: "general_ledger_id"}

	sourceOne, err := ds.CreateBalance(*sourceBalanceOne)
	require.NoError(t, err)
	sourceTwo, err := ds.CreateBalance(*sourceBalanceTwo)
	require.NoError(t, err)
	dest, err := ds.CreateBalance(*destBalance)
	require.NoError(t, err)

	txn := &model.Transaction{
		Reference: txnRef,
		Sources: []model.Distribution{
			{Identifier: sourceOne.BalanceID, Distribution: "50%"},
			{Identifier: sourceTwo.BalanceID, Distribution: "50%"},
		},
		Destination:    dest.BalanceID,
		Amount:         500,
		Currency:       "USD",
		AllowOverdraft: true,
		Precision:      100,
		MetaData:       map[string]interface{}{"test": true},
		SkipQueue:      false, // Key: Will be queued
	}

	queuedTxn, err := blnkInstance.QueueTransaction(ctx, txn)
	require.NoError(t, err, "Failed to queue multi-source transaction")
	require.Equal(t, StatusQueued, queuedTxn.Status, "Multi-source transaction should initially be QUEUED")

	// Poll for each child transaction to be APPLIED
	pollInterval := 200 * time.Millisecond
	timeoutDuration := 10 * time.Second

	childTxnRef1 := fmt.Sprintf("%s_1_q", queuedTxn.Reference) // Based on how RecordTransaction creates child refs
	childTxnRef2 := fmt.Sprintf("%s_2_q", queuedTxn.Reference)

	fmt.Printf("Polling for multi-source child transaction 1: %s\n", childTxnRef1)
	appliedChild1, err := pollForTransactionStatus(ctx, ds, childTxnRef1, StatusApplied, pollInterval, timeoutDuration)
	require.NoError(t, err, fmt.Sprintf("Polling failed for multi-source child transaction 1 (%s). Error: %v", childTxnRef1, err))
	require.NotNil(t, appliedChild1, "Applied child transaction 1 should not be nil")
	require.Equal(t, StatusApplied, appliedChild1.Status)

	fmt.Printf("Polling for multi-source child transaction 2: %s\n", childTxnRef2)
	appliedChild2, err := pollForTransactionStatus(ctx, ds, childTxnRef2, StatusApplied, pollInterval, timeoutDuration)
	require.NoError(t, err, fmt.Sprintf("Polling failed for multi-source child transaction 2 (%s). Error: %v", childTxnRef2, err))
	require.NotNil(t, appliedChild2, "Applied child transaction 2 should not be nil")
	require.Equal(t, StatusApplied, appliedChild2.Status)

	// Verify balances
	updatedSourceOne, _ := ds.GetBalanceByIDLite(sourceOne.BalanceID)
	updatedSourceTwo, _ := ds.GetBalanceByIDLite(sourceTwo.BalanceID)
	updatedDest, _ := ds.GetBalanceByIDLite(dest.BalanceID)

	expectedDebitEachSource := big.NewInt(int64(-250) * 100) // 50% of 500
	expectedCreditDest := big.NewInt(int64(500) * 100)

	assert.Equal(t, 0, updatedSourceOne.Balance.Cmp(expectedDebitEachSource), "Source one balance incorrect")
	assert.Equal(t, 0, updatedSourceTwo.Balance.Cmp(expectedDebitEachSource), "Source two balance incorrect")
	assert.Equal(t, 0, updatedDest.Balance.Cmp(expectedCreditDest), "Destination balance incorrect")
}

func TestMultipleDestinationsTransactionFlowAsyncPolled(t *testing.T) {
	// Skip in short mode
	if testing.Short() {
		t.Skip("Skipping multi-destination async flow test in short mode")
	}

	ctx := context.Background()
	cnf := &config.Configuration{
		Redis: config.RedisConfig{
			Dns: "localhost:6379",
		},
		DataSource: config.DataSourceConfig{
			Dns: "postgres://postgres:password@localhost:5432/blnk?sslmode=disable",
		},
		Queue: config.QueueConfig{
			WebhookQueue:     "webhook_queue_test_md_async",
			IndexQueue:       "index_queue_test_md_async",
			TransactionQueue: "transaction_queue_test_md_async",
			NumberOfQueues:   1,
		},
		Server: config.ServerConfig{
			SecretKey: "test-secret-md-async",
		},
		Transaction: config.TransactionConfig{
			BatchSize:        100,
			MaxQueueSize:     1000,
			LockDuration:     time.Second * 30,
			IndexQueuePrefix: "test_index_md_async",
		},
	}
	config.ConfigStore.Store(cnf)

	ds, err := database.NewDataSource(cnf)
	require.NoError(t, err, "Failed to create datasource")

	blnkInstance, err := NewBlnk(ds)
	require.NoError(t, err, "Failed to create Blnk instance")

	// Construct the specific transaction queue name
	transactionQueueName := fmt.Sprintf("%s_%d", cnf.Queue.TransactionQueue, 1)

	// Start the test Asynq worker
	cleanupWorker := startTestAsynqWorker(t, cnf, blnkInstance, transactionQueueName)
	defer cleanupWorker()

	txnRef := "txn_" + model.GenerateUUIDWithSuffix("test_md_async")

	sourceBalance := &model.Balance{Currency: "USD", LedgerID: "general_ledger_id"}
	destBalanceOne := &model.Balance{Currency: "USD", LedgerID: "general_ledger_id"}
	destBalanceTwo := &model.Balance{Currency: "USD", LedgerID: "general_ledger_id"}

	source, err := ds.CreateBalance(*sourceBalance)
	require.NoError(t, err)
	destOne, err := ds.CreateBalance(*destBalanceOne)
	require.NoError(t, err)
	destTwo, err := ds.CreateBalance(*destBalanceTwo)
	require.NoError(t, err)

	txn := &model.Transaction{
		Reference: txnRef,
		Source:    source.BalanceID,
		Destinations: []model.Distribution{
			{Identifier: destOne.BalanceID, Distribution: "50%"},
			{Identifier: destTwo.BalanceID, Distribution: "50%"},
		},
		Amount:         500,
		Currency:       "USD",
		AllowOverdraft: true,
		Precision:      100,
		MetaData:       map[string]interface{}{"test": true},
		SkipQueue:      false, // Key: Will be queued
	}

	queuedTxn, err := blnkInstance.QueueTransaction(ctx, txn)
	require.NoError(t, err, "Failed to queue multi-destination transaction")
	require.Equal(t, StatusQueued, queuedTxn.Status, "Multi-destination transaction should initially be QUEUED")

	// Poll for each child transaction to be APPLIED
	pollInterval := 200 * time.Millisecond
	timeoutDuration := 10 * time.Second

	childTxnRef1 := fmt.Sprintf("%s_1_q", queuedTxn.Reference)
	childTxnRef2 := fmt.Sprintf("%s_2_q", queuedTxn.Reference)

	fmt.Printf("Polling for multi-destination child transaction 1: %s\n", childTxnRef1)
	appliedChild1, err := pollForTransactionStatus(ctx, ds, childTxnRef1, StatusApplied, pollInterval, timeoutDuration)
	require.NoError(t, err, fmt.Sprintf("Polling failed for multi-destination child transaction 1 (%s). Error: %v", childTxnRef1, err))
	require.NotNil(t, appliedChild1, "Applied child transaction 1 should not be nil")
	require.Equal(t, StatusApplied, appliedChild1.Status)

	fmt.Printf("Polling for multi-destination child transaction 2: %s\n", childTxnRef2)
	appliedChild2, err := pollForTransactionStatus(ctx, ds, childTxnRef2, StatusApplied, pollInterval, timeoutDuration)
	require.NoError(t, err, fmt.Sprintf("Polling failed for multi-destination child transaction 2 (%s). Error: %v", childTxnRef2, err))
	require.NotNil(t, appliedChild2, "Applied child transaction 2 should not be nil")
	require.Equal(t, StatusApplied, appliedChild2.Status)

	// Verify balances
	updatedSource, _ := ds.GetBalanceByIDLite(source.BalanceID)
	updatedDestOne, _ := ds.GetBalanceByIDLite(destOne.BalanceID)
	updatedDestTwo, _ := ds.GetBalanceByIDLite(destTwo.BalanceID)

	expectedDebitSource := big.NewInt(int64(-500) * 100)
	expectedCreditEachDest := big.NewInt(int64(250) * 100) // 50% of 500

	assert.Equal(t, 0, updatedSource.Balance.Cmp(expectedDebitSource), "Source balance incorrect")
	assert.Equal(t, 0, updatedDestOne.Balance.Cmp(expectedCreditEachDest), "Destination one balance incorrect")
	assert.Equal(t, 0, updatedDestTwo.Balance.Cmp(expectedCreditEachDest), "Destination two balance incorrect")
}

func TestInflightTransactionWithOverdraftOnCommit(t *testing.T) {
	// Test scenario:
	// 1. Create a source balance with 500 USD
	// 2. Create an inflight transaction for 1000 USD (exceeds balance) with AllowOverdraft: false
	// 3. The transaction should be QUEUED initially, then REJECTED due to insufficient funds
	// 4. Attempt to commit the rejected transaction
	// 5. Verify that balances remain unchanged since the transaction was rejected

	// Skip in short mode
	if testing.Short() {
		t.Skip("Skipping inflight overdraft test in short mode")
	}

	ctx := context.Background()
	cnf := &config.Configuration{
		Redis: config.RedisConfig{
			Dns: "localhost:6379",
		},
		DataSource: config.DataSourceConfig{
			Dns: "postgres://postgres:password@localhost:5432/blnk?sslmode=disable",
		},
		Queue: config.QueueConfig{
			WebhookQueue:     "webhook_queue_test_overdraft",
			IndexQueue:       "index_queue_test_overdraft",
			TransactionQueue: "transaction_queue_test_overdraft",
			NumberOfQueues:   1,
		},
		Server: config.ServerConfig{
			SecretKey: "test-secret",
		},
		Transaction: config.TransactionConfig{
			BatchSize:        100,
			MaxQueueSize:     1000,
			LockDuration:     time.Second * 30,
			IndexQueuePrefix: "test_index",
		},
	}
	config.ConfigStore.Store(cnf)

	// Construct the specific transaction queue name
	transactionQueueName := fmt.Sprintf("%s_%d", cnf.Queue.TransactionQueue, 1)

	ds, err := database.NewDataSource(cnf)
	require.NoError(t, err, "Failed to create datasource")

	blnk, err := NewBlnk(ds)
	require.NoError(t, err, "Failed to create Blnk instance")

	// Start the test Asynq worker to process queued transactions
	cleanupWorker := startTestAsynqWorker(t, cnf, blnk, transactionQueueName)
	defer cleanupWorker()

	// Step 1: Create source and destination balances
	sourceBalance := &model.Balance{
		Currency: "USD",
		LedgerID: "general_ledger_id",
	}
	destBalance := &model.Balance{
		Currency: "USD",
		LedgerID: "general_ledger_id",
	}

	source, err := ds.CreateBalance(*sourceBalance)
	require.NoError(t, err, "Failed to create source balance")

	dest, err := ds.CreateBalance(*destBalance)
	require.NoError(t, err, "Failed to create destination balance")

	// Verify source balance after deposit
	sourceAfterDeposit, err := ds.GetBalanceByIDLite(source.BalanceID)
	require.NoError(t, err, "Failed to get source balance after deposit")

	expectedSourceBalance := big.NewInt(0)
	require.Equal(t, 0, sourceAfterDeposit.Balance.Cmp(expectedSourceBalance),
		"Source balance should be 0")

	// Step 3: Create an inflight transaction with amount greater than source balance (1000 > 500)
	inflightAmount := 1000.0 // This is greater than the source balance of 500
	txnRef := "overdraft_txn_" + model.GenerateUUIDWithSuffix("test")
	txn := &model.Transaction{
		Reference:      txnRef,
		Source:         source.BalanceID,
		Destination:    dest.BalanceID,
		Amount:         inflightAmount,
		Inflight:       true,
		Currency:       "USD",
		AllowOverdraft: false, // Don't allow overdraft - this will cause rejection
		Precision:      100,
		MetaData:       map[string]interface{}{"test": "overdraft_test"},
		// Not using SkipQueue so it goes through the queue
	}

	// Queue the inflight transaction
	// When an inflight transaction is queued, it should be in QUEUED status initially
	queuedTxn, err := blnk.QueueTransaction(ctx, txn)
	require.NoError(t, err, "Failed to queue inflight transaction")
	require.Equal(t, StatusQueued, queuedTxn.Status, "Transaction should be QUEUED")

	// Poll for the transaction to be processed by the worker
	// Since the transaction amount exceeds the balance and AllowOverdraft is false,
	// it should be rejected due to insufficient funds
	// The rejected transaction will have a "_q" suffix added to its reference
	rejectedTxn, err := pollForTransactionStatus(ctx, ds, txnRef+"_q", StatusRejected, 500*time.Millisecond, 5*time.Second)
	require.NoError(t, err, "Failed to poll for rejected transaction status")
	require.Equal(t, StatusRejected, rejectedTxn.Status, "Transaction should be REJECTED due to insufficient funds")

	// Step 4: Now attempt to commit using the commit worker
	// This simulates the actual flow where a commit request goes through the queue
	// ProcessTransactionInBatches will use the CommitWorker to process the transaction
	committedTxns, err := blnk.ProcessTransactionInBatches(ctx, queuedTxn.TransactionID, big.NewInt(0), 1, false,
		blnk.GetInflightTransactionsByParentID, blnk.CommitWorker)

	// Since the transaction was rejected, the commit attempt should either:
	// 1. Return an error (no inflight transactions found)
	// 2. Return an empty result
	if err != nil {
		t.Logf("Commit failed as expected for rejected transaction: %v", err)
		require.Contains(t, strings.ToLower(err.Error()), "no transaction", "Error should indicate no transactions to commit")
	} else {
		// If no error, the result should be empty or contain no APPLIED transactions
		require.Equal(t, 0, len(committedTxns), "Should have no committed transactions for a rejected transaction")
	}

	// Step 5: Verify balances remain unchanged after attempting to commit rejected transaction
	finalSource, err := ds.GetBalanceByIDLite(source.BalanceID)
	require.NoError(t, err, "Failed to get final source balance")

	finalDest, err := ds.GetBalanceByIDLite(dest.BalanceID)
	require.NoError(t, err, "Failed to get final destination balance")

	// Source should still have its initial balance of 500
	// Destination should still be 0
	// No overdraft should have occurred since the transaction was rejected
	require.Equal(t, 0, finalSource.Balance.Cmp(expectedSourceBalance),
		"Source balance should remain 500 (unchanged) after rejected transaction")
	require.Equal(t, 0, finalDest.Balance.Cmp(big.NewInt(0)),
		"Destination balance should remain 0 (unchanged) after rejected transaction")

	// Inflight balances should be 0 since the transaction was rejected
	require.Equal(t, 0, finalSource.InflightBalance.Cmp(big.NewInt(0)),
		"Source inflight balance should be 0 after rejection")
	require.Equal(t, 0, finalDest.InflightBalance.Cmp(big.NewInt(0)),
		"Destination inflight balance should be 0 after rejection")

	// Verify the rejected transaction status remains REJECTED
	verifyTxn, err := ds.GetTransactionByRef(ctx, txnRef+"_q")
	require.NoError(t, err, "Failed to get rejected transaction")
	require.Equal(t, StatusRejected, verifyTxn.Status, "Transaction should remain REJECTED")

	// Verify the original queued transaction
	originalQueuedTxn, err := ds.GetTransaction(ctx, queuedTxn.TransactionID)
	require.NoError(t, err, "Failed to get original queued transaction")
	require.Equal(t, StatusQueued, originalQueuedTxn.Status, "Original transaction should remain QUEUED")

	t.Logf("Test TestInflightTransactionWithOverdraftOnCommit completed successfully - balances unchanged after attempting to commit rejected transaction")
}

func TestCreateBulkTransactionsSyncNonAtomic(t *testing.T) {
	// Skip in short mode
	if testing.Short() {
		t.Skip("Skipping bulk transaction test in short mode")
	}

	ctx := context.Background()
	cnf := &config.Configuration{
		Redis: config.RedisConfig{
			Dns: "localhost:6379",
		},
		DataSource: config.DataSourceConfig{
			Dns: "postgres://postgres:password@localhost:5432/blnk?sslmode=disable",
		},
		Queue: config.QueueConfig{
			WebhookQueue:     "webhook_queue_test_bulk",
			IndexQueue:       "index_queue_test_bulk",
			TransactionQueue: "transaction_queue_test_bulk",
			NumberOfQueues:   1,
		},
		Server: config.ServerConfig{
			SecretKey: "test-secret",
		},
		Transaction: config.TransactionConfig{
			BatchSize:        100,
			MaxQueueSize:     1000,
			LockDuration:     time.Second * 30,
			IndexQueuePrefix: "test_index",
		},
	}
	config.ConfigStore.Store(cnf)

	ds, err := database.NewDataSource(cnf)
	require.NoError(t, err, "Failed to create datasource")

	blnk, err := NewBlnk(ds)
	require.NoError(t, err, "Failed to create Blnk instance")

	// Create test balances
	sourceBalance := &model.Balance{
		Currency: "USD",
		LedgerID: "general_ledger_id",
	}
	destBalance1 := &model.Balance{
		Currency: "USD",
		LedgerID: "general_ledger_id",
	}
	destBalance2 := &model.Balance{
		Currency: "USD",
		LedgerID: "general_ledger_id",
	}

	source, err := ds.CreateBalance(*sourceBalance)
	require.NoError(t, err, "Failed to create source balance")

	dest1, err := ds.CreateBalance(*destBalance1)
	require.NoError(t, err, "Failed to create destination balance 1")

	dest2, err := ds.CreateBalance(*destBalance2)
	require.NoError(t, err, "Failed to create destination balance 2")

	// Create bulk transactions
	transactions := []*model.Transaction{
		{
			Reference:      "bulk_txn_1_" + model.GenerateUUIDWithSuffix("test"),
			Source:         source.BalanceID,
			Destination:    dest1.BalanceID,
			Amount:         100,
			Currency:       "USD",
			AllowOverdraft: true,
			Precision:      100,
			MetaData:       map[string]interface{}{"test": "bulk_1"},
		},
		{
			Reference:      "bulk_txn_2_" + model.GenerateUUIDWithSuffix("test"),
			Source:         source.BalanceID,
			Destination:    dest2.BalanceID,
			Amount:         200,
			Currency:       "USD",
			AllowOverdraft: true,
			Precision:      100,
			MetaData:       map[string]interface{}{"test": "bulk_2"},
		},
	}

	req := &model.BulkTransactionRequest{
		Transactions: transactions,
		Atomic:       false, // Non-atomic
		Inflight:     false, // Standard transactions
		RunAsync:     false, // Synchronous
		SkipQueue:    true,  // Process immediately
	}

	// Execute bulk transaction
	result, err := blnk.CreateBulkTransactions(ctx, req)
	require.NoError(t, err, "Failed to create bulk transactions")
	require.NotNil(t, result, "Result should not be nil")
	require.Equal(t, "applied", result.Status, "Status should be applied")
	require.Equal(t, 2, result.TransactionCount, "Should have processed 2 transactions")
	require.NotEmpty(t, result.BatchID, "Batch ID should not be empty")

	// Verify balances
	updatedSource, err := ds.GetBalanceByIDLite(source.BalanceID)
	require.NoError(t, err, "Failed to get updated source balance")

	updatedDest1, err := ds.GetBalanceByIDLite(dest1.BalanceID)
	require.NoError(t, err, "Failed to get updated destination balance 1")

	updatedDest2, err := ds.GetBalanceByIDLite(dest2.BalanceID)
	require.NoError(t, err, "Failed to get updated destination balance 2")

	// Expected balance changes
	expectedSourceDebit := big.NewInt(int64(-300) * 100) // -100 - 200 = -300
	expectedDest1Credit := big.NewInt(int64(100) * 100)  // +100
	expectedDest2Credit := big.NewInt(int64(200) * 100)  // +200

	require.Equal(t, 0, updatedSource.Balance.Cmp(expectedSourceDebit),
		"Source balance should be debited by total amount")
	require.Equal(t, 0, updatedDest1.Balance.Cmp(expectedDest1Credit),
		"Destination 1 balance should be credited by first transaction amount")
	require.Equal(t, 0, updatedDest2.Balance.Cmp(expectedDest2Credit),
		"Destination 2 balance should be credited by second transaction amount")
}

func TestCreateBulkTransactionsSyncAtomicSuccess(t *testing.T) {
	// Skip in short mode
	if testing.Short() {
		t.Skip("Skipping bulk transaction test in short mode")
	}

	ctx := context.Background()
	cnf := &config.Configuration{
		Redis: config.RedisConfig{
			Dns: "localhost:6379",
		},
		DataSource: config.DataSourceConfig{
			Dns: "postgres://postgres:password@localhost:5432/blnk?sslmode=disable",
		},
		Queue: config.QueueConfig{
			WebhookQueue:     "webhook_queue_test_bulk_atomic",
			IndexQueue:       "index_queue_test_bulk_atomic",
			TransactionQueue: "transaction_queue_test_bulk_atomic",
			NumberOfQueues:   1,
		},
		Server: config.ServerConfig{
			SecretKey: "test-secret",
		},
		Transaction: config.TransactionConfig{
			BatchSize:        100,
			MaxQueueSize:     1000,
			LockDuration:     time.Second * 30,
			IndexQueuePrefix: "test_index",
		},
	}
	config.ConfigStore.Store(cnf)

	ds, err := database.NewDataSource(cnf)
	require.NoError(t, err, "Failed to create datasource")

	blnk, err := NewBlnk(ds)
	require.NoError(t, err, "Failed to create Blnk instance")

	// Create test balances
	sourceBalance := &model.Balance{
		Currency: "USD",
		LedgerID: "general_ledger_id",
	}
	destBalance1 := &model.Balance{
		Currency: "USD",
		LedgerID: "general_ledger_id",
	}
	destBalance2 := &model.Balance{
		Currency: "USD",
		LedgerID: "general_ledger_id",
	}

	source, err := ds.CreateBalance(*sourceBalance)
	require.NoError(t, err, "Failed to create source balance")

	dest1, err := ds.CreateBalance(*destBalance1)
	require.NoError(t, err, "Failed to create destination balance 1")

	dest2, err := ds.CreateBalance(*destBalance2)
	require.NoError(t, err, "Failed to create destination balance 2")

	// Create bulk transactions
	transactions := []*model.Transaction{
		{
			Reference:      "bulk_atomic_txn_1_" + model.GenerateUUIDWithSuffix("test"),
			Source:         source.BalanceID,
			Destination:    dest1.BalanceID,
			Amount:         100,
			Currency:       "USD",
			AllowOverdraft: true,
			Precision:      100,
			MetaData:       map[string]interface{}{"test": "bulk_atomic_1"},
		},
		{
			Reference:      "bulk_atomic_txn_2_" + model.GenerateUUIDWithSuffix("test"),
			Source:         source.BalanceID,
			Destination:    dest2.BalanceID,
			Amount:         200,
			Currency:       "USD",
			AllowOverdraft: true,
			Precision:      100,
			MetaData:       map[string]interface{}{"test": "bulk_atomic_2"},
		},
	}

	req := &model.BulkTransactionRequest{
		Transactions: transactions,
		Atomic:       true,  // Atomic
		Inflight:     false, // Standard transactions
		RunAsync:     false, // Synchronous
		SkipQueue:    true,  // Process immediately
	}

	// Execute bulk transaction
	result, err := blnk.CreateBulkTransactions(ctx, req)
	require.NoError(t, err, "Failed to create bulk transactions")
	require.NotNil(t, result, "Result should not be nil")
	require.Equal(t, "applied", result.Status, "Status should be applied")
	require.Equal(t, 2, result.TransactionCount, "Should have processed 2 transactions")
	require.NotEmpty(t, result.BatchID, "Batch ID should not be empty")

	// Verify balances
	updatedSource, err := ds.GetBalanceByIDLite(source.BalanceID)
	require.NoError(t, err, "Failed to get updated source balance")

	updatedDest1, err := ds.GetBalanceByIDLite(dest1.BalanceID)
	require.NoError(t, err, "Failed to get updated destination balance 1")

	updatedDest2, err := ds.GetBalanceByIDLite(dest2.BalanceID)
	require.NoError(t, err, "Failed to get updated destination balance 2")

	// Expected balance changes
	expectedSourceDebit := big.NewInt(int64(-300) * 100) // -100 - 200 = -300
	expectedDest1Credit := big.NewInt(int64(100) * 100)  // +100
	expectedDest2Credit := big.NewInt(int64(200) * 100)  // +200

	require.Equal(t, 0, updatedSource.Balance.Cmp(expectedSourceDebit),
		"Source balance should be debited by total amount")
	require.Equal(t, 0, updatedDest1.Balance.Cmp(expectedDest1Credit),
		"Destination 1 balance should be credited by first transaction amount")
	require.Equal(t, 0, updatedDest2.Balance.Cmp(expectedDest2Credit),
		"Destination 2 balance should be credited by second transaction amount")
}

func TestCreateBulkTransactionsInflightAsync(t *testing.T) {
	// Skip in short mode
	if testing.Short() {
		t.Skip("Skipping bulk inflight transaction test in short mode")
	}

	ctx := context.Background()
	cnf := &config.Configuration{
		Redis: config.RedisConfig{
			Dns: "localhost:6379",
		},
		DataSource: config.DataSourceConfig{
			Dns: "postgres://postgres:password@localhost:5432/blnk?sslmode=disable",
		},
		Queue: config.QueueConfig{
			WebhookQueue:     "webhook_queue_test_bulk_inflight",
			IndexQueue:       "index_queue_test_bulk_inflight",
			TransactionQueue: "transaction_queue_test_bulk_inflight",
			NumberOfQueues:   1,
		},
		Server: config.ServerConfig{
			SecretKey: "test-secret",
		},
		Transaction: config.TransactionConfig{
			BatchSize:        100,
			MaxQueueSize:     1000,
			LockDuration:     time.Second * 30,
			IndexQueuePrefix: "test_index",
		},
	}
	config.ConfigStore.Store(cnf)

	ds, err := database.NewDataSource(cnf)
	require.NoError(t, err, "Failed to create datasource")

	blnk, err := NewBlnk(ds)
	require.NoError(t, err, "Failed to create Blnk instance")

	// Start the test Asynq worker to process inflight transactions
	transactionQueueName := fmt.Sprintf("%s_%d", cnf.Queue.TransactionQueue, 1)
	cleanupWorker := startTestAsynqWorker(t, cnf, blnk, transactionQueueName)
	defer cleanupWorker()

	// Create test balances
	sourceBalance := &model.Balance{
		Currency: "USD",
		LedgerID: "general_ledger_id",
	}
	destBalance1 := &model.Balance{
		Currency: "USD",
		LedgerID: "general_ledger_id",
	}
	destBalance2 := &model.Balance{
		Currency: "USD",
		LedgerID: "general_ledger_id",
	}

	source, err := ds.CreateBalance(*sourceBalance)
	require.NoError(t, err, "Failed to create source balance")

	dest1, err := ds.CreateBalance(*destBalance1)
	require.NoError(t, err, "Failed to create destination balance 1")

	dest2, err := ds.CreateBalance(*destBalance2)
	require.NoError(t, err, "Failed to create destination balance 2")

	// Create bulk inflight transactions
	transactions := []*model.Transaction{
		{
			Reference:      "bulk_inflight_txn_1_" + model.GenerateUUIDWithSuffix("test"),
			Source:         source.BalanceID,
			Destination:    dest1.BalanceID,
			Amount:         100,
			Currency:       "USD",
			AllowOverdraft: true,
			Precision:      100,
			MetaData:       map[string]interface{}{"test": "bulk_inflight_1"},
		},
		{
			Reference:      "bulk_inflight_txn_2_" + model.GenerateUUIDWithSuffix("test"),
			Source:         source.BalanceID,
			Destination:    dest2.BalanceID,
			Amount:         200,
			Currency:       "USD",
			AllowOverdraft: true,
			Precision:      100,
			MetaData:       map[string]interface{}{"test": "bulk_inflight_2"},
		},
	}

	req := &model.BulkTransactionRequest{
		Transactions: transactions,
		Atomic:       false, // Non-atomic
		Inflight:     true,  // Inflight transactions
		RunAsync:     true,  // Asynchronous processing
		SkipQueue:    false, // Use queue for async processing
	}

	// Execute bulk inflight transaction asynchronously
	result, err := blnk.CreateBulkTransactions(ctx, req)
	require.NoError(t, err, "Failed to create bulk inflight transactions")
	require.NotNil(t, result, "Result should not be nil")
	require.Equal(t, "processing", result.Status, "Status should be processing for async")
	require.NotEmpty(t, result.BatchID, "Batch ID should not be empty")

	// Poll for the inflight transactions to be processed by the worker
	pollInterval := 500 * time.Millisecond
	timeoutDuration := 15 * time.Second

	// Poll for each inflight transaction to be created
	inflightTxnRef1 := fmt.Sprintf("%s_q", transactions[0].Reference)
	inflightTxnRef2 := fmt.Sprintf("%s_q", transactions[1].Reference)

	inflightTxn1, err := pollForTransactionStatus(ctx, ds, inflightTxnRef1, StatusInflight, pollInterval, timeoutDuration)
	require.NoError(t, err, "Failed to poll for first inflight transaction")
	require.Equal(t, StatusInflight, inflightTxn1.Status, "First transaction should be INFLIGHT")

	inflightTxn2, err := pollForTransactionStatus(ctx, ds, inflightTxnRef2, StatusInflight, pollInterval, timeoutDuration)
	require.NoError(t, err, "Failed to poll for second inflight transaction")
	require.Equal(t, StatusInflight, inflightTxn2.Status, "Second transaction should be INFLIGHT")

	// Verify balances after async processing - should show inflight changes
	updatedSource, err := ds.GetBalanceByIDLite(source.BalanceID)
	require.NoError(t, err, "Failed to get updated source balance")

	updatedDest1, err := ds.GetBalanceByIDLite(dest1.BalanceID)
	require.NoError(t, err, "Failed to get updated destination balance 1")

	updatedDest2, err := ds.GetBalanceByIDLite(dest2.BalanceID)
	require.NoError(t, err, "Failed to get updated destination balance 2")

	// Expected inflight balance changes
	expectedSourceInflightDebit := big.NewInt(int64(-300) * 100) // -100 - 200 = -300
	expectedDest1InflightCredit := big.NewInt(int64(100) * 100)  // +100
	expectedDest2InflightCredit := big.NewInt(int64(200) * 100)  // +200

	// Main balances should remain zero
	require.Equal(t, 0, updatedSource.Balance.Cmp(big.NewInt(0)),
		"Source main balance should remain zero")
	require.Equal(t, 0, updatedDest1.Balance.Cmp(big.NewInt(0)),
		"Destination 1 main balance should remain zero")
	require.Equal(t, 0, updatedDest2.Balance.Cmp(big.NewInt(0)),
		"Destination 2 main balance should remain zero")

	// Inflight balances should reflect the transactions
	require.Equal(t, 0, updatedSource.InflightBalance.Cmp(expectedSourceInflightDebit),
		"Source inflight balance should be debited by total amount")
	require.Equal(t, 0, updatedDest1.InflightBalance.Cmp(expectedDest1InflightCredit),
		"Destination 1 inflight balance should be credited by first transaction amount")
	require.Equal(t, 0, updatedDest2.InflightBalance.Cmp(expectedDest2InflightCredit),
		"Destination 2 inflight balance should be credited by second transaction amount")

	// Now commit all inflight transactions using the commit worker with batch ID
	committedTxns, err := blnk.ProcessTransactionInBatches(ctx, result.BatchID, big.NewInt(0), 1, false,
		blnk.GetInflightTransactionsByParentID, blnk.CommitWorker)
	require.NoError(t, err, "Failed to commit inflight transactions")
	require.Equal(t, 2, len(committedTxns), "Should have committed 2 transactions")

	// Poll for both commit transactions to be processed
	for _, commitTxn := range committedTxns {
		appliedCommitTxn, err := pollForTransactionStatus(ctx, ds, commitTxn.Reference, StatusApplied, pollInterval, timeoutDuration)
		require.NoError(t, err, "Failed to poll for commit transaction")
		require.Equal(t, StatusApplied, appliedCommitTxn.Status, "Commit transaction should be APPLIED")
	}

	// Verify balances after committing all transactions
	sourceAfterCommit, err := ds.GetBalanceByIDLite(source.BalanceID)
	require.NoError(t, err, "Failed to get source balance after commit")

	dest1AfterCommit, err := ds.GetBalanceByIDLite(dest1.BalanceID)
	require.NoError(t, err, "Failed to get destination 1 balance after commit")

	dest2AfterCommit, err := ds.GetBalanceByIDLite(dest2.BalanceID)
	require.NoError(t, err, "Failed to get destination 2 balance after commit")

	// After committing both transactions, all amounts should be in main balances
	expectedSourceAfterCommit := big.NewInt(int64(-300) * 100) // -300 committed (both transactions)
	expectedDest1AfterCommit := big.NewInt(int64(100) * 100)   // +100 committed
	expectedDest2AfterCommit := big.NewInt(int64(200) * 100)   // +200 committed

	require.Equal(t, 0, sourceAfterCommit.Balance.Cmp(expectedSourceAfterCommit),
		"Source balance should reflect both committed transactions")
	require.Equal(t, 0, sourceAfterCommit.InflightBalance.Cmp(big.NewInt(0)),
		"Source inflight balance should be zero after all commits")
	require.Equal(t, 0, dest1AfterCommit.Balance.Cmp(expectedDest1AfterCommit),
		"Destination 1 balance should reflect committed amount")
	require.Equal(t, 0, dest1AfterCommit.InflightBalance.Cmp(big.NewInt(0)),
		"Destination 1 inflight balance should be zero after commit")
	require.Equal(t, 0, dest2AfterCommit.Balance.Cmp(expectedDest2AfterCommit),
		"Destination 2 balance should reflect committed amount")
	require.Equal(t, 0, dest2AfterCommit.InflightBalance.Cmp(big.NewInt(0)),
		"Destination 2 inflight balance should be zero after commit")
}

func TestCreateBulkTransactionsAsync(t *testing.T) {
	// Skip in short mode
	if testing.Short() {
		t.Skip("Skipping async bulk transaction test in short mode")
	}

	ctx := context.Background()
	cnf := &config.Configuration{
		Redis: config.RedisConfig{
			Dns: "localhost:6379",
		},
		DataSource: config.DataSourceConfig{
			Dns: "postgres://postgres:password@localhost:5432/blnk?sslmode=disable",
		},
		Queue: config.QueueConfig{
			WebhookQueue:     "webhook_queue_test_bulk_async",
			IndexQueue:       "index_queue_test_bulk_async",
			TransactionQueue: "transaction_queue_test_bulk_async",
			NumberOfQueues:   1,
		},
		Server: config.ServerConfig{
			SecretKey: "test-secret",
		},
		Transaction: config.TransactionConfig{
			BatchSize:        100,
			MaxQueueSize:     1000,
			LockDuration:     time.Second * 30,
			IndexQueuePrefix: "test_index",
		},
	}
	config.ConfigStore.Store(cnf)

	ds, err := database.NewDataSource(cnf)
	require.NoError(t, err, "Failed to create datasource")

	blnk, err := NewBlnk(ds)
	require.NoError(t, err, "Failed to create Blnk instance")

	// Create test balances
	sourceBalance := &model.Balance{
		Currency: "USD",
		LedgerID: "general_ledger_id",
	}
	destBalance1 := &model.Balance{
		Currency: "USD",
		LedgerID: "general_ledger_id",
	}
	destBalance2 := &model.Balance{
		Currency: "USD",
		LedgerID: "general_ledger_id",
	}

	source, err := ds.CreateBalance(*sourceBalance)
	require.NoError(t, err, "Failed to create source balance")

	dest1, err := ds.CreateBalance(*destBalance1)
	require.NoError(t, err, "Failed to create destination balance 1")

	dest2, err := ds.CreateBalance(*destBalance2)
	require.NoError(t, err, "Failed to create destination balance 2")

	// Create bulk transactions
	transactions := []*model.Transaction{
		{
			Reference:      "bulk_async_txn_1_" + model.GenerateUUIDWithSuffix("test"),
			Source:         source.BalanceID,
			Destination:    dest1.BalanceID,
			Amount:         100,
			Currency:       "USD",
			AllowOverdraft: true,
			Precision:      100,
			MetaData:       map[string]interface{}{"test": "bulk_async_1"},
		},
		{
			Reference:      "bulk_async_txn_2_" + model.GenerateUUIDWithSuffix("test"),
			Source:         source.BalanceID,
			Destination:    dest2.BalanceID,
			Amount:         200,
			Currency:       "USD",
			AllowOverdraft: true,
			Precision:      100,
			MetaData:       map[string]interface{}{"test": "bulk_async_2"},
		},
	}

	req := &model.BulkTransactionRequest{
		Transactions: transactions,
		Atomic:       false, // Non-atomic
		Inflight:     false, // Standard transactions
		RunAsync:     true,  // Asynchronous
		SkipQueue:    true,  // Process immediately when picked up
	}

	// Execute bulk transaction asynchronously
	result, err := blnk.CreateBulkTransactions(ctx, req)
	require.NoError(t, err, "Failed to create async bulk transactions")
	require.NotNil(t, result, "Result should not be nil")
	require.Equal(t, "processing", result.Status, "Status should be processing for async")
	require.NotEmpty(t, result.BatchID, "Batch ID should not be empty")

	// Wait a bit for async processing to complete
	time.Sleep(2 * time.Second)

	// Verify balances were updated by the async processing
	updatedSource, err := ds.GetBalanceByIDLite(source.BalanceID)
	require.NoError(t, err, "Failed to get updated source balance")

	updatedDest1, err := ds.GetBalanceByIDLite(dest1.BalanceID)
	require.NoError(t, err, "Failed to get updated destination balance 1")

	updatedDest2, err := ds.GetBalanceByIDLite(dest2.BalanceID)
	require.NoError(t, err, "Failed to get updated destination balance 2")

	// Expected balance changes
	expectedSourceDebit := big.NewInt(int64(-300) * 100) // -100 - 200 = -300
	expectedDest1Credit := big.NewInt(int64(100) * 100)  // +100
	expectedDest2Credit := big.NewInt(int64(200) * 100)  // +200

	require.Equal(t, 0, updatedSource.Balance.Cmp(expectedSourceDebit),
		"Source balance should be debited by total amount")
	require.Equal(t, 0, updatedDest1.Balance.Cmp(expectedDest1Credit),
		"Destination 1 balance should be credited by first transaction amount")
	require.Equal(t, 0, updatedDest2.Balance.Cmp(expectedDest2Credit),
		"Destination 2 balance should be credited by second transaction amount")
}

func TestCreateBulkTransactionsNonAtomicPartialFailure(t *testing.T) {
	// Skip in short mode
	if testing.Short() {
		t.Skip("Skipping bulk non-atomic partial failure test in short mode")
	}

	ctx := context.Background()
	cnf := &config.Configuration{
		Redis: config.RedisConfig{
			Dns: "localhost:6379",
		},
		DataSource: config.DataSourceConfig{
			Dns: "postgres://postgres:password@localhost:5432/blnk?sslmode=disable",
		},
		Queue: config.QueueConfig{
			WebhookQueue:     "webhook_queue_test_bulk_non_atomic_fail",
			IndexQueue:       "index_queue_test_bulk_non_atomic_fail",
			TransactionQueue: "transaction_queue_test_bulk_non_atomic_fail",
			NumberOfQueues:   1,
		},
		Server: config.ServerConfig{
			SecretKey: "test-secret",
		},
		Transaction: config.TransactionConfig{
			BatchSize:        100,
			MaxQueueSize:     1000,
			LockDuration:     time.Second * 30,
			IndexQueuePrefix: "test_index",
		},
	}
	config.ConfigStore.Store(cnf)

	ds, err := database.NewDataSource(cnf)
	require.NoError(t, err, "Failed to create datasource")

	blnk, err := NewBlnk(ds)
	require.NoError(t, err, "Failed to create Blnk instance")

	// Create test balances
	sourceBalance := &model.Balance{
		Currency: "USD",
		LedgerID: "general_ledger_id",
	}
	destBalance1 := &model.Balance{
		Currency: "USD",
		LedgerID: "general_ledger_id",
	}

	source, err := ds.CreateBalance(*sourceBalance)
	require.NoError(t, err, "Failed to create source balance")

	dest1, err := ds.CreateBalance(*destBalance1)
	require.NoError(t, err, "Failed to create destination balance 1")

	// Create bulk transactions - second one will fail due to invalid destination
	transactions := []*model.Transaction{
		{
			Reference:      "bulk_non_atomic_fail_txn_1_" + model.GenerateUUIDWithSuffix("test"),
			Source:         source.BalanceID,
			Destination:    dest1.BalanceID,
			Amount:         100,
			Currency:       "USD",
			AllowOverdraft: true,
			Precision:      100,
			MetaData:       map[string]interface{}{"test": "bulk_non_atomic_fail_1"},
		},
		{
			Reference:      "bulk_non_atomic_fail_txn_2_" + model.GenerateUUIDWithSuffix("test"),
			Source:         source.BalanceID,
			Destination:    "invalid_destination_id", // This will cause failure
			Amount:         200,
			Currency:       "USD",
			AllowOverdraft: true,
			Precision:      100,
			MetaData:       map[string]interface{}{"test": "bulk_non_atomic_fail_2"},
		},
	}

	req := &model.BulkTransactionRequest{
		Transactions: transactions,
		Atomic:       false, // Non-atomic - first transaction should remain
		Inflight:     false, // Standard transactions
		RunAsync:     false, // Synchronous
		SkipQueue:    true,  // Process immediately
	}

	// Execute bulk transaction - should fail but first transaction should remain
	result, err := blnk.CreateBulkTransactions(ctx, req)
	require.Error(t, err, "Bulk transaction should fail due to invalid destination")
	require.NotNil(t, result, "Result should not be nil even on failure")
	require.Equal(t, "failed", result.Status, "Status should be failed")
	require.NotEmpty(t, result.BatchID, "Batch ID should not be empty")
	require.Contains(t, result.Error, "Previous transactions were not rolled back",
		"Error should mention that previous transactions were not rolled back")

	// Verify that the first transaction was processed but the second was not
	updatedSource, err := ds.GetBalanceByIDLite(source.BalanceID)
	require.NoError(t, err, "Failed to get updated source balance")

	updatedDest1, err := ds.GetBalanceByIDLite(dest1.BalanceID)
	require.NoError(t, err, "Failed to get updated destination balance 1")

	// First transaction should have been processed (non-atomic behavior)
	expectedSourceDebit := big.NewInt(int64(-100) * 100) // Only first transaction processed
	expectedDest1Credit := big.NewInt(int64(100) * 100)  // Only first transaction processed

	require.Equal(t, 0, updatedSource.Balance.Cmp(expectedSourceDebit),
		"Source balance should reflect only the first transaction (non-atomic)")
	require.Equal(t, 0, updatedDest1.Balance.Cmp(expectedDest1Credit),
		"Destination 1 balance should reflect only the first transaction (non-atomic)")
}

func TestGetTransaction_Mock(t *testing.T) {
	cnf := &config.Configuration{
		Redis: config.RedisConfig{Dns: "localhost:6379"},
		Queue: config.QueueConfig{
			WebhookQueue:     "webhook_queue",
			TransactionQueue: "transaction_queue",
			NumberOfQueues:   1,
		},
	}
	config.ConfigStore.Store(cnf)

	t.Run("Success", func(t *testing.T) {
		mockDS := new(mocks.MockDataSource)

		txnID := "txn_123"
		expectedTxn := &model.Transaction{
			TransactionID: txnID,
			Amount:        100.0,
			Reference:     "ref_123",
			Status:        "APPLIED",
			PreciseAmount: big.NewInt(10000),
		}

		mockDS.On("GetTransaction", mock.Anything, txnID).Return(expectedTxn, nil)

		blnk := &Blnk{datasource: mockDS, config: cnf}
		result, err := blnk.GetTransaction(context.Background(), txnID)

		assert.NoError(t, err)
		assert.NotNil(t, result)
		assert.Equal(t, txnID, result.TransactionID)
		assert.Equal(t, "APPLIED", result.Status)
		mockDS.AssertExpectations(t)
	})

	t.Run("Not Found", func(t *testing.T) {
		mockDS := new(mocks.MockDataSource)

		txnID := "txn_nonexistent"
		mockDS.On("GetTransaction", mock.Anything, txnID).Return((*model.Transaction)(nil), fmt.Errorf("not found"))

		blnk := &Blnk{datasource: mockDS, config: cnf}
		result, err := blnk.GetTransaction(context.Background(), txnID)

		assert.Error(t, err)
		assert.Nil(t, result)
		mockDS.AssertExpectations(t)
	})
}

func TestGetAllTransactions_Mock(t *testing.T) {
	cnf := &config.Configuration{
		Redis: config.RedisConfig{Dns: "localhost:6379"},
		Queue: config.QueueConfig{
			WebhookQueue:     "webhook_queue",
			TransactionQueue: "transaction_queue",
			NumberOfQueues:   1,
		},
	}
	config.ConfigStore.Store(cnf)

	t.Run("Success", func(t *testing.T) {
		mockDS := new(mocks.MockDataSource)

		expectedTxns := []model.Transaction{
			{TransactionID: "txn_1", Amount: 100.0, Status: "APPLIED"},
			{TransactionID: "txn_2", Amount: 200.0, Status: "APPLIED"},
		}

		mockDS.On("GetAllTransactions", 10, 0).Return(expectedTxns, nil)

		blnk := &Blnk{datasource: mockDS, config: cnf}
		result, err := blnk.GetAllTransactions(10, 0)

		assert.NoError(t, err)
		assert.Len(t, result, 2)
		assert.Equal(t, "txn_1", result[0].TransactionID)
		assert.Equal(t, "txn_2", result[1].TransactionID)
		mockDS.AssertExpectations(t)
	})

	t.Run("Empty Result", func(t *testing.T) {
		mockDS := new(mocks.MockDataSource)

		mockDS.On("GetAllTransactions", 10, 100).Return([]model.Transaction{}, nil)

		blnk := &Blnk{datasource: mockDS, config: cnf}
		result, err := blnk.GetAllTransactions(10, 100)

		assert.NoError(t, err)
		assert.Len(t, result, 0)
		mockDS.AssertExpectations(t)
	})

	t.Run("Database Error", func(t *testing.T) {
		mockDS := new(mocks.MockDataSource)

		mockDS.On("GetAllTransactions", 10, 0).Return([]model.Transaction(nil), fmt.Errorf("database error"))

		blnk := &Blnk{datasource: mockDS, config: cnf}
		result, err := blnk.GetAllTransactions(10, 0)

		assert.Error(t, err)
		assert.Nil(t, result)
		mockDS.AssertExpectations(t)
	})
}

func TestGetTransactionByRef_Mock(t *testing.T) {
	cnf := &config.Configuration{
		Redis: config.RedisConfig{Dns: "localhost:6379"},
		Queue: config.QueueConfig{
			WebhookQueue:     "webhook_queue",
			TransactionQueue: "transaction_queue",
			NumberOfQueues:   1,
		},
	}
	config.ConfigStore.Store(cnf)

	t.Run("Success", func(t *testing.T) {
		mockDS := new(mocks.MockDataSource)

		reference := "ref_unique_123"
		expectedTxn := model.Transaction{
			TransactionID: "txn_found",
			Reference:     reference,
			Amount:        500.0,
			Status:        "APPLIED",
		}

		mockDS.On("GetTransactionByRef", mock.Anything, reference).Return(expectedTxn, nil)

		blnk := &Blnk{datasource: mockDS, config: cnf}
		result, err := blnk.GetTransactionByRef(context.Background(), reference)

		assert.NoError(t, err)
		assert.Equal(t, "txn_found", result.TransactionID)
		assert.Equal(t, reference, result.Reference)
		mockDS.AssertExpectations(t)
	})

	t.Run("Not Found", func(t *testing.T) {
		mockDS := new(mocks.MockDataSource)

		reference := "ref_nonexistent"
		mockDS.On("GetTransactionByRef", mock.Anything, reference).Return(model.Transaction{}, fmt.Errorf("not found"))

		blnk := &Blnk{datasource: mockDS, config: cnf}
		result, err := blnk.GetTransactionByRef(context.Background(), reference)

		assert.Error(t, err)
		assert.Equal(t, model.Transaction{}, result)
		mockDS.AssertExpectations(t)
	})
}

func TestUpdateTransactionStatus_Mock(t *testing.T) {
	cnf := &config.Configuration{
		Redis: config.RedisConfig{Dns: "localhost:6379"},
		Queue: config.QueueConfig{
			WebhookQueue:     "webhook_queue",
			TransactionQueue: "transaction_queue",
			NumberOfQueues:   1,
		},
	}
	config.ConfigStore.Store(cnf)

	t.Run("Success", func(t *testing.T) {
		mockDS := new(mocks.MockDataSource)

		txnID := "txn_to_update"
		newStatus := "VOID"

		mockDS.On("UpdateTransactionStatus", mock.Anything, txnID, newStatus).Return(nil)

		blnk := &Blnk{datasource: mockDS, config: cnf}
		err := blnk.UpdateTransactionStatus(context.Background(), txnID, newStatus)

		assert.NoError(t, err)
		mockDS.AssertExpectations(t)
	})

	t.Run("Error", func(t *testing.T) {
		mockDS := new(mocks.MockDataSource)

		txnID := "txn_nonexistent"
		newStatus := "VOID"

		mockDS.On("UpdateTransactionStatus", mock.Anything, txnID, newStatus).Return(fmt.Errorf("not found"))

		blnk := &Blnk{datasource: mockDS, config: cnf}
		err := blnk.UpdateTransactionStatus(context.Background(), txnID, newStatus)

		assert.Error(t, err)
		mockDS.AssertExpectations(t)
	})
}

func TestGetRefundableTransactionsByParentID_Mock(t *testing.T) {
	cnf := &config.Configuration{
		Redis: config.RedisConfig{Dns: "localhost:6379"},
		Queue: config.QueueConfig{
			WebhookQueue:     "webhook_queue",
			TransactionQueue: "transaction_queue",
			NumberOfQueues:   1,
		},
	}
	config.ConfigStore.Store(cnf)

	t.Run("Success with transactions", func(t *testing.T) {
		mockDS := new(mocks.MockDataSource)

		parentID := "bulk_parent_123"
		expectedTxns := []*model.Transaction{
			{TransactionID: "txn_child_1", ParentTransaction: parentID, Status: "APPLIED"},
			{TransactionID: "txn_child_2", ParentTransaction: parentID, Status: "APPLIED"},
		}

		mockDS.On("GetRefundableTransactionsByParentID", mock.Anything, parentID, 100, int64(0)).Return(expectedTxns, nil)

		blnk := &Blnk{datasource: mockDS, config: cnf}
		result, err := blnk.GetRefundableTransactionsByParentID(context.Background(), parentID, 100, 0)

		assert.NoError(t, err)
		assert.Len(t, result, 2)
		assert.Equal(t, parentID, result[0].ParentTransaction)
		assert.Equal(t, parentID, result[1].ParentTransaction)
		mockDS.AssertExpectations(t)
	})

	t.Run("No refundable transactions", func(t *testing.T) {
		mockDS := new(mocks.MockDataSource)

		parentID := "bulk_no_children"
		mockDS.On("GetRefundableTransactionsByParentID", mock.Anything, parentID, 100, int64(0)).Return([]*model.Transaction{}, nil)

		blnk := &Blnk{datasource: mockDS, config: cnf}
		result, err := blnk.GetRefundableTransactionsByParentID(context.Background(), parentID, 100, 0)

		assert.NoError(t, err)
		assert.Len(t, result, 0)
		mockDS.AssertExpectations(t)
	})

	t.Run("Database Error", func(t *testing.T) {
		mockDS := new(mocks.MockDataSource)

		parentID := "bulk_error"
		mockDS.On("GetRefundableTransactionsByParentID", mock.Anything, parentID, 100, int64(0)).Return(([]*model.Transaction)(nil), fmt.Errorf("db error"))

		blnk := &Blnk{datasource: mockDS, config: cnf}
		result, err := blnk.GetRefundableTransactionsByParentID(context.Background(), parentID, 100, 0)

		assert.Error(t, err)
		assert.Nil(t, result)
		mockDS.AssertExpectations(t)
	})
}

func TestLogRollbackResult_Blnk(t *testing.T) {
	cnf := &config.Configuration{
		Redis: config.RedisConfig{Dns: "localhost:6379"},
		Queue: config.QueueConfig{
			WebhookQueue:     "webhook_queue",
			TransactionQueue: "transaction_queue",
			NumberOfQueues:   1,
		},
	}
	config.ConfigStore.Store(cnf)

	mockDS := new(mocks.MockDataSource)
	blnk := &Blnk{datasource: mockDS, config: cnf}

	t.Run("Log Success", func(t *testing.T) {
		blnk.logRollbackResult("batch_123", "voided", nil)
	})

	t.Run("Log Failure", func(t *testing.T) {
		blnk.logRollbackResult("batch_456", "voided", fmt.Errorf("rollback failed"))
	})
}

func TestContains(t *testing.T) {
	t.Run("Item exists", func(t *testing.T) {
		slice := []string{"apple", "banana", "cherry"}
		assert.True(t, contains(slice, "banana"))
	})

	t.Run("Item does not exist", func(t *testing.T) {
		slice := []string{"apple", "banana", "cherry"}
		assert.False(t, contains(slice, "grape"))
	})

	t.Run("Empty slice", func(t *testing.T) {
		slice := []string{}
		assert.False(t, contains(slice, "anything"))
	})

	t.Run("First element", func(t *testing.T) {
		slice := []string{"first", "second", "third"}
		assert.True(t, contains(slice, "first"))
	})

	t.Run("Last element", func(t *testing.T) {
		slice := []string{"first", "second", "last"}
		assert.True(t, contains(slice, "last"))
	})
}

func TestGetAccountByNumber_Mock(t *testing.T) {
	cnf := &config.Configuration{
		Redis: config.RedisConfig{Dns: "localhost:6379"},
		Queue: config.QueueConfig{
			WebhookQueue:     "webhook_queue",
			TransactionQueue: "transaction_queue",
			NumberOfQueues:   1,
		},
	}
	config.ConfigStore.Store(cnf)

	t.Run("Success", func(t *testing.T) {
		mockDS := new(mocks.MockDataSource)
		accountNumber := "1234567890"
		expectedAccount := &model.Account{
			AccountID: "acc_123",
			Number:    accountNumber,
			Name:      "Test Account",
		}

		mockDS.On("GetAccountByNumber", accountNumber).Return(expectedAccount, nil)

		blnk := &Blnk{datasource: mockDS, config: cnf}
		result, err := blnk.GetAccountByNumber(accountNumber)

		assert.NoError(t, err)
		assert.NotNil(t, result)
		assert.Equal(t, accountNumber, result.Number)
		mockDS.AssertExpectations(t)
	})

	t.Run("Not Found", func(t *testing.T) {
		mockDS := new(mocks.MockDataSource)
		accountNumber := "nonexistent"

		mockDS.On("GetAccountByNumber", accountNumber).Return((*model.Account)(nil), fmt.Errorf("not found"))

		blnk := &Blnk{datasource: mockDS, config: cnf}
		result, err := blnk.GetAccountByNumber(accountNumber)

		assert.Error(t, err)
		assert.Nil(t, result)
		mockDS.AssertExpectations(t)
	})
}

func TestConvertToStructFieldName_Blnk(t *testing.T) {
	tests := []struct {
		input    string
		expected string
	}{
		{"firstName", "FirstName"},
		{"email", "Email"},
		{"phone", "Phone"},
		{"", ""},
		{"A", "A"},
	}

	for _, tc := range tests {
		t.Run(tc.input, func(t *testing.T) {
			result := convertToStructFieldName(tc.input)
			assert.Equal(t, tc.expected, result)
		})
	}
}

func TestUpdateBalanceIdentity_Mock(t *testing.T) {
	cnf := &config.Configuration{
		Redis: config.RedisConfig{Dns: "localhost:6379"},
		Queue: config.QueueConfig{
			WebhookQueue:     "webhook_queue",
			TransactionQueue: "transaction_queue",
			NumberOfQueues:   1,
		},
	}
	config.ConfigStore.Store(cnf)

	t.Run("Success", func(t *testing.T) {
		mockDS := new(mocks.MockDataSource)
		balanceID := "bln_123"
		identityID := "idt_456"

		mockDS.On("GetIdentityByID", identityID).Return(&model.Identity{IdentityID: identityID}, nil)
		mockDS.On("GetBalanceByIDLite", balanceID).Return(&model.Balance{BalanceID: balanceID}, nil)
		mockDS.On("UpdateBalanceIdentity", balanceID, identityID).Return(nil)

		blnk := &Blnk{datasource: mockDS, config: cnf}
		err := blnk.UpdateBalanceIdentity(balanceID, identityID)

		assert.NoError(t, err)
		mockDS.AssertExpectations(t)
	})

	t.Run("Identity not found", func(t *testing.T) {
		mockDS := new(mocks.MockDataSource)
		balanceID := "bln_123"
		identityID := "idt_nonexistent"

		mockDS.On("GetIdentityByID", identityID).Return((*model.Identity)(nil), fmt.Errorf("not found"))

		blnk := &Blnk{datasource: mockDS, config: cnf}
		err := blnk.UpdateBalanceIdentity(balanceID, identityID)

		assert.Error(t, err)
		assert.Contains(t, err.Error(), "identity validation failed")
		mockDS.AssertExpectations(t)
	})

	t.Run("Balance not found", func(t *testing.T) {
		mockDS := new(mocks.MockDataSource)
		balanceID := "bln_nonexistent"
		identityID := "idt_456"

		mockDS.On("GetIdentityByID", identityID).Return(&model.Identity{IdentityID: identityID}, nil)
		mockDS.On("GetBalanceByIDLite", balanceID).Return((*model.Balance)(nil), fmt.Errorf("not found"))

		blnk := &Blnk{datasource: mockDS, config: cnf}
		err := blnk.UpdateBalanceIdentity(balanceID, identityID)

		assert.Error(t, err)
		assert.Contains(t, err.Error(), "balance validation failed")
		mockDS.AssertExpectations(t)
	})

	t.Run("Update fails", func(t *testing.T) {
		mockDS := new(mocks.MockDataSource)
		balanceID := "bln_123"
		identityID := "idt_456"

		mockDS.On("GetIdentityByID", identityID).Return(&model.Identity{IdentityID: identityID}, nil)
		mockDS.On("GetBalanceByIDLite", balanceID).Return(&model.Balance{BalanceID: balanceID}, nil)
		mockDS.On("UpdateBalanceIdentity", balanceID, identityID).Return(fmt.Errorf("update failed"))

		blnk := &Blnk{datasource: mockDS, config: cnf}
		err := blnk.UpdateBalanceIdentity(balanceID, identityID)

		assert.Error(t, err)
		mockDS.AssertExpectations(t)
	})
}

func TestGetInflightTransactionsByParentID_Mock(t *testing.T) {
	cnf := &config.Configuration{
		Redis: config.RedisConfig{Dns: "localhost:6379"},
		Queue: config.QueueConfig{
			WebhookQueue:     "webhook_queue",
			TransactionQueue: "transaction_queue",
			NumberOfQueues:   1,
		},
	}
	config.ConfigStore.Store(cnf)

	t.Run("Success with transactions", func(t *testing.T) {
		mockDS := new(mocks.MockDataSource)
		parentID := "bulk_parent_123"
		expectedTxns := []*model.Transaction{
			{TransactionID: "txn_child_1", ParentTransaction: parentID, Status: StatusInflight},
			{TransactionID: "txn_child_2", ParentTransaction: parentID, Status: StatusInflight},
		}

		mockDS.On("GetInflightTransactionsByParentID", mock.Anything, parentID, 100, int64(0)).Return(expectedTxns, nil)

		blnk := &Blnk{datasource: mockDS, config: cnf}
		result, err := blnk.GetInflightTransactionsByParentID(context.Background(), parentID, 100, 0)

		assert.NoError(t, err)
		assert.Len(t, result, 2)
		mockDS.AssertExpectations(t)
	})

	t.Run("Empty result", func(t *testing.T) {
		mockDS := new(mocks.MockDataSource)
		parentID := "bulk_no_inflight"

		mockDS.On("GetInflightTransactionsByParentID", mock.Anything, parentID, 100, int64(0)).Return([]*model.Transaction{}, nil)

		blnk := &Blnk{datasource: mockDS, config: cnf}
		result, err := blnk.GetInflightTransactionsByParentID(context.Background(), parentID, 100, 0)

		assert.NoError(t, err)
		assert.Len(t, result, 0)
		mockDS.AssertExpectations(t)
	})
}

func TestGetBalanceByID_Mock(t *testing.T) {
	cnf := &config.Configuration{
		Redis: config.RedisConfig{Dns: "localhost:6379"},
		Queue: config.QueueConfig{
			WebhookQueue:     "webhook_queue",
			TransactionQueue: "transaction_queue",
			NumberOfQueues:   1,
		},
	}
	config.ConfigStore.Store(cnf)

	t.Run("Success", func(t *testing.T) {
		mockDS := new(mocks.MockDataSource)
		balanceID := "bln_test_123"
		expectedBalance := &model.Balance{
			BalanceID: balanceID,
			Currency:  "USD",
			Balance:   big.NewInt(10000),
		}

		mockDS.On("GetBalanceByID", balanceID, []string(nil), false).Return(expectedBalance, nil)

		blnk := &Blnk{datasource: mockDS, config: cnf}
		result, err := blnk.GetBalanceByID(context.Background(), balanceID, nil, false)

		assert.NoError(t, err)
		assert.NotNil(t, result)
		assert.Equal(t, balanceID, result.BalanceID)
		mockDS.AssertExpectations(t)
	})

	t.Run("Not found", func(t *testing.T) {
		mockDS := new(mocks.MockDataSource)
		balanceID := "bln_nonexistent"

		mockDS.On("GetBalanceByID", balanceID, []string(nil), false).Return((*model.Balance)(nil), fmt.Errorf("not found"))

		blnk := &Blnk{datasource: mockDS, config: cnf}
		result, err := blnk.GetBalanceByID(context.Background(), balanceID, nil, false)

		assert.Error(t, err)
		assert.Nil(t, result)
		mockDS.AssertExpectations(t)
	})
}

func TestGetAllBalances_Mock(t *testing.T) {
	cnf := &config.Configuration{
		Redis: config.RedisConfig{Dns: "localhost:6379"},
		Queue: config.QueueConfig{
			WebhookQueue:     "webhook_queue",
			TransactionQueue: "transaction_queue",
			NumberOfQueues:   1,
		},
	}
	config.ConfigStore.Store(cnf)

	t.Run("Success", func(t *testing.T) {
		mockDS := new(mocks.MockDataSource)
		expectedBalances := []model.Balance{
			{BalanceID: "bln_1", Currency: "USD"},
			{BalanceID: "bln_2", Currency: "EUR"},
		}

		mockDS.On("GetAllBalances", 10, 0).Return(expectedBalances, nil)

		blnk := &Blnk{datasource: mockDS, config: cnf}
		result, err := blnk.GetAllBalances(context.Background(), 10, 0)

		assert.NoError(t, err)
		assert.Len(t, result, 2)
		mockDS.AssertExpectations(t)
	})
}

func TestCreateAPIKey_Mock(t *testing.T) {
	cnf := &config.Configuration{
		Redis: config.RedisConfig{Dns: "localhost:6379"},
		Queue: config.QueueConfig{
			WebhookQueue:     "webhook_queue",
			TransactionQueue: "transaction_queue",
			NumberOfQueues:   1,
		},
	}
	config.ConfigStore.Store(cnf)

	t.Run("Success", func(t *testing.T) {
		mockDS := new(mocks.MockDataSource)
		name := "Test Key"
		ownerID := "owner_123"
		scopes := []string{"read", "write"}
		expiresAt := time.Now().Add(24 * time.Hour)

		expectedKey := &model.APIKey{
			APIKeyID:  "api_key_123",
			Name:      name,
			OwnerID:   ownerID,
			Scopes:    scopes,
			ExpiresAt: expiresAt,
		}

		mockDS.On("CreateAPIKey", mock.Anything, name, ownerID, scopes, expiresAt).Return(expectedKey, nil)

		blnk := &Blnk{datasource: mockDS, config: cnf}
		result, err := blnk.CreateAPIKey(context.Background(), name, ownerID, scopes, expiresAt)

		assert.NoError(t, err)
		assert.NotNil(t, result)
		assert.Equal(t, name, result.Name)
		mockDS.AssertExpectations(t)
	})

	t.Run("Error", func(t *testing.T) {
		mockDS := new(mocks.MockDataSource)
		mockDS.On("CreateAPIKey", mock.Anything, mock.Anything, mock.Anything, mock.Anything, mock.Anything).
			Return((*model.APIKey)(nil), fmt.Errorf("creation failed"))

		blnk := &Blnk{datasource: mockDS, config: cnf}
		result, err := blnk.CreateAPIKey(context.Background(), "name", "owner", []string{}, time.Now())

		assert.Error(t, err)
		assert.Nil(t, result)
		mockDS.AssertExpectations(t)
	})
}

func TestListAPIKeys_Mock(t *testing.T) {
	cnf := &config.Configuration{
		Redis: config.RedisConfig{Dns: "localhost:6379"},
		Queue: config.QueueConfig{
			WebhookQueue:     "webhook_queue",
			TransactionQueue: "transaction_queue",
			NumberOfQueues:   1,
		},
	}
	config.ConfigStore.Store(cnf)

	t.Run("Success", func(t *testing.T) {
		mockDS := new(mocks.MockDataSource)
		ownerID := "owner_123"
		expectedKeys := []*model.APIKey{
			{APIKeyID: "key_1", Name: "Key 1"},
			{APIKeyID: "key_2", Name: "Key 2"},
		}

		mockDS.On("ListAPIKeys", mock.Anything, ownerID).Return(expectedKeys, nil)

		blnk := &Blnk{datasource: mockDS, config: cnf}
		result, err := blnk.ListAPIKeys(context.Background(), ownerID)

		assert.NoError(t, err)
		assert.Len(t, result, 2)
		mockDS.AssertExpectations(t)
	})

	t.Run("Empty result", func(t *testing.T) {
		mockDS := new(mocks.MockDataSource)
		mockDS.On("ListAPIKeys", mock.Anything, "owner_empty").Return([]*model.APIKey{}, nil)

		blnk := &Blnk{datasource: mockDS, config: cnf}
		result, err := blnk.ListAPIKeys(context.Background(), "owner_empty")

		assert.NoError(t, err)
		assert.Len(t, result, 0)
		mockDS.AssertExpectations(t)
	})
}

func TestRevokeAPIKey_Mock(t *testing.T) {
	cnf := &config.Configuration{
		Redis: config.RedisConfig{Dns: "localhost:6379"},
		Queue: config.QueueConfig{
			WebhookQueue:     "webhook_queue",
			TransactionQueue: "transaction_queue",
			NumberOfQueues:   1,
		},
	}
	config.ConfigStore.Store(cnf)

	t.Run("Success", func(t *testing.T) {
		mockDS := new(mocks.MockDataSource)
		keyID := "key_123"
		ownerID := "owner_123"

		mockDS.On("RevokeAPIKey", mock.Anything, keyID, ownerID).Return(nil)

		blnk := &Blnk{datasource: mockDS, config: cnf}
		err := blnk.RevokeAPIKey(context.Background(), keyID, ownerID)

		assert.NoError(t, err)
		mockDS.AssertExpectations(t)
	})

	t.Run("Key not found", func(t *testing.T) {
		mockDS := new(mocks.MockDataSource)
		mockDS.On("RevokeAPIKey", mock.Anything, "nonexistent", "owner").Return(fmt.Errorf("not found"))

		blnk := &Blnk{datasource: mockDS, config: cnf}
		err := blnk.RevokeAPIKey(context.Background(), "nonexistent", "owner")

		assert.Error(t, err)
		mockDS.AssertExpectations(t)
	})
}

func TestGetAPIKeyByKey_Mock(t *testing.T) {
	cnf := &config.Configuration{
		Redis: config.RedisConfig{Dns: "localhost:6379"},
		Queue: config.QueueConfig{
			WebhookQueue:     "webhook_queue",
			TransactionQueue: "transaction_queue",
			NumberOfQueues:   1,
		},
	}
	config.ConfigStore.Store(cnf)

	t.Run("Success", func(t *testing.T) {
		mockDS := new(mocks.MockDataSource)
		keyString := "abc123xyz"
		expectedKey := &model.APIKey{
			APIKeyID: "key_123",
			Key:      keyString,
			Name:     "Test Key",
		}

		mockDS.On("GetAPIKey", mock.Anything, keyString).Return(expectedKey, nil)

		blnk := &Blnk{datasource: mockDS, config: cnf}
		result, err := blnk.GetAPIKeyByKey(context.Background(), keyString)

		assert.NoError(t, err)
		assert.NotNil(t, result)
		assert.Equal(t, keyString, result.Key)
		mockDS.AssertExpectations(t)
	})

	t.Run("Not found", func(t *testing.T) {
		mockDS := new(mocks.MockDataSource)
		mockDS.On("GetAPIKey", mock.Anything, "invalid").Return((*model.APIKey)(nil), fmt.Errorf("not found"))

		blnk := &Blnk{datasource: mockDS, config: cnf}
		result, err := blnk.GetAPIKeyByKey(context.Background(), "invalid")

		assert.Error(t, err)
		assert.Nil(t, result)
		mockDS.AssertExpectations(t)
	})
}

func TestUpdateLastUsed_Mock(t *testing.T) {
	cnf := &config.Configuration{
		Redis: config.RedisConfig{Dns: "localhost:6379"},
		Queue: config.QueueConfig{
			WebhookQueue:     "webhook_queue",
			TransactionQueue: "transaction_queue",
			NumberOfQueues:   1,
		},
	}
	config.ConfigStore.Store(cnf)

	t.Run("Success", func(t *testing.T) {
		mockDS := new(mocks.MockDataSource)
		keyID := "key_123"

		mockDS.On("UpdateLastUsed", mock.Anything, keyID).Return(nil)

		blnk := &Blnk{datasource: mockDS, config: cnf}
		err := blnk.UpdateLastUsed(context.Background(), keyID)

		assert.NoError(t, err)
		mockDS.AssertExpectations(t)
	})

	t.Run("Error", func(t *testing.T) {
		mockDS := new(mocks.MockDataSource)
		mockDS.On("UpdateLastUsed", mock.Anything, "invalid").Return(fmt.Errorf("update failed"))

		blnk := &Blnk{datasource: mockDS, config: cnf}
		err := blnk.UpdateLastUsed(context.Background(), "invalid")

		assert.Error(t, err)
		mockDS.AssertExpectations(t)
	})
}

func TestGetReconciliation_Mock(t *testing.T) {
	cnf := &config.Configuration{
		Redis: config.RedisConfig{Dns: "localhost:6379"},
		Queue: config.QueueConfig{
			WebhookQueue:     "webhook_queue",
			TransactionQueue: "transaction_queue",
			NumberOfQueues:   1,
		},
	}
	config.ConfigStore.Store(cnf)

	t.Run("Success", func(t *testing.T) {
		mockDS := new(mocks.MockDataSource)
		reconciliationID := "recon_123"
		expected := &model.Reconciliation{
			ReconciliationID: reconciliationID,
			Status:           "completed",
		}

		mockDS.On("GetReconciliation", mock.Anything, reconciliationID).Return(expected, nil)

		blnk := &Blnk{datasource: mockDS, config: cnf}
		result, err := blnk.GetReconciliation(context.Background(), reconciliationID)

		assert.NoError(t, err)
		assert.NotNil(t, result)
		assert.Equal(t, reconciliationID, result.ReconciliationID)
		mockDS.AssertExpectations(t)
	})

	t.Run("Not found", func(t *testing.T) {
		mockDS := new(mocks.MockDataSource)
		mockDS.On("GetReconciliation", mock.Anything, "nonexistent").Return((*model.Reconciliation)(nil), fmt.Errorf("not found"))

		blnk := &Blnk{datasource: mockDS, config: cnf}
		result, err := blnk.GetReconciliation(context.Background(), "nonexistent")

		assert.Error(t, err)
		assert.Nil(t, result)
		mockDS.AssertExpectations(t)
	})
}

func TestGetBalanceAtTime_Mock(t *testing.T) {
	cnf := &config.Configuration{
		Redis: config.RedisConfig{Dns: "localhost:6379"},
		Queue: config.QueueConfig{
			WebhookQueue:     "webhook_queue",
			TransactionQueue: "transaction_queue",
			NumberOfQueues:   1,
		},
	}
	config.ConfigStore.Store(cnf)

	t.Run("Success", func(t *testing.T) {
		mockDS := new(mocks.MockDataSource)
		balanceID := "bln_123"
		targetTime := time.Now().Add(-24 * time.Hour)
		expected := &model.Balance{
			BalanceID: balanceID,
			Balance:   big.NewInt(5000),
		}

		mockDS.On("GetBalanceAtTime", mock.Anything, balanceID, targetTime, false).Return(expected, nil)

		blnk := &Blnk{datasource: mockDS, config: cnf}
		result, err := blnk.GetBalanceAtTime(context.Background(), balanceID, targetTime, false)

		assert.NoError(t, err)
		assert.NotNil(t, result)
		assert.Equal(t, balanceID, result.BalanceID)
		mockDS.AssertExpectations(t)
	})

	t.Run("Error", func(t *testing.T) {
		mockDS := new(mocks.MockDataSource)
		targetTime := time.Now()
		mockDS.On("GetBalanceAtTime", mock.Anything, "invalid", targetTime, false).Return((*model.Balance)(nil), fmt.Errorf("not found"))

		blnk := &Blnk{datasource: mockDS, config: cnf}
		result, err := blnk.GetBalanceAtTime(context.Background(), "invalid", targetTime, false)

		assert.Error(t, err)
		assert.Nil(t, result)
		mockDS.AssertExpectations(t)
	})
}