Those who slipped under the radar

pubnub · Sep 30, 2024 · a4f849f · a4f849f
1 parent 3acbac6
commit a4f849f
Show file tree

Hide file tree

Showing 6 changed files with 10 additions and 7 deletions.
diff --git 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/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
@@ -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
@@ -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
@@ -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
@@ -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):