Dev (#148)

[bytom/bytom.git] / blockchain / txdb / snapshot_test.go

diff --git a/blockchain/txdb/snapshot_test.go b/blockchain/txdb/snapshot_test.go
index 75d1098..ad03729 100644 (file)
--- a/blockchain/txdb/snapshot_test.go
+++ b/blockchain/txdb/snapshot_test.go
@@ -8,8 +8,39 @@ import (
        dbm "github.com/tendermint/tmlibs/db"
 
        "github.com/bytom/protocol/bc"
+       "github.com/bytom/protocol/state"
 )
 
+func TestProtoSnapshotTree(t *testing.T) {
+       testDB := dbm.NewDB("testdb", "leveldb", "temp")
+       defer os.RemoveAll("temp")
+
+       hashes := []bc.Hash{}
+       insertSnap := state.Empty()
+
+       hash := bc.Hash{}
+       for i := uint64(0); i <= uint64(10); i++ {
+               hash.V0 = i
+               hashes = append(hashes, hash)
+               insertSnap.Tree.Insert(hash.Bytes())
+       }
+
+       if err := saveSnapshot(testDB, insertSnap, &hash); err != nil {
+               t.Errorf(err.Error())
+       }
+
+       popSnap, err := getSnapshot(testDB, &hash)
+       if err != nil {
+               t.Errorf(err.Error())
+       }
+
+       for _, h := range hashes {
+               if !popSnap.Tree.Contains(h.Bytes()) {
+                       t.Errorf("%s isn't in the snap tree", h.String())
+               }
+       }
+}
+
 func TestCleanSnapshotDB(t *testing.T) {
        testDB := dbm.NewDB("testdb", "leveldb", "temp")
        defer os.RemoveAll("temp")