test: cosmosclient.BankBalances

Author: tbruyelle

ignite/pkg/cosmosclient/bank.go

@@ -18,7 +18,7 @@ func (c Client) BankBalances(ctx context.Context, address string, pagination *qu
 		Pagination: pagination,
 	}
 
-	resp, err := banktypes.NewQueryClient(c.context).AllBalances(ctx, req)
+	resp, err := c.bankQueryClient.AllBalances(ctx, req)
 	if err != nil {
 		return nil, err
 	}

ignite/pkg/cosmosclient/bank_test.go

@@ -0,0 +1,41 @@
+package cosmosclient
+
+import (
+	"context"
+	"testing"
+
+	"cosmossdk.io/math"
+	sdk "github.com/cosmos/cosmos-sdk/types"
+	"github.com/cosmos/cosmos-sdk/types/query"
+	banktypes "github.com/cosmos/cosmos-sdk/x/bank/types"
+	"github.com/stretchr/testify/assert"
+	"github.com/stretchr/testify/require"
+)
+
+func TestClientBankBalances(t *testing.T) {
+	var (
+		ctx              = context.Background()
+		address          = "address"
+		pagination       = &query.PageRequest{Offset: 1}
+		expectedBalances = sdk.NewCoins(
+			sdk.NewCoin("token", math.NewInt(1000)),
+			sdk.NewCoin("stake", math.NewInt(2000)),
+		)
+	)
+	c := newClient(t, func(s suite) {
+		req := &banktypes.QueryAllBalancesRequest{
+			Address:    address,
+			Pagination: pagination,
+		}
+
+		s.bankQueryClient.EXPECT().AllBalances(ctx, req).
+			Return(&banktypes.QueryAllBalancesResponse{
+				Balances: expectedBalances,
+			}, nil)
+	})
+
+	balances, err := c.BankBalances(ctx, address, pagination)
+
+	require.NoError(t, err)
+	assert.Equal(t, expectedBalances, balances)
+}