add test

action/protocol/staking/handlers_test.go

@@ -15,6 +15,7 @@ import (
 	"time"
 
 	"github.com/golang/mock/gomock"
+	"github.com/mohae/deepcopy"
 	"github.com/pkg/errors"
 	"github.com/stretchr/testify/require"
 
@@ -531,6 +532,27 @@ func TestProtocol_HandleCandidateRegister(t *testing.T) {
 			nil,
 			iotextypes.ReceiptStatus_ErrCandidateConflict,
 		},
+		// register without self-stake
+		{
+			1201000,
+			identityset.Address(27),
+			1,
+			"test",
+			identityset.Address(28).String(),
+			identityset.Address(29).String(),
+			identityset.Address(30).String(),
+			"0",
+			"0",
+			uint32(10000),
+			false,
+			nil,
+			uint64(1000000),
+			uint64(1000000),
+			big.NewInt(1),
+			true,
+			nil,
+			iotextypes.ReceiptStatus_Success,
+		},
 	}
 
 	for _, test := range tests {
@@ -553,7 +575,9 @@ func TestProtocol_HandleCandidateRegister(t *testing.T) {
 			BlockTimeStamp: time.Now(),
 			GasLimit:       test.blkGasLimit,
 		})
-		ctx = genesis.WithGenesisContext(ctx, genesis.Default)
+		g := deepcopy.Copy(genesis.Default).(genesis.Genesis)
+		g.ToBeEnabledBlockHeight = 0
+		ctx = genesis.WithGenesisContext(ctx, g)
 		ctx = protocol.WithFeatureCtx(protocol.WithFeatureWithHeightCtx(ctx))
 		require.Equal(test.err, errors.Cause(p.Validate(ctx, act, sm)))
 		if test.err != nil {
@@ -570,16 +594,24 @@ func TestProtocol_HandleCandidateRegister(t *testing.T) {
 		if test.err == nil && test.status == iotextypes.ReceiptStatus_Success {
 			// check the special create bucket and candidate register log
 			tLogs := r.TransactionLogs()
-			require.Equal(2, len(tLogs))
-			cLog := tLogs[0]
-			require.Equal(test.caller.String(), cLog.Sender)
-			require.Equal(address.StakingBucketPoolAddr, cLog.Recipient)
-			require.Equal(test.amountStr, cLog.Amount.String())
-
-			cLog = tLogs[1]
-			require.Equal(test.caller.String(), cLog.Sender)
-			require.Equal(address.RewardingPoolAddr, cLog.Recipient)
-			require.Equal(p.config.RegistrationConsts.Fee.String(), cLog.Amount.String())
+			if test.amountStr == "0" {
+				require.Equal(1, len(tLogs))
+				cLog := tLogs[0]
+				require.Equal(test.caller.String(), cLog.Sender)
+				require.Equal(address.RewardingPoolAddr, cLog.Recipient)
+				require.Equal(p.config.RegistrationConsts.Fee.String(), cLog.Amount.String())
+			} else {
+				require.Equal(2, len(tLogs))
+				cLog := tLogs[0]
+				require.Equal(test.caller.String(), cLog.Sender)
+				require.Equal(address.StakingBucketPoolAddr, cLog.Recipient)
+				require.Equal(test.amountStr, cLog.Amount.String())
+
+				cLog = tLogs[1]
+				require.Equal(test.caller.String(), cLog.Sender)
+				require.Equal(address.RewardingPoolAddr, cLog.Recipient)
+				require.Equal(p.config.RegistrationConsts.Fee.String(), cLog.Amount.String())
+			}
 
 			// test candidate
 			candidate, _, err := csr.getCandidate(act.OwnerAddress())