fix type

markall93 · Sep 11, 2018 · 6a97d39 · 6a97d39
1 parent 1d0faec
commit 6a97d39
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 4 deletions.
diff --git a/x/vote/handler_test.go b/x/vote/handler_test.go
@@ -127,10 +127,20 @@ func TestRevokeBasic(t *testing.T) {
 
 	// user1 can revoke voter candidancy now
 	referenceList := &model.ReferenceList{
-		AllValidators: []types.AccountKey{},
+		AllValidators: []types.AccountKey{user1},
 	}
 	vm.storage.SetReferenceList(ctx, referenceList)
 	msg5 := NewRevokeStakeMsg("user1")
+	result2 := handler(ctx, msg5)
+	assert.Equal(t, ErrValidatorCannotRevoke().Result(), result2)
+	// invalid user cannot revoke
+	invalidMsg := NewRevokeStakeMsg("wqwdqwdasdsa")
+	resultInvalid := handler(ctx, invalidMsg)
+	assert.Equal(t, model.ErrVoterNotFound().Result(), resultInvalid)
+	// user1 can revoke voter candidancy now
+	referenceList = &model.ReferenceList{
+		AllValidators: []types.AccountKey{},
+	}
 
 	vm.storage.SetReferenceList(ctx, referenceList)
 	result3 := handler(ctx, msg5)

diff --git a/x/vote/wire.go b/x/vote/wire.go
@@ -6,9 +6,9 @@ import (
 
 // Register concrete types on wire codec
 func RegisterWire(cdc *wire.Codec) {
-	cdc.RegisterConcrete(StakeInMsg{}, "lino/voteDeposit", nil)
-	cdc.RegisterConcrete(RevokeStakeMsg{}, "lino/voteRevoke", nil)
-	cdc.RegisterConcrete(StakeOutMsg{}, "lino/voteWithdraw", nil)
+	cdc.RegisterConcrete(StakeInMsg{}, "lino/stakeIn", nil)
+	cdc.RegisterConcrete(RevokeStakeMsg{}, "lino/revokeStake", nil)
+	cdc.RegisterConcrete(StakeOutMsg{}, "lino/stakeOut", nil)
 	cdc.RegisterConcrete(DelegateMsg{}, "lino/delegate", nil)
 	cdc.RegisterConcrete(DelegatorWithdrawMsg{}, "lino/delegateWithdraw", nil)
 	cdc.RegisterConcrete(RevokeDelegationMsg{}, "lino/delegateRevoke", nil)