Add WalletBalanceService

Author: XZB-1248

v2/client.go

@@ -769,6 +769,10 @@ func (c *Client) NewGetAssetDetailService() *GetAssetDetailService {
 	return &GetAssetDetailService{c: c}
 }
 
+func (c *Client) NewWalletBalanceService() *WalletBalanceService {
+	return &WalletBalanceService{c: c}
+}
+
 // NewAveragePriceService init average price service
 func (c *Client) NewAveragePriceService() *AveragePriceService {
 	return &AveragePriceService{c: c}

v2/wallet_balance_service.go

@@ -0,0 +1,51 @@
+package binance
+
+import (
+	"context"
+	"encoding/json"
+	"net/http"
+)
+
+// WalletBalanceService fetches all user wallet balance.
+//
+// See https://developers.binance.com/docs/wallet/asset/query-user-wallet-balance
+type WalletBalanceService struct {
+	c          *Client
+	quoteAsset *string
+}
+
+// QuoteAsset sets the quoteAsset parameter.
+func (s *WalletBalanceService) QuoteAsset(asset string) *WalletBalanceService {
+	s.quoteAsset = &asset
+	return s
+}
+
+// Do sends the request.
+func (s *WalletBalanceService) Do(ctx context.Context, opts ...RequestOption) (res []*WalletBalance, err error) {
+	r := &request{
+		method:   http.MethodGet,
+		endpoint: "/sapi/v1/asset/wallet/balance",
+		secType:  secTypeSigned,
+	}
+	if s.quoteAsset != nil {
+		r.setParam("quoteAsset", *s.quoteAsset)
+	}
+
+	data, err := s.c.callAPI(ctx, r, opts...)
+	if err != nil {
+		return nil, err
+	}
+	res = make([]*WalletBalance, 0)
+	err = json.Unmarshal(data, &res)
+	if err != nil {
+		return nil, err
+	}
+	return res, nil
+}
+
+// WalletBalanceResponse defines the response of WalletBalanceService
+type WalletBalance struct {
+	Activate   bool   `json:"activate"`
+	Balance    string `json:"balance"`
+	WalletName string `json:"walletName"`
+}

v2/wallet_balance_service_test.go

@@ -0,0 +1,150 @@
+package binance
+
+import (
+	"testing"
+
+	"github.com/stretchr/testify/suite"
+)
+
+type walletBalanceServiceTestSuite struct {
+	baseTestSuite
+}
+
+func TestWalletBalanceService(t *testing.T) {
+	suite.Run(t, new(walletBalanceServiceTestSuite))
+}
+
+func (s *walletBalanceServiceTestSuite) TestWalletBalance() {
+	data := []byte(`[
+	{
+		"activate": true,
+		"balance": "0.00000551",
+		"walletName": "Spot"
+	},
+	{
+		"activate": true,
+		"balance": "0",
+		"walletName": "Funding"
+	},
+	{
+		"activate": true,
+		"balance": "0",
+		"walletName": "Cross Margin"
+	},
+	{
+		"activate": true,
+		"balance": "0",
+		"walletName": "Isolated Margin"
+	},
+	{
+		"activate": true,
+		"balance": "0",
+		"walletName": "USDⓈ-M Futures"
+	},
+	{
+		"activate": true,
+		"balance": "0",
+		"walletName": "COIN-M Futures"
+	},
+	{
+		"activate": true,
+		"balance": "0",
+		"walletName": "Earn"
+	},
+	{
+		"activate": true,
+		"balance": "0",
+		"walletName": "Options"
+	},
+	{
+		"activate": true,
+		"balance": "0",
+		"walletName": "Trading Bots"
+	},
+	{
+		"activate": true,
+		"balance": "0.01630125",
+		"walletName": "Copy Trading"
+	}
+]`)
+	s.mockDo(data, nil)
+	defer s.assertDo()
+
+	quoteAsset := "USDT"
+
+	s.assertReq(func(r *request) {
+		e := newSignedRequest().setParam("quoteAsset", quoteAsset)
+		s.assertRequestEqual(e, r)
+	})
+
+	res, err := s.client.NewWalletBalanceService().QuoteAsset(quoteAsset).Do(newContext())
+	s.r().NoError(err)
+
+	e := []*WalletBalance{
+		{
+			Activate:   true,
+			Balance:    "0.00000551",
+			WalletName: "Spot",
+		},
+		{
+			Activate:   true,
+			Balance:    "0",
+			WalletName: "Funding",
+		},
+		{
+			Activate:   true,
+			Balance:    "0",
+			WalletName: "Cross Margin",
+		},
+		{
+			Activate:   true,
+			Balance:    "0",
+			WalletName: "Isolated Margin",
+		},
+		{
+			Activate:   true,
+			Balance:    "0",
+			WalletName: "USDⓈ-M Futures",
+		},
+		{
+			Activate:   true,
+			Balance:    "0",
+			WalletName: "COIN-M Futures",
+		},
+		{
+			Activate:   true,
+			Balance:    "0",
+			WalletName: "Earn",
+		},
+		{
+			Activate:   true,
+			Balance:    "0",
+			WalletName: "Options",
+		},
+		{
+			Activate:   true,
+			Balance:    "0",
+			WalletName: "Trading Bots",
+		},
+		{
+			Activate:   true,
+			Balance:    "0.01630125",
+			WalletName: "Copy Trading",
+		},
+	}
+	s.assertWalletBalancesEqual(e, res)
+}
+
+func (s *walletBalanceServiceTestSuite) assertWalletBalanceEqual(e, a *WalletBalance) {
+	r := s.r()
+	r.Equal(e.Activate, a.Activate, "Activate")
+	r.Equal(e.Balance, a.Balance, "Balance")
+	r.Equal(e.WalletName, a.WalletName, "WalletName")
+}
+
+func (s *walletBalanceServiceTestSuite) assertWalletBalancesEqual(e, a []*WalletBalance) {
+	s.r().Len(e, len(a))
+	for i := range e {
+		s.assertWalletBalanceEqual(e[i], a[i])
+	}
+}