From 41753e517c68575589c485b3f9570db94e59bcd0 Mon Sep 17 00:00:00 2001 From: Mission Liao Date: Thu, 6 Dec 2018 14:30:05 +0800 Subject: test: allow to log in test.State (#359) --- core/test/governance_test.go | 10 +++++++--- 1 file changed, 7 insertions(+), 3 deletions(-) (limited to 'core/test/governance_test.go') diff --git a/core/test/governance_test.go b/core/test/governance_test.go index 01993f9..30f1469 100644 --- a/core/test/governance_test.go +++ b/core/test/governance_test.go @@ -21,6 +21,7 @@ import ( "testing" "time" + "github.com/dexon-foundation/dexon-consensus/common" "github.com/stretchr/testify/suite" ) @@ -33,10 +34,12 @@ func (s *GovernanceTestSuite) TestEqual() { // Setup a base governance. _, genesisNodes, err := NewKeys(20) req.NoError(err) - g1, err := NewGovernance(genesisNodes, 100*time.Millisecond, 2) + g1, err := NewGovernance(NewState( + genesisNodes, 100*time.Millisecond, &common.NullLogger{}, true), 2) req.NoError(err) // Create a governance with different lambda. - g2, err := NewGovernance(genesisNodes, 50*time.Millisecond, 2) + g2, err := NewGovernance(NewState( + genesisNodes, 50*time.Millisecond, &common.NullLogger{}, true), 2) req.NoError(err) req.False(g1.Equal(g2, true)) // Create configs for 3 rounds for g1. @@ -68,7 +71,8 @@ func (s *GovernanceTestSuite) TestRegisterChange() { req := s.Require() _, genesisNodes, err := NewKeys(20) req.NoError(err) - g, err := NewGovernance(genesisNodes, 100*time.Millisecond, 2) + g, err := NewGovernance(NewState( + genesisNodes, 100*time.Millisecond, &common.NullLogger{}, true), 2) req.NoError(err) // Unable to register change for genesis round. req.Error(g.RegisterConfigChange(0, StateChangeNumChains, uint32(32))) -- cgit v1.2.3