You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
134 lines
3.7 KiB
134 lines
3.7 KiB
package client
|
|
|
|
import (
|
|
"bytes"
|
|
"math/big"
|
|
"strings"
|
|
"testing"
|
|
|
|
"github.com/ethereum/go-ethereum/common"
|
|
"github.com/ethereum/go-ethereum/crypto"
|
|
client "github.com/harmony-one/harmony/api/client/service/proto"
|
|
proto "github.com/harmony-one/harmony/api/client/service/proto"
|
|
"github.com/harmony-one/harmony/core/state"
|
|
common2 "github.com/harmony-one/harmony/internal/common"
|
|
|
|
"github.com/ethereum/go-ethereum/ethdb"
|
|
"github.com/ethereum/go-ethereum/params"
|
|
"github.com/harmony-one/harmony/consensus"
|
|
"github.com/harmony-one/harmony/core"
|
|
"github.com/harmony-one/harmony/core/vm"
|
|
)
|
|
|
|
var (
|
|
// Test accounts
|
|
testBankKey, _ = crypto.GenerateKey()
|
|
testBankAddress = crypto.PubkeyToAddress(testBankKey.PublicKey)
|
|
testBankFunds = big.NewInt(8000000000000000000)
|
|
|
|
chainConfig = params.TestChainConfig
|
|
)
|
|
|
|
func TestGetFreeToken(test *testing.T) {
|
|
hash := common.Hash{}
|
|
hash.SetBytes([]byte("hello"))
|
|
server := NewServer(func() (*state.DB, error) {
|
|
return nil, nil
|
|
}, func(common.Address) common.Hash {
|
|
return hash
|
|
}, nil)
|
|
|
|
testBankKey, _ := crypto.GenerateKey()
|
|
testBankAddress := crypto.PubkeyToAddress(testBankKey.PublicKey)
|
|
response, err := server.GetFreeToken(nil, &client.GetFreeTokenRequest{Address: testBankAddress.Bytes()})
|
|
|
|
if err != nil {
|
|
test.Errorf("Failed to get free token")
|
|
}
|
|
if bytes.Compare(response.TxId, hash.Bytes()) != 0 {
|
|
test.Errorf("Wrong transaction id is returned")
|
|
}
|
|
}
|
|
|
|
func TestFetchAccountState(test *testing.T) {
|
|
var (
|
|
database = ethdb.NewMemDatabase()
|
|
gspec = core.Genesis{
|
|
Config: chainConfig,
|
|
Alloc: core.GenesisAlloc{testBankAddress: {Balance: testBankFunds}},
|
|
ShardID: 10,
|
|
}
|
|
)
|
|
|
|
genesis := gspec.MustCommit(database)
|
|
_ = genesis
|
|
chain, _ := core.NewBlockChain(database, nil, gspec.Config, consensus.NewFaker(), vm.Config{}, nil)
|
|
|
|
hash := common.Hash{}
|
|
hash.SetBytes([]byte("hello"))
|
|
server := NewServer(func() (*state.DB, error) {
|
|
return chain.State()
|
|
}, func(common.Address) common.Hash {
|
|
return hash
|
|
}, nil)
|
|
|
|
response, err := server.FetchAccountState(nil, &client.FetchAccountStateRequest{Address: testBankAddress.Bytes()})
|
|
|
|
if err != nil {
|
|
test.Errorf("Failed to get free token")
|
|
}
|
|
|
|
if bytes.Compare(response.Balance, testBankFunds.Bytes()) != 0 {
|
|
test.Errorf("Wrong balance is returned")
|
|
}
|
|
|
|
if response.Nonce != 0 {
|
|
test.Errorf("Wrong nonce is returned")
|
|
}
|
|
}
|
|
|
|
func TestGetStakingContractInfo(test *testing.T) {
|
|
var (
|
|
database = ethdb.NewMemDatabase()
|
|
gspec = core.Genesis{
|
|
Config: chainConfig,
|
|
Alloc: core.GenesisAlloc{testBankAddress: {Balance: testBankFunds}},
|
|
ShardID: 10,
|
|
}
|
|
)
|
|
|
|
genesis := gspec.MustCommit(database)
|
|
_ = genesis
|
|
chain, _ := core.NewBlockChain(database, nil, gspec.Config, consensus.NewFaker(), vm.Config{}, nil)
|
|
|
|
hash := common.Hash{}
|
|
hash.SetBytes([]byte("hello"))
|
|
deployedStakingContractAddress := common.Address{}
|
|
deployedStakingContractAddress.SetBytes([]byte("stakingContractAddress"))
|
|
server := NewServer(func() (*state.DB, error) {
|
|
return chain.State()
|
|
}, func(common.Address) common.Hash {
|
|
return hash
|
|
}, func() common.Address {
|
|
return deployedStakingContractAddress
|
|
})
|
|
|
|
response, err := server.GetStakingContractInfo(nil, &proto.StakingContractInfoRequest{Address: testBankAddress.Bytes()})
|
|
if err != nil {
|
|
test.Fatal("GetStakingContractInfo failed:", err)
|
|
}
|
|
|
|
if bytes.Compare(response.Balance, testBankFunds.Bytes()) != 0 {
|
|
test.Errorf("Wrong balance is returned")
|
|
}
|
|
|
|
if strings.Compare(response.ContractAddress, common2.MustAddressToBech32(deployedStakingContractAddress)) != 0 {
|
|
test.Errorf("Wrong ContractAddress is returned (expected %#v, got %#v)",
|
|
common2.MustAddressToBech32(deployedStakingContractAddress),
|
|
response.ContractAddress)
|
|
}
|
|
|
|
if response.Nonce != 0 {
|
|
test.Errorf("Wrong nonce is returned")
|
|
}
|
|
}
|
|
|