diff options
author | Mission Liao <mission.liao@dexon.org> | 2018-08-15 23:10:21 +0800 |
---|---|---|
committer | GitHub <noreply@github.com> | 2018-08-15 23:10:21 +0800 |
commit | d3107b56cbef1f05baddb64880c3e97d7eda87a4 (patch) | |
tree | b94d05f101f61c9808ae1681a28dde4c37a5068f /core/test/stopper.go | |
parent | 39f1d8ae529805fa410d3ed08358c568343705a5 (diff) | |
download | tangerine-consensus-d3107b56cbef1f05baddb64880c3e97d7eda87a4.tar tangerine-consensus-d3107b56cbef1f05baddb64880c3e97d7eda87a4.tar.gz tangerine-consensus-d3107b56cbef1f05baddb64880c3e97d7eda87a4.tar.bz2 tangerine-consensus-d3107b56cbef1f05baddb64880c3e97d7eda87a4.tar.lz tangerine-consensus-d3107b56cbef1f05baddb64880c3e97d7eda87a4.tar.xz tangerine-consensus-d3107b56cbef1f05baddb64880c3e97d7eda87a4.tar.zst tangerine-consensus-d3107b56cbef1f05baddb64880c3e97d7eda87a4.zip |
test: add test.Scheduler (#58)
When simulating execution of core.Consensus by passing packets through golang-channel or real-socket, we need to utilize time.Sleep and time.Now to simulate the required network/proposing latency. It's problematic when we try to test a simulation with long network latency.
Instead, Scheduler would try to execute the event with minimum timestamp, thus time.Sleep is replaced with Scheduler.nextTick, and time.Now is replaced with Event.Time.
Changes:
- Add test.Scheduler.
- Add test.Stopper interface to provide encapsulate different stop conditions for scheduler.
- Add a reference implementation for test.Stopper, it will stop scheduler when all validators confirmed X blocks proposed from themselves.
- Add a test scenario on core.Consensus that all validators are not byzantine.
Diffstat (limited to 'core/test/stopper.go')
-rw-r--r-- | core/test/stopper.go | 86 |
1 files changed, 86 insertions, 0 deletions
diff --git a/core/test/stopper.go b/core/test/stopper.go new file mode 100644 index 0000000..da4d205 --- /dev/null +++ b/core/test/stopper.go @@ -0,0 +1,86 @@ +// 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 test + +import ( + "sync" + + "github.com/dexon-foundation/dexon-consensus-core/blockdb" + "github.com/dexon-foundation/dexon-consensus-core/core/types" +) + +// StopByConfirmedBlocks would make sure each validators confirms +// at least X blocks proposed by itself. +type StopByConfirmedBlocks struct { + apps map[types.ValidatorID]*App + dbs map[types.ValidatorID]blockdb.BlockDatabase + lastCheckDelivered map[types.ValidatorID]int + confirmedBlocks map[types.ValidatorID]int + blockCount int + lock sync.Mutex +} + +// NewStopByConfirmedBlocks construct an StopByConfirmedBlocks instance. +func NewStopByConfirmedBlocks( + blockCount int, + apps map[types.ValidatorID]*App, + dbs map[types.ValidatorID]blockdb.BlockDatabase) *StopByConfirmedBlocks { + + confirmedBlocks := make(map[types.ValidatorID]int) + for vID := range apps { + confirmedBlocks[vID] = 0 + } + return &StopByConfirmedBlocks{ + apps: apps, + dbs: dbs, + lastCheckDelivered: make(map[types.ValidatorID]int), + confirmedBlocks: confirmedBlocks, + blockCount: blockCount, + } +} + +// ShouldStop implements Stopper interface. +func (s *StopByConfirmedBlocks) ShouldStop(vID types.ValidatorID) bool { + s.lock.Lock() + defer s.lock.Unlock() + + // Accumulate confirmed blocks proposed by this validator in this round. + lastChecked := s.lastCheckDelivered[vID] + currentConfirmedBlocks := s.confirmedBlocks[vID] + db := s.dbs[vID] + s.apps[vID].Check(func(app *App) { + for _, h := range app.DeliverSequence[lastChecked:] { + b, err := db.Get(h) + if err != nil { + panic(err) + } + if b.ProposerID == vID { + currentConfirmedBlocks++ + } + } + s.lastCheckDelivered[vID] = len(app.DeliverSequence) + }) + s.confirmedBlocks[vID] = currentConfirmedBlocks + // Check if all validators confirmed at least 'blockCount' blocks. + for _, v := range s.confirmedBlocks { + if v < s.blockCount { + return false + } + } + return true +} |