modify the store code

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

diff --git a/database/cache_test.go b/database/cache_test.go
index 198d4cd..1419377 100644 (file)
--- a/database/cache_test.go
+++ b/database/cache_test.go
@@ -55,7 +55,7 @@ func TestBlockCache(t *testing.T) {
        for i := 0; i < 10; i++ {
                block := newBlock(uint64(i))
                hash := block.Hash()
-               if b, _ := cache.getBlockHeader(&hash); b != nil {
+               if _, ok := cache.lruBlockHeaders.Get(hash); ok {
                        t.Fatalf("find old block")
                }
        }
@@ -63,7 +63,7 @@ func TestBlockCache(t *testing.T) {
        for i := 10; i < maxCachedBlockHeaders+10; i++ {
                block := newBlock(uint64(i))
                hash := block.Hash()
-               if b, _ := cache.getBlockHeader(&hash); b == nil {
+               if _, ok := cache.lruBlockHeaders.Get(hash); !ok {
                        t.Fatalf("can't find new block")
                }
        }
@@ -75,14 +75,14 @@ func TestBlockCache(t *testing.T) {
 
        for i := 0; i < 10; i++ {
                voteResult := newVoteResult(uint64(i))
-               if v, _ := cache.getVoteResult(voteResult.Seq); v != nil {
+               if _, ok := cache.lruVoteResults.Get(voteResult.Seq); ok {
                        t.Fatalf("find old vote result")
                }
        }
 
        for i := 10; i < maxCachedVoteResults+10; i++ {
                voteResult := newVoteResult(uint64(i))
-               if v, _ := cache.getVoteResult(voteResult.Seq); v == nil {
+               if _, ok := cache.lruVoteResults.Get(voteResult.Seq); !ok {
                        t.Fatalf("can't find new vote result")
                }
        }