diff options
| author | Mission Liao <mission.liao@dexon.org> | 2019-02-15 14:18:59 +0800 |
|---|---|---|
| committer | Jimmy Hu <jimmy.hu@dexon.org> | 2019-02-19 10:48:50 +0800 |
| commit | 4dbdc22e355cf1f6f0c39af1b2f3737b7527bc0c (patch) | |
| tree | 625b7d34aa700d072ffb8e68dc89ed3936b76d29 /integration_test/utils.go | |
| parent | e4825619fb2499f5f534537c1a4d52d3e0bcacfe (diff) | |
| download | dexon-consensus-4dbdc22e355cf1f6f0c39af1b2f3737b7527bc0c.tar.gz dexon-consensus-4dbdc22e355cf1f6f0c39af1b2f3737b7527bc0c.tar.zst dexon-consensus-4dbdc22e355cf1f6f0c39af1b2f3737b7527bc0c.zip | |
big-bang: single chain (#446)
Diffstat (limited to 'integration_test/utils.go')
| -rw-r--r-- | integration_test/utils.go | 140 |
1 files changed, 0 insertions, 140 deletions
diff --git a/integration_test/utils.go b/integration_test/utils.go deleted file mode 100644 index 2fbfa1a..0000000 --- a/integration_test/utils.go +++ /dev/null @@ -1,140 +0,0 @@ -// Copyright 2018 The dexon-consensus Authors -// This file is part of the dexon-consensus library. -// -// The dexon-consensus 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 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 library. If not, see -// <http://www.gnu.org/licenses/>. - -package integration - -import ( - "errors" - "time" - - "github.com/dexon-foundation/dexon-consensus/core" - "github.com/dexon-foundation/dexon-consensus/core/crypto" - "github.com/dexon-foundation/dexon-consensus/core/db" - "github.com/dexon-foundation/dexon-consensus/core/test" - "github.com/dexon-foundation/dexon-consensus/core/types" -) - -func genRoundEndTimes( - configs []*types.Config, dMoment time.Time) (ends []time.Time) { - now := dMoment - for _, config := range configs { - now = now.Add(config.RoundInterval) - ends = append(ends, now) - } - return -} - -// loadAllConfigs loads all prepared configuration from governance, -// starts from round 0. -func loadAllConfigs(gov core.Governance) (configs []*types.Config) { - var round uint64 - for { - config := gov.Configuration(round) - if config == nil { - break - } - configs = append(configs, config) - round++ - } - return -} - -// decideOwnChains compute which chainIDs belongs to this node. -func decideOwnChains(numChains uint32, numNodes, id int) (own []uint32) { - var cur = uint32(id) - if numNodes == 0 { - panic(errors.New("attempt to arrange chains on 0 nodes")) - } - for { - if cur >= numChains { - break - } - own = append(own, cur) - cur += uint32(numNodes) - } - return -} - -// PrepareNodes setups nodes for testing. -func PrepareNodes( - gov *test.Governance, - prvKeys []crypto.PrivateKey, - maxNumChains uint32, - networkLatency, proposingLatency test.LatencyModel) ( - nodes map[types.NodeID]*Node, err error) { - if maxNumChains == 0 { - err = errors.New("zero NumChains is unexpected") - return - } - // Setup nodes, count of nodes is derived from the count of private keys - // hold in Governance. - nodes = make(map[types.NodeID]*Node) - dMoment := time.Now().UTC() - broadcastTargets := make(map[types.NodeID]struct{}) - for idx, prvKey := range prvKeys { - nID := types.NewNodeID(prvKey.PublicKey()) - broadcastTargets[nID] = struct{}{} - // Decides which chains are owned by this node. - if nodes[nID], err = newNode( - gov, - prvKey, - dMoment, - decideOwnChains(maxNumChains, len(prvKeys), idx), - networkLatency, - proposingLatency); err != nil { - return - } - } - // Assign broadcast targets. - for _, n := range nodes { - n.setBroadcastTargets(broadcastTargets) - n.gov().State().SwitchToRemoteMode() - } - return -} - -// VerifyApps is a helper to check delivery between test.Apps -func VerifyApps(apps map[types.NodeID]*test.App) (err error) { - for vFrom, fromApp := range apps { - if err = fromApp.Verify(); err != nil { - return - } - for vTo, toApp := range apps { - if vFrom == vTo { - continue - } - if err = fromApp.Compare(toApp); err != nil { - return - } - } - } - return -} - -// CollectAppAndDBFromNodes collects test.App and db.Database -// from nodes. -func CollectAppAndDBFromNodes(nodes map[types.NodeID]*Node) ( - apps map[types.NodeID]*test.App, - dbs map[types.NodeID]db.Database) { - apps = make(map[types.NodeID]*test.App) - dbs = make(map[types.NodeID]db.Database) - for nID, node := range nodes { - apps[nID] = node.app() - dbs[nID] = node.db() - } - return -} |