feat(staking/validator): add fn to generate random SocialHandleURIs

cosmos · Sep 21, 2024 · c894be1 · c894be1
1 parent 05bea00
commit c894be1
Show file tree

Hide file tree

Showing 3 changed files with 42 additions and 1 deletion.
diff --git a/x/staking/simulation/msg_factory.go b/x/staking/simulation/msg_factory.go
@@ -39,12 +39,17 @@ func MsgCreateValidatorFactory(k *keeper.Keeper) simsx.SimMsgFactoryFn[*types.Ms
 		}
 
 		selfDelegation := valOper.LiquidBalance().RandSubsetCoin(reporter, bondDenom)
+
 		description := types.NewDescription(
 			r.StringN(10),
 			r.StringN(10),
 			r.StringN(10),
 			r.StringN(10),
 			r.StringN(10),
+			types.Metadata{
+				ProfilePicUri:    RandURIOfHostLength(r.Rand, 10),
+				SocialHandleUris: RandSocialHandles(r.Rand, 2, 10),
+			},
 		)
 
 		maxCommission := math.LegacyNewDecWithPrec(int64(r.IntInRange(0, 100)), 2)
@@ -138,7 +143,10 @@ func MsgEditValidatorFactory(k *keeper.Keeper) simsx.SimMsgFactoryFn[*types.MsgE
 		}
 		valOpAddrBz := must(k.ValidatorAddressCodec().StringToBytes(val.GetOperator()))
 		valOper := testData.GetAccountbyAccAddr(reporter, valOpAddrBz)
-		d := types.NewDescription(r.StringN(10), r.StringN(10), r.StringN(10), r.StringN(10), r.StringN(10))
+		d := types.NewDescription(r.StringN(10), r.StringN(10), r.StringN(10), r.StringN(10), r.StringN(10), types.Metadata{
+			ProfilePicUri:    RandURIOfHostLength(r.Rand, 10),
+			SocialHandleUris: RandSocialHandles(r.Rand, 2, 10),
+		})
 
 		msg := types.NewMsgEditValidator(val.GetOperator(), d, &newCommissionRate, nil)
 		return []simsx.SimAccount{valOper}, msg

diff --git a/x/staking/simulation/rand_util.go b/x/staking/simulation/rand_util.go
@@ -22,3 +22,15 @@ func RandURIOfHostLength(r *rand.Rand, n int) string {
 
 	return uri.String()
 }
+
+// RandSocialHandleURIs returns a string array of length num with uris.
+func RandSocialHandleURIs(r *rand.Rand, num int, uriHostLength int) []string {
+	if num == 0 {
+		return []string{}
+	}
+	var socialHandles []string
+	for i := 0; i < num; i++ {
+		socialHandles = append(socialHandles, RandURIOfHostLength(r, uriHostLength))
+	}
+	return socialHandles
+}
diff --git a/x/staking/simulation/rand_util_test.go b/x/staking/simulation/rand_util_test.go
@@ -37,3 +37,24 @@ func TestRandURIOfHostLength(t *testing.T) {
 		})
 	}
 }
+
+func TestRandSocialHandleURIs(t *testing.T) {
+	t.Parallel()
+	r := rand.New(rand.NewSource(time.Now().Unix()))
+	tests := []struct {
+		name string
+		n    int
+		want int
+	}{
+		{"0-handles", 0, 0},
+		{"10-handles", 10, 10},
+		{"100-handles", 100, 100},
+	}
+	for _, tt := range tests {
+		tc := tt
+		t.Run(tc.name, func(t *testing.T) {
+			uris := simulation.RandSocialHandleURIs(r, tc.n, 10)
+			require.Equal(t, tc.want, len(uris))
+		})
+	}
+}