Fix batch test in network

src/azure-cli/azure/cli/command_modules/network/private_link_resource_and_endpoint_connections/resource_providers/batch_provider.py

@@ -5,6 +5,7 @@
 from azure.cli.core.commands.client_factory import get_mgmt_service_client
 from azure.mgmt.batch import BatchManagement
 from azure.mgmt.batch.models import PrivateLinkServiceConnectionState, PrivateLinkServiceConnectionStatus
+from azure.mgmt.batch.models import PrivateEndpointConnection
 from knack.log import get_logger
 from . import GeneralPrivateEndpointClient
 
@@ -32,14 +33,16 @@ def _update_private_endpoint_connection_status(self,
                 status=PrivateLinkServiceConnectionStatus.rejected,
                 description=description)
 
+        parameters = PrivateEndpointConnection(private_link_service_connection_state=new_status)
+
         client = get_mgmt_service_client(
             cmd.cli_ctx,
             BatchManagement).private_endpoint_connection
-        return client.update(
+        return client.begin_update(
             resource_group_name=resource_group_name,
             account_name=resource_name,
             private_endpoint_connection_name=private_endpoint_connection_name,
-            private_link_service_connection_state=new_status)
+            parameters=parameters)
 
     def remove_private_endpoint_connection(self, cmd, resource_group_name, resource_name, name):
         logger.error("Microsoft.Batch/batchAccounts does not currently support deleting private endpoint "