abci_test.go

package internal_test

import (
	sdk "github.com/line/lbm-sdk/types"
	"github.com/line/lbm-sdk/x/foundation"
	"github.com/line/lbm-sdk/x/foundation/keeper/internal"
)

func (s *KeeperTestSuite) TestBeginBlocker() {
	ctx, _ := s.ctx.CacheContext()

	taxRatio := sdk.MustNewDecFromStr("0.123456789")
	s.impl.SetParams(ctx, foundation.Params{
		FoundationTax: taxRatio,
	})

	before := s.impl.GetTreasury(ctx)
	s.Require().Equal(1, len(before))
	s.Require().Equal(sdk.NewDecFromInt(s.balance), before[0].Amount)

	// collect
	internal.BeginBlocker(ctx, s.impl)

	tax := sdk.NewDecFromInt(s.balance).MulTruncate(taxRatio).TruncateInt()
	// ensure the behavior does not change
	s.Require().Equal(sdk.NewInt(121932631), tax)

	expectedAfter := s.balance.Add(tax)
	after := s.impl.GetTreasury(ctx)
	s.Require().Equal(1, len(after))
	s.Require().Equal(sdk.NewDecFromInt(expectedAfter), after[0].Amount)
}

func (s *KeeperTestSuite) TestEndBlocker() {
	ctx, _ := s.ctx.CacheContext()

	// check preconditions
	for name, tc := range map[string]struct {
		id     uint64
		status foundation.ProposalStatus
	}{
		"active proposal": {
			s.activeProposal,
			foundation.PROPOSAL_STATUS_SUBMITTED,
		},
		"voted proposal": {
			s.votedProposal,
			foundation.PROPOSAL_STATUS_SUBMITTED,
		},
		"withdrawn proposal": {
			s.withdrawnProposal,
			foundation.PROPOSAL_STATUS_WITHDRAWN,
		},
		"invalid proposal": {
			s.invalidProposal,
			foundation.PROPOSAL_STATUS_SUBMITTED,
		},
	} {
		s.Run(name, func() {
			proposal, err := s.impl.GetProposal(ctx, tc.id)
			s.Require().NoError(err)
			s.Require().NotNil(proposal)
			s.Require().Equal(tc.status, proposal.Status)
		})
	}

	// voting periods end
	votingPeriod := s.impl.GetFoundationInfo(ctx).GetDecisionPolicy().GetVotingPeriod()
	ctx = ctx.WithBlockTime(ctx.BlockTime().Add(votingPeriod))
	internal.EndBlocker(ctx, s.impl)

	for name, tc := range map[string]struct {
		id      uint64
		removed bool
		status  foundation.ProposalStatus
	}{
		"active proposal": {
			id:     s.activeProposal,
			status: foundation.PROPOSAL_STATUS_ACCEPTED,
		},
		"voted proposal": {
			id:     s.votedProposal,
			status: foundation.PROPOSAL_STATUS_REJECTED,
		},
		"withdrawn proposal": {
			id:      s.withdrawnProposal,
			removed: true,
		},
		"invalid proposal": {
			id:     s.invalidProposal,
			status: foundation.PROPOSAL_STATUS_ACCEPTED,
		},
	} {
		s.Run(name, func() {
			proposal, err := s.impl.GetProposal(ctx, tc.id)
			if tc.removed {
				s.Require().Error(err)
				return
			}
			s.Require().NoError(err)
			s.Require().NotNil(proposal)
			s.Require().Equal(tc.status, proposal.Status)
		})
	}

	// proposals expire
	maxExecutionPeriod := foundation.DefaultConfig().MaxExecutionPeriod
	ctx = ctx.WithBlockTime(ctx.BlockTime().Add(maxExecutionPeriod))
	internal.EndBlocker(ctx, s.impl)

	// all proposals must be pruned
	s.Require().Empty(s.impl.GetProposals(ctx))
}