Function TestGetBalanceByIndicator_Success

database/balance_test.go:917–954

(t *testing.T)

}

func TestGetBalanceByIndicator_Success(t *testing.T) {
	db, mock, err := sqlmock.New()
	assert.NoError(t, err)
	defer func() { _ = db.Close() }()

	ds := Datasource{Conn: db}

	query := `
       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 ledgerforge.balances
       WHERE indicator = $1 AND currency = $2
    `

	mock.ExpectQuery(regexp.QuoteMeta(query)).
		WithArgs("ACC001", "USD").
		WillReturnRows(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(
			"bln_123", "ACC001", "USD", 100, "ldg_001",
			"100000", "60000", "40000",
			"5000", "3000", "2000",
			time.Now(), 1, false, "FIFO", "",
		))

	balance, err := ds.GetBalanceByIndicator("ACC001", "USD")
	assert.NoError(t, err)
	assert.NotNil(t, balance)
	assert.Equal(t, "bln_123", balance.BalanceID)
	assert.Equal(t, "ACC001", balance.Indicator)
	assert.Equal(t, "USD", balance.Currency)
	assert.Equal(t, "100000", balance.Balance.String())

	err = mock.ExpectationsWereMet()
	assert.NoError(t, err)
}

func TestGetBalanceByIndicator_NotFound(t *testing.T) {
	db, mock, err := sqlmock.New()