account_test.go

package kumex

import (
	"testing"
)

func TestApiService_AccountOverview(t *testing.T) {

	s := NewApiServiceFromEnv()
	p := map[string]string{}
	p["currency"] = "USDT"

	rsp, err := s.AccountOverview(p)
	if err != nil {
		t.Fatal(err)
	}
	o := &AccountModel{}
	if err := rsp.ReadData(o); err != nil {
		t.Fatal(err)
	}
	t.Log(ToJsonString(o))
	switch {
	case o.AccountEquity == 0:
		t.Error("Empty key 'accountEquity'")
	case o.MarginBalance < 0:
		t.Error("Empty key 'marginBalance'")
	case o.PositionMargin < 0:
		t.Error("Empty key 'positionMargin'")
	case o.OrderMargin < 0:
		t.Error("Empty key 'orderMargin'")
	case o.FrozenFunds < 0:
		t.Error("Empty key 'frozenFunds'")
	case o.AvailableBalance <= 0:
		t.Error("Empty key 'availableBalance'")
	case o.Currency == "":
		t.Error("Empty key 'currency'")
	}
}

func TestApiService_TransactionHistory(t *testing.T) {
	t.SkipNow()
	s := NewApiServiceFromEnv()
	p := map[string]string{}
	pp := &PaginationParam{CurrentPage: 1, PageSize: 10}
	rsp, err := s.TransactionHistory(p, pp)
	if err != nil {
		t.Fatal(err)
	}
	ds := TransactionHistoryListModel{}
	if _, err := rsp.ReadPaginationData(&ds); err != nil {
		t.Fatal(err)
	}

	for _, d := range ds {
		t.Log(ToJsonString(d))
		switch {
		case d.Time == "":
			t.Error("Empty key 'time'")
		case d.Type == "":
			t.Error("Empty key 'type'")
		case d.Amount == "":
			t.Error("Empty key 'amount'")
		case d.Status == "":
			t.Error("Empty key 'status'")
		case d.Fee == "":
			t.Error("Empty key 'fee'")
		case d.AccountEquity == "":
			t.Error("Empty key 'accountEquity'")
		case d.Remarks == "":
			t.Error("Empty key 'remark'")
		case d.Offset == "":
			t.Error("Empty key 'Offset'")
		}
	}
}

func TestApiService_CreateSubApiKey(t *testing.T) {
	t.SkipNow()
	s := NewApiServiceFromEnv()
	p := map[string]string{
		"subName":    "TestSubAccount1Fen",
		"passphrase": "123abcABC",
		"remark":     "re",
	}
	rsp, err := s.CreateSubApiKey(p)
	if err != nil {
		t.Fatal(err)
	}
	w := &CreateSubApiKeyRes{}
	if err := rsp.ReadData(w); err != nil {
		t.Fatal(err)
	}
	t.Log(ToJsonString(w))
}

func TestApiService_SubApiKeys(t *testing.T) {
	s := NewApiServiceFromEnv()
	rsp, err := s.SubApiKeys("", "TestSubAccount1Fen")
	if err != nil {
		t.Fatal(err)
	}
	w := SubApiKeysModel{}
	if err := rsp.ReadData(&w); err != nil {
		t.Fatal(err)
	}
	for _, model := range w {
		t.Log(ToJsonString(model))
	}
}

func TestApiService_ModifySubApiKey(t *testing.T) {
	s := NewApiServiceFromEnv()
	p := map[string]string{
		"subName":    "TestSubAccount1Fen",
		"apiKey":     "649cfa412b8f770001f5eec1",
		"passphrase": "123abcABC",
		"permission": "Trade",
	}
	rsp, err := s.ModifySubApiKey(p)
	if err != nil {
		t.Fatal(err)
	}
	w := &ModifySubApiKeyRes{}
	if err := rsp.ReadData(w); err != nil {
		t.Fatal(err)
	}
	t.Log(ToJsonString(w))
}

func TestApiService_DeleteSubApiKey(t *testing.T) {
	s := NewApiServiceFromEnv()
	rsp, err := s.DeleteSubApiKey("649cfa412b8f770001f5eec1", "123abcABC", "TestSubAccount1Fen")
	if err != nil {
		t.Fatal(err)
	}
	w := &DeleteSubApiKeyRes{}
	if err := rsp.ReadData(w); err != nil {
		t.Fatal(err)
	}
	t.Log(ToJsonString(w))
}

func TestApiService_SubAccountsBalance(t *testing.T) {
	s := NewApiServiceFromEnv()
	rsp, err := s.SubAccountsBalance("USDT")
	if err != nil {
		t.Fatal(err)
	}
	w := SubAccountBalanceModel{}
	if err := rsp.ReadData(&w); err != nil {
		t.Fatal(err)
	}

	t.Log(ToJsonString(w.Summary))
	for _, account := range w.Accounts {
		t.Log(ToJsonString(account))
	}
}