account_stakers_query_test.go

package hedera

import (
	"fmt"
	"github.com/stretchr/testify/assert"
	"strings"
	"testing"
)

func TestSerializeStakersQuery(t *testing.T) {
	query := NewAccountStakersQuery().
		SetAccountID(AccountID{Account: 3}).
		Query

	assert.Equal(t, `cryptoGetProxyStakers:{header:{}accountID:{accountNum:3}}`, strings.ReplaceAll(query.pb.String(), " ", ""))
}

func TestAccountStakersQuery_Execute(t *testing.T) {
	client := newTestClient(t, false)

	_, err := NewAccountStakersQuery().
		SetAccountID(client.GetOperatorAccountID()).
		SetMaxQueryPayment(NewHbar(1)).
		SetQueryPayment(HbarFromTinybar(25)).
		Execute(client)
	assert.Error(t, err)
}

func TestAccountStakersQueryCost_Execute(t *testing.T) {
	client := newTestClient(t, false)

	accountStakers := NewAccountStakersQuery().
		SetMaxQueryPayment(NewHbar(1)).
		SetAccountID(client.GetOperatorAccountID())

	cost, err := accountStakers.GetCost(client)
	assert.Error(t, err)

	_, err = accountStakers.SetQueryPayment(cost).Execute(client)
	assert.Error(t, err)
}

func TestAccountStakersQueryCost_BigMax_Execute(t *testing.T) {
	client := newTestClient(t, false)

	accountStakers := NewAccountStakersQuery().
		SetMaxQueryPayment(NewHbar(100000)).
		SetAccountID(client.GetOperatorAccountID())

	cost, err := accountStakers.GetCost(client)
	assert.Error(t, err)

	_, err = accountStakers.SetQueryPayment(cost).Execute(client)
	assert.Error(t, err)
}

func TestAccountStakersQueryCost_SmallMax_Execute(t *testing.T) {
	client := newTestClient(t, false)

	accountStakers := NewAccountStakersQuery().
		SetMaxQueryPayment(HbarFromTinybar(25)).
		SetAccountID(client.GetOperatorAccountID())

	cost, err := accountStakers.GetCost(client)
	assert.Error(t, err)

	_, err = accountStakers.SetQueryPayment(cost).Execute(client)
	assert.Error(t, err)
}

func TestAccountStakersQueryCost_InsufficientFee_Execute(t *testing.T) {
	client := newTestClient(t, false)

	accountStakers := NewAccountStakersQuery().
		SetMaxQueryPayment(NewHbar(1)).
		SetAccountID(client.GetOperatorAccountID())

	_, err := accountStakers.GetCost(client)
	assert.Error(t, err)

	_, err = accountStakers.SetQueryPayment(HbarFromTinybar(1)).Execute(client)
	if err != nil {
		assert.Equal(t, fmt.Sprintf("exceptional precheck status NOT_SUPPORTED"), err.Error())
	}
}

func TestAccountStakersNoAccountID_Execute(t *testing.T) {
	client := newTestClient(t, false)

	_, err := NewAccountStakersQuery().
		Execute(client)
	assert.Error(t, err)
	if err != nil {
		assert.Equal(t, fmt.Sprintf("exceptional precheck status NOT_SUPPORTED"), err.Error())
	}
}