From 2a11d0ef886167f9022da0bee55e76eb7067a5ee Mon Sep 17 00:00:00 2001
From: Sebastian Molenda <sebastian.molenda@pubnub.com>
Date: Mon, 30 Sep 2024 15:56:59 +0200
Subject: [PATCH] Those who slipped under the radar

---
 pubnub/endpoints/access/grant_token.py                        | 4 ++--
 .../endpoints/channel_groups/add_channel_to_channel_group.py  | 2 +-
 .../channel_groups/remove_channel_from_channel_group.py       | 2 +-
 pubnub/endpoints/objects_v2/memberships/get_memberships.py    | 4 ++--
 pubnub/endpoints/objects_v2/uuid/get_uuid.py                  | 3 +++
 pubnub/endpoints/objects_v2/uuid/remove_uuid.py               | 2 +-
 pubnub/endpoints/objects_v2/uuid/set_uuid.py                  | 2 +-
 7 files changed, 11 insertions(+), 8 deletions(-)

diff --git a/pubnub/endpoints/access/grant_token.py b/pubnub/endpoints/access/grant_token.py
index aabb14f1..8f2da50d 100644
--- a/pubnub/endpoints/access/grant_token.py
+++ b/pubnub/endpoints/access/grant_token.py
@@ -124,8 +124,8 @@ def validate_params(self):
     def create_response(self, envelope) -> PNGrantTokenResult:
         return PNGrantTokenResult.from_json(envelope['data'])
 
-    def sync(self):
-        return super().sync()
+    def sync(self) -> PNGrantTokenResultEnvelope:
+        return PNGrantTokenResultEnvelope(super().sync())
 
     def is_auth_required(self):
         return False
diff --git a/pubnub/endpoints/channel_groups/add_channel_to_channel_group.py b/pubnub/endpoints/channel_groups/add_channel_to_channel_group.py
index 4438934b..cdcfdab1 100644
--- a/pubnub/endpoints/channel_groups/add_channel_to_channel_group.py
+++ b/pubnub/endpoints/channel_groups/add_channel_to_channel_group.py
@@ -58,7 +58,7 @@ def is_auth_required(self):
     def create_response(self, envelope) -> PNChannelGroupsAddChannelResult:
         return PNChannelGroupsAddChannelResult()
 
-    def sync(self):
+    def sync(self) -> PNChannelGroupsAddChannelResultEnvelope:
         return PNChannelGroupsAddChannelResultEnvelope(super().sync())
 
     def request_timeout(self):
diff --git a/pubnub/endpoints/channel_groups/remove_channel_from_channel_group.py b/pubnub/endpoints/channel_groups/remove_channel_from_channel_group.py
index 38bf73fa..e5b06fae 100644
--- a/pubnub/endpoints/channel_groups/remove_channel_from_channel_group.py
+++ b/pubnub/endpoints/channel_groups/remove_channel_from_channel_group.py
@@ -60,7 +60,7 @@ def is_auth_required(self):
     def create_response(self, envelope) -> PNChannelGroupsRemoveChannelResult:
         return PNChannelGroupsRemoveChannelResult()
 
-    def sync(self):
+    def sync(self) -> PNChannelGroupsRemoveChannelResultEnvelope:
         return PNChannelGroupsRemoveChannelResultEnvelope(super().sync())
 
     def request_timeout(self):
diff --git a/pubnub/endpoints/objects_v2/memberships/get_memberships.py b/pubnub/endpoints/objects_v2/memberships/get_memberships.py
index 8e8d7d45..12a331c6 100644
--- a/pubnub/endpoints/objects_v2/memberships/get_memberships.py
+++ b/pubnub/endpoints/objects_v2/memberships/get_memberships.py
@@ -35,8 +35,8 @@ def validate_specific_params(self):
     def create_response(self, envelope) -> PNGetMembershipsResult:
         return PNGetMembershipsResult(envelope)
 
-    def sync(self):
-        return super().sync()
+    def sync(self) -> PNGetMembershipsResultEnvelope:
+        return PNGetMembershipsResultEnvelope(super().sync())
 
     def operation_type(self):
         return PNOperationType.PNGetMembershipsOperation
diff --git a/pubnub/endpoints/objects_v2/uuid/get_uuid.py b/pubnub/endpoints/objects_v2/uuid/get_uuid.py
index b6a1b6e1..5672c6f6 100644
--- a/pubnub/endpoints/objects_v2/uuid/get_uuid.py
+++ b/pubnub/endpoints/objects_v2/uuid/get_uuid.py
@@ -31,6 +31,9 @@ def validate_specific_params(self):
     def create_response(self, envelope) -> PNGetUUIDMetadataResult:
         return PNGetUUIDMetadataResult(envelope)
 
+    def sync(self) -> PNGetUUIDMetadataResultEnvelope:
+        return PNGetUUIDMetadataResultEnvelope(super().sync())
+
     def operation_type(self):
         return PNOperationType.PNGetUuidMetadataOperation
 
diff --git a/pubnub/endpoints/objects_v2/uuid/remove_uuid.py b/pubnub/endpoints/objects_v2/uuid/remove_uuid.py
index 24e068cb..c18b282a 100644
--- a/pubnub/endpoints/objects_v2/uuid/remove_uuid.py
+++ b/pubnub/endpoints/objects_v2/uuid/remove_uuid.py
@@ -27,7 +27,7 @@ def validate_specific_params(self):
     def create_response(self, envelope) -> PNRemoveUUIDMetadataResult:
         return PNRemoveUUIDMetadataResult(envelope)
 
-    def sync(self):
+    def sync(self) -> PNRemoveUUIDMetadataResultEnvelope:
         return PNRemoveUUIDMetadataResultEnvelope(super().sync())
 
     def operation_type(self):
diff --git a/pubnub/endpoints/objects_v2/uuid/set_uuid.py b/pubnub/endpoints/objects_v2/uuid/set_uuid.py
index 63962d04..c1d17c1f 100644
--- a/pubnub/endpoints/objects_v2/uuid/set_uuid.py
+++ b/pubnub/endpoints/objects_v2/uuid/set_uuid.py
@@ -68,7 +68,7 @@ def validate_specific_params(self):
     def create_response(self, envelope) -> PNSetUUIDMetadataResult:
         return PNSetUUIDMetadataResult(envelope)
 
-    def sync(self):
+    def sync(self) -> PNSetUUIDMetadataResultEnvelope:
         return PNSetUUIDMetadataResultEnvelope(super().sync())
 
     def operation_type(self):