// 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) GetStateForConfigAtRound(_ uint64) *vm.GovernanceState {
	return &vm.GovernanceState{g.statedb}
}

func (g *govStateFetcher) NotarySetNodeKeyAddresses(round uint64) (map[common.Address]struct{}, error) {
	return make(map[common.Address]struct{}), nil
}

type DexconTestSuite struct {
	suite.Suite

	config  *params.DexconConfig
	memDB   *ethdb.MemDatabase
	stateDB *state.StateDB
	s       *vm.GovernanceState
}

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.GovernanceState{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.RoundLength = 3600
	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.SetCRS(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 / numBlocksInCurRound
	// 0.1875 * 1e18 * 3600 * 1000 / (86400 * 1000 * 365 * 3600) = 5945585996.96
	d.Require().Equal(big.NewInt(5945585996), consensus.calculateBlockReward(0))
}

func TestDexcon(t *testing.T) {
	suite.Run(t, new(DexconTestSuite))
}