Fixed issues

Author: vchavatapalli

tableauserverclient/server/endpoint/datasources_endpoint.py

@@ -328,7 +328,7 @@ def update_connections(
         username: Optional[str] = None,
         password: Optional[str] = None,
         embed_password: Optional[bool] = None,
-    ) -> Iterable[str]:
+    ) -> list[ConnectionItem]:
         """
         Bulk updates one or more datasource connections by LUID.
 
@@ -368,8 +368,8 @@ def update_connections(
             embed_password=embed_password,
         )
         server_response = self.put_request(url, request_body)
-        connection_items = list(ConnectionItem.from_response(server_response.content, self.parent_srv.namespace))
-        updated_ids = [conn.id for conn in connection_items]
+        connection_items = ConnectionItem.from_response(server_response.content, self.parent_srv.namespace)
+        updated_ids: list[str] = [conn.id for conn in connection_items]
 
         logger.info(f"Updated connections for datasource {datasource_item.id}: {', '.join(updated_ids)}")
         return connection_items

tableauserverclient/server/endpoint/workbooks_endpoint.py

@@ -335,7 +335,7 @@ def update_connections(
         username: Optional[str] = None,
         password: Optional[str] = None,
         embed_password: Optional[bool] = None,
-    ) -> Iterable[str]:
+    ) -> list[ConnectionItem]:
         """
         Bulk updates one or more workbook connections by LUID, including authenticationType, username, password, and embedPassword.
 
@@ -377,8 +377,8 @@ def update_connections(
 
         # Send request
         server_response = self.put_request(url, request_body)
-        connection_items = list(ConnectionItem.from_response(server_response.content, self.parent_srv.namespace))
-        updated_ids = [conn.id for conn in connection_items]
+        connection_items = ConnectionItem.from_response(server_response.content, self.parent_srv.namespace)
+        updated_ids: list[str] = [conn.id for conn in connection_items]
 
         logger.info(f"Updated connections for workbook {workbook_item.id}: {', '.join(updated_ids)}")
         return connection_items