Merge pull request #466 from Bytom/update-mov

[bytom/vapor.git] / test / utxo_view / utxo_view_test.go

diff --git a/test/utxo_view/utxo_view_test.go b/test/utxo_view/utxo_view_test.go
index ac7cfff..c3119e3 100644 (file)
--- a/test/utxo_view/utxo_view_test.go
+++ b/test/utxo_view/utxo_view_test.go
@@ -4,16 +4,16 @@ import (
        "os"
        "testing"
 
-       "github.com/vapor/testutil"
+       "github.com/bytom/vapor/testutil"
 
        "github.com/golang/protobuf/proto"
 
-       "github.com/vapor/database"
-       dbm "github.com/vapor/database/leveldb"
-       "github.com/vapor/database/storage"
-       "github.com/vapor/protocol/bc"
-       "github.com/vapor/protocol/bc/types"
-       "github.com/vapor/protocol/state"
+       "github.com/bytom/vapor/database"
+       dbm "github.com/bytom/vapor/database/leveldb"
+       "github.com/bytom/vapor/database/storage"
+       "github.com/bytom/vapor/protocol/bc"
+       "github.com/bytom/vapor/protocol/bc/types"
+       "github.com/bytom/vapor/protocol/state"
 )
 
 func TestAttachOrDetachBlocks(t *testing.T) {
@@ -404,7 +404,8 @@ func TestAttachOrDetachBlocks(t *testing.T) {
                        },
                },
        }
-       node := blockNode(types.MapBlock(&mockBlocks[0].Block).BlockHeader)
+
+       mockBlockHeader := &mockBlocks[0].Block.BlockHeader
        defer os.RemoveAll("temp")
        for index, c := range cases {
                testDB := dbm.NewDB("testdb", "leveldb", "temp")
@@ -414,7 +415,7 @@ func TestAttachOrDetachBlocks(t *testing.T) {
                for k, v := range c.before {
                        utxoViewpoint.Entries[k] = v
                }
-               if err := store.SaveChainStatus(node, node, utxoViewpoint, map[uint64]*state.VoteResult{}); err != nil {
+               if err := store.SaveChainStatus(mockBlockHeader, mockBlockHeader, []*types.BlockHeader{mockBlockHeader}, utxoViewpoint, []*state.ConsensusResult{}); err != nil {
                        t.Error(err)
                }
 
@@ -436,7 +437,7 @@ func TestAttachOrDetachBlocks(t *testing.T) {
                                t.Error(err)
                        }
                }
-               if err := store.SaveChainStatus(node, node, utxoViewpoint, map[uint64]*state.VoteResult{}); err != nil {
+               if err := store.SaveChainStatus(mockBlockHeader, mockBlockHeader, []*types.BlockHeader{mockBlockHeader}, utxoViewpoint, []*state.ConsensusResult{}); err != nil {
                        t.Error(err)
                }