1 files changed, 107 insertions, 0 deletions
diff --git a/consensus/dexcon/dexcon_test.go b/consensus/dexcon/dexcon_test.go
new file mode 100644
index 000000000..7ba1be876
--- /dev/null
+++ b/consensus/dexcon/dexcon_test.go
@@ -0,0 +1,107 @@
+// Copyright 2019 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 dexcon
+
+import (
+	"math/big"
+	"testing"
+
+	"github.com/stretchr/testify/suite"
+
+	"github.com/dexon-foundation/dexon/common"
+	"github.com/dexon-foundation/dexon/core/state"
+	"github.com/dexon-foundation/dexon/core/vm"
+	"github.com/dexon-foundation/dexon/crypto"
+	"github.com/dexon-foundation/dexon/ethdb"
+	"github.com/dexon-foundation/dexon/params"
+)
+
+type GovStateFetcher struct {
+	statedb *state.StateDB
+}
+
+func (g *GovStateFetcher) GetGovStateHelperAtRound(_ uint64) *vm.GovernanceStateHelper {
+	return &vm.GovernanceStateHelper{g.statedb}
+}
+
+type DexconTestSuite struct {
+	suite.Suite
+
+	config  *params.DexconConfig
+	memDB   *ethdb.MemDatabase
+	stateDB *state.StateDB
+	s       *vm.GovernanceStateHelper
+}
+
+func (d *DexconTestSuite) SetupTest() {
+	memDB := ethdb.NewMemDatabase()
+	stateDB, err := state.New(common.Hash{}, state.NewDatabase(memDB))
+	if err != nil {
+		panic(err)
+	}
+	d.memDB = memDB
+	d.stateDB = stateDB
+	d.s = &vm.GovernanceStateHelper{stateDB}
+
+	config := params.TestnetChainConfig.Dexcon
+	config.LockupPeriod = 1000
+	config.NextHalvingSupply = new(big.Int).Mul(big.NewInt(1e18), big.NewInt(2.5e9))
+	config.LastHalvedAmount = new(big.Int).Mul(big.NewInt(1e18), big.NewInt(1.5e9))
+	config.MiningVelocity = 0.1875
+	config.RoundInterval = 3600000
+	config.NumChains = 6
+	config.MinBlockInterval = 1000
+
+	d.config = config
+
+	// Give governance contract balance so it will not be deleted because of being an empty state object.
+	stateDB.AddBalance(vm.GovernanceContractAddress, big.NewInt(1))
+
+	// Genesis CRS.
+	crs := crypto.Keccak256Hash([]byte(config.GenesisCRSText))
+	d.s.PushCRS(crs)
+
+	// Round 0 height.
+	d.s.PushRoundHeight(big.NewInt(0))
+
+	// Governance configuration.
+	d.s.UpdateConfiguration(config)
+
+	d.stateDB.Commit(true)
+}
+
+func (d *DexconTestSuite) TestBlockRewardCalculation() {
+	consensus := New()
+	consensus.SetGovStateFetcher(&GovStateFetcher{d.stateDB})
+
+	d.s.IncTotalStaked(big.NewInt(1e18))
+
+	// blockReard = miningVelocity * totalStaked * roundInterval / aYear / numBlocksInPrevRound
+	// 0.1875 * 1e18 * 3600 * 1000 / (86400 * 1000 * 365 * 6 * 3600) = 990930999.4926434
+	d.Require().Equal(big.NewInt(990930999), consensus.calculateBlockReward(0, d.stateDB))
+
+	// Round 1
+	d.s.PushRoundHeight(big.NewInt(4000 * 6))
+
+	// 0.1875 * 1e18 * 3600 * 1000 / (86400 * 1000 * 365 * 6 * 4000) = 891837899
+	d.Require().Equal(big.NewInt(891837899), consensus.calculateBlockReward(1, d.stateDB))
+}
+
+func TestDexcon(t *testing.T) {
+	suite.Run(t, new(DexconTestSuite))
+}