fix tests and add stringer interface

cosmos · Nov 30, 2023 · 2408dd4 · 2408dd4
1 parent a3bf1da
commit 2408dd4
Show file tree

Hide file tree

Showing 3 changed files with 14 additions and 2 deletions.
diff --git a/tests/integration/staking/keeper/validator_test.go b/tests/integration/staking/keeper/validator_test.go
@@ -155,7 +155,7 @@ func TestSlashToZeroPowerRemoved(t *testing.T) {
 	applyValidatorSetUpdates(t, f.sdkCtx, f.stakingKeeper, -1)
 	// validator should be unbonding
 	validator, _ = f.stakingKeeper.GetValidator(f.sdkCtx, addrVals[0])
-	assert.Equal(t, validator.GetStatus(), types.Unbonding)
+	assert.Equal(t, validator.GetStatus(), sdk.Unbonding)
 }
 
 // test how the validators are sorted, tests GetBondedValidatorsByPower

diff --git a/types/staking.go b/types/staking.go
@@ -58,6 +58,18 @@ const (
 	Bonded BondStatus = 3
 )
 
+// BondStatus_name is the string representation of BondStatus.
+var bondStatus_name = map[int32]string{
+	0: "BOND_STATUS_UNSPECIFIED",
+	1: "BOND_STATUS_UNBONDED",
+	2: "BOND_STATUS_UNBONDING",
+	3: "BOND_STATUS_BONDED",
+}
+
+func (x BondStatus) String() string {
+	return bondStatus_name[int32(x)]
+}
+
 // DelegationI delegation bond for a delegated proof of stake system
 type DelegationI interface {
 	GetDelegatorAddr() string  // delegator string for the bond

diff --git a/x/staking/keeper/grpc_query.go b/x/staking/keeper/grpc_query.go
@@ -44,7 +44,7 @@ func (k Querier) Validators(ctx context.Context, req *types.QueryValidatorsReque
 	valStore := prefix.NewStore(store, types.ValidatorsKey)
 
 	validators, pageRes, err := query.GenericFilteredPaginate(k.cdc, valStore, req.Pagination, func(key []byte, val *types.Validator) (*types.Validator, error) {
-		if req.Status != "" && !strings.EqualFold(string(val.GetStatus()), req.Status) { // TODO test
+		if req.Status != "" && !strings.EqualFold(val.GetStatus().String(), req.Status) {
 			return nil, nil
 		}