Skip to content
Open
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
3 changes: 3 additions & 0 deletions x/billing/keeper/keeper.go
Original file line number Diff line number Diff line change
Expand Up @@ -76,6 +76,9 @@ func (k Keeper) GetAuthority() string {
}

func (k Keeper) SetPlatformValidators(ctx sdk.Context, vals []string) error {
if err := types.ValidatePlatformValidators(vals); err != nil {
return err
}
params := k.GetParams(ctx)
params.PlatformValidators = vals
k.SetParams(ctx, params)
Expand Down
11 changes: 11 additions & 0 deletions x/billing/types/params.go
Original file line number Diff line number Diff line change
Expand Up @@ -66,11 +66,22 @@ func validatePlatformValidatorsParam(v interface{}) error {
if !ok {
return fmt.Errorf("could not unmarshal platform-validators parm for validation")
}
return ValidatePlatformValidators(vals)
}

// ValidatePlatformValidators validates that all platform validator addresses are valid
// and that there are no duplicates.
func ValidatePlatformValidators(vals []string) error {
seen := make(map[string]bool)
for _, val := range vals {
_, err := sdk.AccAddressFromBech32(val)
if err != nil {
return fmt.Errorf("invalid platform validator address: %s", val)
}
if seen[val] {
return fmt.Errorf("duplicate platform validator address: %s", val)
}
seen[val] = true
}
return nil
}
85 changes: 85 additions & 0 deletions x/billing/types/params_test.go
Original file line number Diff line number Diff line change
@@ -0,0 +1,85 @@
package types

import (
"testing"

sdk "github.com/cosmos/cosmos-sdk/types"
"github.com/stretchr/testify/require"
)

func init() {
config := sdk.GetConfig()
config.SetBech32PrefixForAccount("saga", "sagapub")
config.SetBech32PrefixForValidator("sagavaloper", "sagavaloperpub")
config.SetBech32PrefixForConsensusNode("sagavalcons", "sagavalconspub")
}

func TestValidatePlatformValidatorsParam(t *testing.T) {
addr1 := "saga14grgksm5pe5u4cf8pvcchfsl8mfg8mvj3c95l0"
addr2 := "saga1nkm3et2qcqgya0ad8wt6e20l6206xdtccw28c9"
addr3 := "saga1zggrvdnjzsfpc7sr7jnw2jl5v9u8vjn263x0s0"

tests := []struct {
name string
input interface{}
wantErr bool
errMsg string
}{
{
name: "valid empty list",
input: []string{},
wantErr: false,
},
{
name: "valid single address",
input: []string{addr1},
wantErr: false,
},
{
name: "valid multiple addresses",
input: []string{addr1, addr2},
wantErr: false,
},
{
name: "invalid address format",
input: []string{"invalid-address"},
wantErr: true,
errMsg: "invalid platform validator address",
},
{
name: "duplicate addresses",
input: []string{addr1, addr1},
wantErr: true,
errMsg: "duplicate platform validator address",
},
{
name: "duplicate addresses with valid in between",
input: []string{addr1, addr2, addr1},
wantErr: true,
errMsg: "duplicate platform validator address",
},
{
name: "three unique addresses",
input: []string{addr1, addr2, addr3},
wantErr: false,
},
{
name: "wrong type",
input: "not a slice",
wantErr: true,
errMsg: "could not unmarshal platform-validators",
},
}

for _, tt := range tests {
t.Run(tt.name, func(t *testing.T) {
err := validatePlatformValidatorsParam(tt.input)
if tt.wantErr {
require.Error(t, err)
require.Contains(t, err.Error(), tt.errMsg)
} else {
require.NoError(t, err)
}
})
}
}
Loading