pr changes

Aliceyyy5539 · Feb 20, 2024 · 76b61f9 · 76b61f9
1 parent 677d7e9
commit 76b61f9
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 4 deletions.
diff --git a/x/projects/types/message_set_admin_policy.go b/x/projects/types/message_set_admin_policy.go
@@ -46,8 +46,10 @@ func (msg *MsgSetPolicy) ValidateBasic() error {
 		return sdkerrors.Wrapf(legacyerrors.ErrInvalidAddress, "invalid creator address (%s)", err)
 	}
 
-	if _, ok := planstypes.Geolocation_name[msg.Policy.GeolocationProfile]; !ok {
-		return sdkerrors.Wrapf(legacyerrors.ErrInvalidType, "invalid geolocationprofile")
+	if msg.Policy != nil {
+		if err := msg.Policy.ValidateBasicPolicy(false); err != nil {
+			return sdkerrors.Wrapf(err, "invalid policy")
+		}
 	}
 
 	return nil

diff --git a/x/projects/types/message_set_subscription_policy.go b/x/projects/types/message_set_subscription_policy.go
@@ -46,8 +46,10 @@ func (msg *MsgSetSubscriptionPolicy) ValidateBasic() error {
 		return sdkerrors.Wrapf(legacyerrors.ErrInvalidAddress, "invalid creator address (%s)", err)
 	}
 
-	if _, ok := planstypes.Geolocation_name[msg.Policy.GeolocationProfile]; !ok {
-		return sdkerrors.Wrapf(legacyerrors.ErrInvalidType, "invalid geolocationprofile")
+	if msg.Policy != nil {
+		if err := msg.Policy.ValidateBasicPolicy(false); err != nil {
+			return sdkerrors.Wrapf(err, "invalid policy")
+		}
 	}
 
 	return nil