diff options
Diffstat (limited to 'core/test/app.go')
-rw-r--r-- | core/test/app.go | 72 |
1 files changed, 72 insertions, 0 deletions
diff --git a/core/test/app.go b/core/test/app.go new file mode 100644 index 0000000..f596afb --- /dev/null +++ b/core/test/app.go @@ -0,0 +1,72 @@ +// 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 ( + "time" + + "github.com/dexon-foundation/dexon-consensus-core/common" + "github.com/dexon-foundation/dexon-consensus-core/core/types" +) + +// App implements Application interface for testing purpose. +type App struct { + Acked map[common.Hash]struct{} + TotalOrdered []*struct { + BlockHashes common.Hashes + Early bool + } + Delivered map[common.Hash]time.Time +} + +// NewApp constructs a TestApp instance. +func NewApp() *App { + return &App{ + Acked: make(map[common.Hash]struct{}), + TotalOrdered: []*struct { + BlockHashes common.Hashes + Early bool + }{}, + Delivered: make(map[common.Hash]time.Time), + } +} + +// StronglyAcked implements Application interface. +func (app *App) StronglyAcked(blockHash common.Hash) { + app.Acked[blockHash] = struct{}{} +} + +// TotalOrderingDeliver implements Application interface. +func (app *App) TotalOrderingDeliver(blocks []*types.Block, early bool) { + var hashes common.Hashes + for _, b := range blocks { + hashes = append(hashes, b.Hash) + } + app.TotalOrdered = append(app.TotalOrdered, &struct { + BlockHashes common.Hashes + Early bool + }{ + BlockHashes: hashes, + Early: early, + }) +} + +// DeliverBlock implements Application interface. +func (app *App) DeliverBlock(blockHash common.Hash, timestamp time.Time) { + app.Delivered[blockHash] = timestamp +} |