Dpos process block (#69)

[bytom/vapor.git] / database / store_test.go

diff --git a/database/store_test.go b/database/store_test.go
index ee8c02c..deffc05 100644 (file)
--- a/database/store_test.go
+++ b/database/store_test.go
@@ -155,11 +155,11 @@ func TestSaveChainStatus(t *testing.T) {
                },
        }
 
-       if err := store.SaveChainStatus(node, view); err != nil {
+       if err := store.SaveChainStatus(node, node, view, map[uint64]*state.VoteResult{}); err != nil {
                t.Fatal(err)
        }
 
-       expectStatus := &protocol.BlockStoreState{Height: node.Height, Hash: &node.Hash}
+       expectStatus := &protocol.BlockStoreState{Height: node.Height, Hash: &node.Hash, IrreversibleHeight: node.Height, IrreversibleHash: &node.Hash}
        if !testutil.DeepEqual(store.GetStoreStatus(), expectStatus) {
                t.Errorf("got block status:%v, expect block status:%v", store.GetStoreStatus(), expectStatus)
        }