package slashing

import (
	"testing"

	"github.com/stretchr/testify/require"

	abci "github.com/tendermint/abci/types"
	tmtypes "github.com/tendermint/tendermint/types"

	sdk "github.com/cosmos/cosmos-sdk/types"
	"github.com/cosmos/cosmos-sdk/x/stake"
)

func TestBeginBlocker(t *testing.T) {
	ctx, ck, sk, keeper := createTestInput(t)
	addr, pk, amt := addrs[2], pks[2], int64(100)

	// bond the validator
	got := stake.NewHandler(sk)(ctx, newTestMsgCreateValidator(addr, pk, amt))
	require.True(t, got.IsOK())
	stake.EndBlocker(ctx, sk)
	require.Equal(t, ck.GetCoins(ctx, addr), sdk.Coins{{sk.GetParams(ctx).BondDenom, initCoins - amt}})
	require.Equal(t, sdk.NewRat(amt), sk.Validator(ctx, addr).GetPower())

	val := abci.Validator{
		PubKey: tmtypes.TM2PB.PubKey(pk),
		Power:  amt,
	}

	// mark the validator as having signed
	req := abci.RequestBeginBlock{
		Validators: []abci.SigningValidator{{
			Validator:       val,
			SignedLastBlock: true,
		}},
	}
	BeginBlocker(ctx, req, keeper)

	info, found := keeper.getValidatorSigningInfo(ctx, pk.Address())
	require.True(t, found)
	require.Equal(t, ctx.BlockHeight(), info.StartHeight)
	require.Equal(t, int64(1), info.IndexOffset)
	require.Equal(t, int64(0), info.JailedUntil)
	require.Equal(t, int64(1), info.SignedBlocksCounter)

	height := int64(0)

	// for 50 blocks, mark the validator as having signed
	for ; height < 50; height++ {
		ctx = ctx.WithBlockHeight(height)
		req = abci.RequestBeginBlock{
			Validators: []abci.SigningValidator{{
				Validator:       val,
				SignedLastBlock: true,
			}},
		}
		BeginBlocker(ctx, req, keeper)
	}

	// for 51 blocks, mark the validator as having not signed
	for ; height < 102; height++ {
		ctx = ctx.WithBlockHeight(height)
		req = abci.RequestBeginBlock{
			Validators: []abci.SigningValidator{{
				Validator:       val,
				SignedLastBlock: false,
			}},
		}
		BeginBlocker(ctx, req, keeper)
	}

	// validator should be revoked
	validator, found := sk.GetValidatorByPubKey(ctx, pk)
	require.True(t, found)
	require.Equal(t, sdk.Unbonded, validator.GetStatus())
}