New APIs: Update multiple connections in a single workbook/datasource (#1638)

Update multiple connections in a single workbook - Takes multiple connection, authType and credentials as input
Update multiple connections in a single datasource - Takes multiple connection, authType and credentials as input

---------

Co-authored-by: Jordan Woods <13803242+jorwoods@users.noreply.github.com>

Author: 2 people

samples/update_connection_auth.py

@@ -51,7 +51,7 @@ def main():
         connection = connections[0]
         connection.username = args.datasource_username
         connection.password = args.datasource_password
-        connection.auth_type = args.authentication_type
+        connection.authentication_type = args.authentication_type
         connection.embed_password = True
 
         updated_connection = update_function(resource, connection)

test/test_datasource.py

@@ -157,6 +157,44 @@ def test_update_copy_fields(server) -> None:
     assert single_datasource._project_name == updated_datasource._project_name
 
 
+    def test_update_connections(self) -> None:
+        populate_xml, response_xml = read_xml_assets(POPULATE_CONNECTIONS_XML, UPDATE_CONNECTIONS_XML)
+
+        with requests_mock.Mocker() as m:
+
+            datasource_id = "9dbd2263-16b5-46e1-9c43-a76bb8ab65fb"
+            connection_luids = ["be786ae0-d2bf-4a4b-9b34-e2de8d2d4488", "a1b2c3d4-e5f6-7a8b-9c0d-123456789abc"]
+
+            datasource = TSC.DatasourceItem(datasource_id)
+            datasource._id = "9dbd2263-16b5-46e1-9c43-a76bb8ab65fb"
+            datasource.owner_id = "dd2239f6-ddf1-4107-981a-4cf94e415794"
+            self.server.version = "3.26"
+
+            url = f"{self.server.baseurl}/{datasource.id}/connections"
+            m.get(
+                "http://test/api/3.26/sites/dad65087-b08b-4603-af4e-2887b8aafc67/datasources/9dbd2263-16b5-46e1-9c43-a76bb8ab65fb/connections",
+                text=populate_xml,
+            )
+            m.put(
+                "http://test/api/3.26/sites/dad65087-b08b-4603-af4e-2887b8aafc67/datasources/9dbd2263-16b5-46e1-9c43-a76bb8ab65fb/connections",
+                text=response_xml,
+            )
+
+            print("BASEURL:", self.server.baseurl)
+            print("Calling PUT on:", f"{self.server.baseurl}/{datasource.id}/connections")
+
+            connection_items = self.server.datasources.update_connections(
+                datasource_item=datasource,
+                connection_luids=connection_luids,
+                authentication_type="auth-keypair",
+                username="testuser",
+                password="testpass",
+                embed_password=True,
+            )
+            updated_ids = [conn.id for conn in connection_items]
+
+            self.assertEqual(updated_ids, connection_luids)
+
     def test_update_connections(self) -> None:
         populate_xml, response_xml = read_xml_assets(POPULATE_CONNECTIONS_XML, UPDATE_CONNECTIONS_XML)
 

test/test_workbook.py

@@ -708,6 +708,38 @@ def test_publish_with_thumbnails_user_id(server: TSC.Server) -> None:
         assert re.search(b'thumbnailsUserId=\\"ee8c6e70-43b6-11e6-af4f-f7b0d8e20761\\"', request_body)
 
 
+    def test_update_workbook_connections(self) -> None:
+        populate_xml, response_xml = read_xml_assets(POPULATE_CONNECTIONS_XML, UPDATE_CONNECTIONS_XML)
+
+        with requests_mock.Mocker() as m:
+            workbook_id = "1a2b3c4d-5e6f-7a8b-9c0d-112233445566"
+            connection_luids = ["abc12345-def6-7890-gh12-ijklmnopqrst", "1234abcd-5678-efgh-ijkl-0987654321mn"]
+
+            workbook = TSC.WorkbookItem(workbook_id)
+            workbook._id = workbook_id
+            self.server.version = "3.26"
+            url = f"{self.server.baseurl}/{workbook_id}/connections"
+            m.get(
+                "http://test/api/3.26/sites/dad65087-b08b-4603-af4e-2887b8aafc67/workbooks/1a2b3c4d-5e6f-7a8b-9c0d-112233445566/connections",
+                text=populate_xml,
+            )
+            m.put(
+                "http://test/api/3.26/sites/dad65087-b08b-4603-af4e-2887b8aafc67/workbooks/1a2b3c4d-5e6f-7a8b-9c0d-112233445566/connections",
+                text=response_xml,
+            )
+
+            connection_items = self.server.workbooks.update_connections(
+                workbook_item=workbook,
+                connection_luids=connection_luids,
+                authentication_type="AD Service Principal",
+                username="svc-client",
+                password="secret-token",
+                embed_password=True,
+            )
+            updated_ids = [conn.id for conn in connection_items]
+
+            self.assertEqual(updated_ids, connection_luids)
+
     def test_update_workbook_connections(self) -> None:
         populate_xml, response_xml = read_xml_assets(POPULATE_CONNECTIONS_XML, UPDATE_CONNECTIONS_XML)