diff options
author | Jimmy Hu <jimmy.hu@dexon.org> | 2018-11-02 10:40:58 +0800 |
---|---|---|
committer | Wei-Ning Huang <w@dexon.org> | 2019-04-09 21:32:52 +0800 |
commit | 119eb5889104d870ed608dcd9ccc7581489054b0 (patch) | |
tree | a340373d962d9bad8ecc8c14783857e7ad77e45c /dex/cache_test.go | |
parent | 2f936e7546afb84d7397e3dac6b577aea27220c6 (diff) | |
download | dexon-119eb5889104d870ed608dcd9ccc7581489054b0.tar dexon-119eb5889104d870ed608dcd9ccc7581489054b0.tar.gz dexon-119eb5889104d870ed608dcd9ccc7581489054b0.tar.bz2 dexon-119eb5889104d870ed608dcd9ccc7581489054b0.tar.lz dexon-119eb5889104d870ed608dcd9ccc7581489054b0.tar.xz dexon-119eb5889104d870ed608dcd9ccc7581489054b0.tar.zst dexon-119eb5889104d870ed608dcd9ccc7581489054b0.zip |
dex: implement PullBlocks/PullVotes (#1)
Diffstat (limited to 'dex/cache_test.go')
-rw-r--r-- | dex/cache_test.go | 144 |
1 files changed, 144 insertions, 0 deletions
diff --git a/dex/cache_test.go b/dex/cache_test.go new file mode 100644 index 000000000..96e07aa17 --- /dev/null +++ b/dex/cache_test.go @@ -0,0 +1,144 @@ +// Copyright 2018 The dexon-consensus-core Authors +// This file is part of the dexon-consensus-core library. +// +// The dexon-consensus-core library is free software: you can redistribute it +// and/or modify it under the terms of the GNU Lesser General Public License as +// published by the Free Software Foundation, either version 3 of the License, +// or (at your option) any later version. +// +// The dexon-consensus-core library is distributed in the hope that it will be +// useful, but WITHOUT ANY WARRANTY; without even the implied warranty of +// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU Lesser +// General Public License for more details. +// +// You should have received a copy of the GNU Lesser General Public License +// along with the dexon-consensus-core library. If not, see +// <http://www.gnu.org/licenses/>. + +package dex + +import ( + "testing" + + coreCommon "github.com/dexon-foundation/dexon-consensus-core/common" + coreTypes "github.com/dexon-foundation/dexon-consensus-core/core/types" +) + +func TestCacheVote(t *testing.T) { + cache := newCache(3) + pos0 := coreTypes.Position{ + Height: uint64(0), + } + pos1 := coreTypes.Position{ + Height: uint64(1), + } + vote1 := &coreTypes.Vote{ + BlockHash: coreCommon.NewRandomHash(), + Position: pos0, + } + vote2 := &coreTypes.Vote{ + BlockHash: coreCommon.NewRandomHash(), + Position: pos0, + } + vote3 := &coreTypes.Vote{ + BlockHash: coreCommon.NewRandomHash(), + Position: pos1, + } + vote4 := &coreTypes.Vote{ + BlockHash: coreCommon.NewRandomHash(), + Position: pos1, + } + cache.addVote(vote1) + cache.addVote(vote2) + cache.addVote(vote3) + + votes := cache.votes(pos0) + if len(votes) != 2 { + t.Errorf("fail to get votes: have %d, want 2", len(votes)) + } + if !votes[0].BlockHash.Equal(vote1.BlockHash) { + t.Errorf("get wrong vote: have %s, want %s", votes[0], vote1) + } + if !votes[1].BlockHash.Equal(vote2.BlockHash) { + t.Errorf("get wrong vote: have %s, want %s", votes[1], vote2) + } + votes = cache.votes(pos1) + if len(votes) != 1 { + t.Errorf("fail to get votes: have %d, want 1", len(votes)) + } + if !votes[0].BlockHash.Equal(vote3.BlockHash) { + t.Errorf("get wrong vote: have %s, want %s", votes[0], vote3) + } + + cache.addVote(vote4) + + votes = cache.votes(pos0) + if len(votes) != 0 { + t.Errorf("fail to get votes: have %d, want 0", len(votes)) + } + votes = cache.votes(pos1) + if len(votes) != 2 { + t.Errorf("fail to get votes: have %d, want 1", len(votes)) + } + if !votes[0].BlockHash.Equal(vote3.BlockHash) { + t.Errorf("get wrong vote: have %s, want %s", votes[0], vote3) + } + if !votes[1].BlockHash.Equal(vote4.BlockHash) { + t.Errorf("get wrong vote: have %s, want %s", votes[1], vote4) + } +} + +func TestCacheBlock(t *testing.T) { + cache := newCache(3) + block1 := &coreTypes.Block{ + Hash: coreCommon.NewRandomHash(), + } + block2 := &coreTypes.Block{ + Hash: coreCommon.NewRandomHash(), + } + block3 := &coreTypes.Block{ + Hash: coreCommon.NewRandomHash(), + } + block4 := &coreTypes.Block{ + Hash: coreCommon.NewRandomHash(), + } + cache.addBlock(block1) + cache.addBlock(block2) + cache.addBlock(block3) + + hashes := coreCommon.Hashes{block1.Hash, block2.Hash, block3.Hash, block4.Hash} + hashMap := map[coreCommon.Hash]struct{}{ + block1.Hash: struct{}{}, + block2.Hash: struct{}{}, + block3.Hash: struct{}{}, + } + blocks := cache.blocks(hashes) + if len(blocks) != 3 { + t.Errorf("fail to get blocks: have %d, want 3", len(blocks)) + } + for _, block := range blocks { + if _, exist := hashMap[block.Hash]; !exist { + t.Errorf("get wrong block: have %s, want %v", block, hashMap) + } + } + + cache.addBlock(block4) + + blocks = cache.blocks(hashes) + hashMap[block4.Hash] = struct{}{} + if len(blocks) != 3 { + t.Errorf("fail to get blocks: have %d, want 3", len(blocks)) + } + hasNewBlock := false + for _, block := range blocks { + if _, exist := hashMap[block.Hash]; !exist { + t.Errorf("get wrong block: have %s, want %v", block, hashMap) + } + if block.Hash.Equal(block4.Hash) { + hasNewBlock = true + } + } + if !hasNewBlock { + t.Errorf("expect block %s in cache, have %v", block4, blocks) + } +} |