GetBalance

This commit is contained in:
arnaucube
2019-04-11 14:05:09 +02:00
parent a91060fa55
commit 85d0bcc5a3
2 changed files with 34 additions and 1 deletions

View File

@@ -2,6 +2,8 @@ package node
import (
"crypto/ecdsa"
"encoding/hex"
"fmt"
"io/ioutil"
"testing"
@@ -122,6 +124,7 @@ func TestFromGenesisToTenBlocks(t *testing.T) {
assert.Nil(t, err)
// create the genesis block
// genesisBlock sends 100 to pubK
genesisBlock, err := node.CreateGenesis(&privK.PublicKey, uint64(100))
assert.Nil(t, err)
assert.NotEqual(t, genesisBlock.Signature, core.Signature{})
@@ -169,6 +172,12 @@ func TestFromGenesisToTenBlocks(t *testing.T) {
err = node.Bc.AddBlock(block)
assert.Nil(t, err)
balance, err := node.Bc.GetBalance(&pubK0)
assert.Nil(t, err)
fmt.Println(hex.EncodeToString(core.PackPubK(&pubK0)[:10]))
fmt.Println("balance in pubK0", balance)
assert.Equal(t, balance, uint64(100))
// add another tx sending coins to the pubK1
privK1, err := core.NewKey()
assert.Nil(t, err)
@@ -199,4 +208,10 @@ func TestFromGenesisToTenBlocks(t *testing.T) {
assert.Nil(t, err)
err = node.Bc.AddBlock(block)
assert.Nil(t, err)
balance, err = node.Bc.GetBalance(&pubK0)
assert.Nil(t, err)
fmt.Println(hex.EncodeToString(core.PackPubK(&pubK0)[:10]))
fmt.Println("balance in pubK0", balance)
assert.Equal(t, balance, uint64(90))
}