Function TestGetBalanceByID_Mock

transaction_test.go:6584–6628

(t *testing.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)

		ledgerforge := &LedgerForge{datasource: mockDS, config: cnf}
		result, err := ledgerforge.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"))

		ledgerforge := &LedgerForge{datasource: mockDS, config: cnf}
		result, err := ledgerforge.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{