Add RegisteredConsumerRewardDenoms to rapid test

cosmos · Sep 20, 2023 · 5841fce · 5841fce
1 parent 55d2686
commit 5841fce
Showing 1 changed file with 19 additions and 11 deletions.
diff --git a/tests/e2e/state_rapid_test.go b/tests/e2e/state_rapid_test.go
@@ -47,23 +47,31 @@ func GetChainStateGen() *rapid.Generator[ChainState] {
 			providerKeys := GetProviderKeysGen().Draw(t, "ProviderKeys")
 			consumerChainQueueSizes := GetConsumerChainQueueSizesGen().Draw(t, "ConsumerChainQueueSizes")
 			globalSlashQueueSize := rapid.Uint().Draw(t, "GlobalSlashQueueSize")
+			registeredConsumerRewardDenoms := GetRegisteredConsumerRewardDenomsGen().Draw(t, "RegisteredConsumerRewardDenoms")
 
 			return ChainState{
-				ValBalances:             &valBalances,
-				Proposals:               &proposals,
-				ValPowers:               &valPowers,
-				RepresentativePowers:    &representativePowers,
-				Params:                  &params,
-				Rewards:                 &rewards,
-				ConsumerChains:          &consumerChains,
-				AssignedKeys:            &assignedKeys,
-				ProviderKeys:            &providerKeys,
-				ConsumerChainQueueSizes: &consumerChainQueueSizes,
-				GlobalSlashQueueSize:    &globalSlashQueueSize,
+				ValBalances:                    &valBalances,
+				Proposals:                      &proposals,
+				ValPowers:                      &valPowers,
+				RepresentativePowers:           &representativePowers,
+				Params:                         &params,
+				Rewards:                        &rewards,
+				ConsumerChains:                 &consumerChains,
+				AssignedKeys:                   &assignedKeys,
+				ProviderKeys:                   &providerKeys,
+				ConsumerChainQueueSizes:        &consumerChainQueueSizes,
+				GlobalSlashQueueSize:           &globalSlashQueueSize,
+				RegisteredConsumerRewardDenoms: &registeredConsumerRewardDenoms,
 			}
 		})
 }
 
+func GetRegisteredConsumerRewardDenomsGen() *rapid.Generator[[]string] {
+	return rapid.Custom(func(t *rapid.T) []string {
+		return rapid.SliceOf(rapid.String()).Draw(t, "RegisteredConsumerRewardDenoms")
+	})
+}
+
 func GetConsumerChainQueueSizesGen() *rapid.Generator[map[ChainID]uint] {
 	return rapid.Custom(func(t *rapid.T) map[ChainID]uint {
 		return rapid.MapOf(GetChainIDGen(), rapid.Uint()).Draw(t, "ConsumerChainQueueSizes")