Functional tests (#543)

[bytom/bytom.git] / wallet / wallet_test.go

diff --git a/wallet/wallet_test.go b/wallet/wallet_test.go
old mode 100755 (executable)
new mode 100644 (file)
index 05138c4..8a6d218
--- a/wallet/wallet_test.go
+++ b/wallet/wallet_test.go
@@ -14,7 +14,6 @@ import (
        "github.com/bytom/asset"
        "github.com/bytom/blockchain/pseudohsm"
        "github.com/bytom/blockchain/txbuilder"
-       cfg "github.com/bytom/config"
        "github.com/bytom/consensus"
        "github.com/bytom/crypto/ed25519/chainkd"
        "github.com/bytom/crypto/sha3pool"
@@ -37,7 +36,7 @@ func TestWalletUpdate(t *testing.T) {
        store := leveldb.NewStore(testDB)
        txPool := protocol.NewTxPool()
 
-       chain, err := protocol.NewChain(bc.Hash{}, store, txPool)
+       chain, err := protocol.NewChain(store, txPool)
        if err != nil {
                t.Fatal(err)
        }
@@ -80,9 +79,9 @@ func TestWalletUpdate(t *testing.T) {
        block := mockSingleBlock(tx)
 
        txStatus := bc.NewTransactionStatus()
-       store.SaveBlock(block, txStatus, consensus.InitialSeed)
+       store.SaveBlock(block, txStatus)
 
-       err = w.attachBlock(block)
+       err = w.AttachBlock(block)
        if err != nil {
                t.Fatal(err)
        }
@@ -111,20 +110,11 @@ func TestExportAndImportPrivKey(t *testing.T) {
        store := leveldb.NewStore(testDB)
        txPool := protocol.NewTxPool()
 
-       chain, err := protocol.NewChain(bc.Hash{}, store, txPool)
+       chain, err := protocol.NewChain(store, txPool)
        if err != nil {
                t.Fatal(err)
        }
 
-       genesisBlock := cfg.GenerateGenesisBlock()
-
-       if err = chain.SaveBlock(genesisBlock); err != nil {
-               t.Fatal(err)
-       }
-       if err = chain.ConnectBlock(genesisBlock); err != nil {
-               t.Fatal(err)
-       }
-
        acntManager := account.NewManager(testDB, chain)
        reg := asset.NewRegistry(testDB, chain)