// 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 test

import (
	"testing"
	"time"

	"github.com/dexon-foundation/dexon-consensus/common"
	"github.com/stretchr/testify/suite"
)

type GovernanceTestSuite struct {
	suite.Suite
}

func (s *GovernanceTestSuite) TestEqual() {
	var req = s.Require()
	// Setup a base governance.
	_, genesisNodes, err := NewKeys(20)
	req.NoError(err)
	g1, err := NewGovernance(NewState(
		1, genesisNodes, 100*time.Millisecond, &common.NullLogger{}, true), 2)
	req.NoError(err)
	// Create a governance with different lambda.
	g2, err := NewGovernance(NewState(
		1, genesisNodes, 50*time.Millisecond, &common.NullLogger{}, true), 2)
	req.NoError(err)
	req.False(g1.Equal(g2, true))
	// Create configs for 3 rounds for g1.
	g1.CatchUpWithRound(3)
	// Make a clone.
	g3 := g1.Clone()
	req.True(g1.Equal(g3, true))
	// Create a new round for g1.
	g1.CatchUpWithRound(4)
	req.False(g1.Equal(g3, true))
	// Make another clone.
	g4 := g1.Clone()
	req.True(g1.Equal(g4, true))
	// Add a node to g4.
	_, newNodes, err := NewKeys(1)
	req.NoError(err)
	g4.State().RequestChange(StateAddNode, newNodes[0])
	g1.CatchUpWithRound(5)
	g4.CatchUpWithRound(5)
	req.False(g1.Equal(g4, true))
	// Make a clone.
	g5 := g1.Clone()
	// Change its roundShift
	g5.roundShift = 3
	req.False(g1.Equal(g5, true))
}

func (s *GovernanceTestSuite) TestRegisterChange() {
	req := s.Require()
	_, genesisNodes, err := NewKeys(20)
	req.NoError(err)
	g, err := NewGovernance(NewState(
		1, genesisNodes, 100*time.Millisecond, &common.NullLogger{}, true), 2)
	req.NoError(err)
	// Unable to register change for genesis round.
	req.Error(g.RegisterConfigChange(0, StateChangeDKGSetSize, uint32(32)))
	// Make some round prepared.
	g.CatchUpWithRound(4)
	req.Equal(g.Configuration(4).DKGSetSize, uint32(20))
	// Unable to register change for prepared round.
	req.Error(g.RegisterConfigChange(4, StateChangeDKGSetSize, uint32(32)))
	// It's ok to make some change when condition is met.
	req.NoError(g.RegisterConfigChange(5, StateChangeDKGSetSize, uint32(32)))
	req.NoError(g.RegisterConfigChange(6, StateChangeDKGSetSize, uint32(32)))
	req.NoError(g.RegisterConfigChange(7, StateChangeDKGSetSize, uint32(40)))
	// In local mode, state for round 6 would be ready after notified with
	// round 2.
	g.NotifyRound(2)
	g.NotifyRound(3)
	// In local mode, state for round 7 would be ready after notified with
	// round 6.
	g.NotifyRound(4)
	// Notify governance to take a snapshot for round 7's configuration.
	g.NotifyRound(5)
	req.Equal(g.Configuration(6).DKGSetSize, uint32(32))
	req.Equal(g.Configuration(7).DKGSetSize, uint32(40))
}

func TestGovernance(t *testing.T) {
	suite.Run(t, new(GovernanceTestSuite))
}