forked from TrueCloudLab/neoneo-go
f3f6662fc9
* Initial draft of the neo-go wallet * Cleanup + more test for util package * integrated wallet into neo-cli partially * base wallet implementation + smartcontract code.
47 lines
1.5 KiB
Go
47 lines
1.5 KiB
Go
package core
|
|
|
|
import (
|
|
"log"
|
|
"os"
|
|
"testing"
|
|
|
|
"github.com/CityOfZion/neo-go/pkg/util"
|
|
)
|
|
|
|
func TestNewBlockchain(t *testing.T) {
|
|
startHash, _ := util.Uint256DecodeString("996e37358dc369912041f966f8c5d8d3a8255ba5dcbd3447f8a82b55db869099")
|
|
bc := NewBlockchain(nil, nil, startHash)
|
|
|
|
want := uint32(0)
|
|
if have := bc.BlockHeight(); want != have {
|
|
t.Fatalf("expected %d got %d", want, have)
|
|
}
|
|
if have := bc.HeaderHeight(); want != have {
|
|
t.Fatalf("expected %d got %d", want, have)
|
|
}
|
|
if have := bc.storedHeaderCount; want != have {
|
|
t.Fatalf("expected %d got %d", want, have)
|
|
}
|
|
if !bc.CurrentBlockHash().Equals(startHash) {
|
|
t.Fatalf("expected current block hash to be %d got %s", startHash, bc.CurrentBlockHash())
|
|
}
|
|
}
|
|
|
|
func TestAddHeaders(t *testing.T) {
|
|
startHash, _ := util.Uint256DecodeString("996e37358dc369912041f966f8c5d8d3a8255ba5dcbd3447f8a82b55db869099")
|
|
bc := NewBlockchain(NewMemoryStore(), log.New(os.Stdout, "", 0), startHash)
|
|
|
|
h1 := &Header{BlockBase: BlockBase{Version: 0, Index: 1, Script: &Witness{}}}
|
|
h2 := &Header{BlockBase: BlockBase{Version: 0, Index: 2, Script: &Witness{}}}
|
|
h3 := &Header{BlockBase: BlockBase{Version: 0, Index: 3, Script: &Witness{}}}
|
|
|
|
if err := bc.AddHeaders(h1, h2, h3); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if want, have := h3.Index, bc.HeaderHeight(); want != have {
|
|
t.Fatalf("expected header height of %d got %d", want, have)
|
|
}
|
|
if want, have := uint32(0), bc.storedHeaderCount; want != have {
|
|
t.Fatalf("expected stored header count to be %d got %d", want, have)
|
|
}
|
|
}
|