@azure/arm-databricks 3.0.0-beta.2 → 3.0.0-beta.4

package/dist/index.js

@@ -2,10 +2,64 @@
 
 Object.defineProperty(exports, '__esModule', { value: true });
 
-var coreClient = require('@azure/core-client');
 var tslib = require('tslib');
+var coreClient = require('@azure/core-client');
+var coreRestPipeline = require('@azure/core-rest-pipeline');
 var coreLro = require('@azure/core-lro');
 
+function _interopNamespace(e) {
+    if (e && e.__esModule) return e;
+    var n = Object.create(null);
+    if (e) {
+        Object.keys(e).forEach(function (k) {
+            if (k !== 'default') {
+                var d = Object.getOwnPropertyDescriptor(e, k);
+                Object.defineProperty(n, k, d.get ? d : {
+                    enumerable: true,
+                    get: function () { return e[k]; }
+                });
+            }
+        });
+    }
+    n["default"] = e;
+    return Object.freeze(n);
+}
+
+var coreClient__namespace = /*#__PURE__*/_interopNamespace(coreClient);
+var coreRestPipeline__namespace = /*#__PURE__*/_interopNamespace(coreRestPipeline);
+
+/*
+ * Copyright (c) Microsoft Corporation.
+ * Licensed under the MIT License.
+ *
+ * Code generated by Microsoft (R) AutoRest Code Generator.
+ * Changes may cause incorrect behavior and will be lost if the code is regenerated.
+ */
+const pageMap = new WeakMap();
+/**
+ * Given a result page from a pageable operation, returns a
+ * continuation token that can be used to begin paging from
+ * that point later.
+ * @param page A result object from calling .byPage() on a paged operation.
+ * @returns The continuation token that can be passed into byPage().
+ */
+function getContinuationToken(page) {
+    var _a;
+    if (typeof page !== "object" || page === null) {
+        return undefined;
+    }
+    return (_a = pageMap.get(page)) === null || _a === void 0 ? void 0 : _a.continuationToken;
+}
+function setContinuationToken(page, continuationToken) {
+    var _a;
+    if (typeof page !== "object" || page === null || !continuationToken) {
+        return;
+    }
+    const pageInfo = (_a = pageMap.get(page)) !== null && _a !== void 0 ? _a : {};
+    pageInfo.continuationToken = continuationToken;
+    pageMap.set(page, pageInfo);
+}
+
 /*
  * Copyright (c) Microsoft Corporation.
  * Licensed under the MIT License.
@@ -13,68 +67,132 @@ var coreLro = require('@azure/core-lro');
  * Code generated by Microsoft (R) AutoRest Code Generator.
  * Changes may cause incorrect behavior and will be lost if the code is regenerated.
  */
+/** Known values of {@link CustomParameterType} that the service accepts. */
+exports.KnownCustomParameterType = void 0;
 (function (KnownCustomParameterType) {
+    /** Bool */
     KnownCustomParameterType["Bool"] = "Bool";
+    /** Object */
     KnownCustomParameterType["Object"] = "Object";
+    /** String */
     KnownCustomParameterType["String"] = "String";
 })(exports.KnownCustomParameterType || (exports.KnownCustomParameterType = {}));
+/** Known values of {@link KeySource} that the service accepts. */
+exports.KnownKeySource = void 0;
 (function (KnownKeySource) {
+    /** Default */
     KnownKeySource["Default"] = "Default";
+    /** MicrosoftKeyvault */
     KnownKeySource["MicrosoftKeyvault"] = "Microsoft.Keyvault";
 })(exports.KnownKeySource || (exports.KnownKeySource = {}));
+/** Known values of {@link ProvisioningState} that the service accepts. */
+exports.KnownProvisioningState = void 0;
 (function (KnownProvisioningState) {
+    /** Accepted */
     KnownProvisioningState["Accepted"] = "Accepted";
+    /** Running */
     KnownProvisioningState["Running"] = "Running";
+    /** Ready */
     KnownProvisioningState["Ready"] = "Ready";
+    /** Creating */
     KnownProvisioningState["Creating"] = "Creating";
+    /** Created */
     KnownProvisioningState["Created"] = "Created";
+    /** Deleting */
     KnownProvisioningState["Deleting"] = "Deleting";
+    /** Deleted */
     KnownProvisioningState["Deleted"] = "Deleted";
+    /** Canceled */
     KnownProvisioningState["Canceled"] = "Canceled";
+    /** Failed */
     KnownProvisioningState["Failed"] = "Failed";
+    /** Succeeded */
     KnownProvisioningState["Succeeded"] = "Succeeded";
+    /** Updating */
     KnownProvisioningState["Updating"] = "Updating";
 })(exports.KnownProvisioningState || (exports.KnownProvisioningState = {}));
+/** Known values of {@link EncryptionKeySource} that the service accepts. */
+exports.KnownEncryptionKeySource = void 0;
 (function (KnownEncryptionKeySource) {
+    /** MicrosoftKeyvault */
     KnownEncryptionKeySource["MicrosoftKeyvault"] = "Microsoft.Keyvault";
 })(exports.KnownEncryptionKeySource || (exports.KnownEncryptionKeySource = {}));
+/** Known values of {@link PrivateLinkServiceConnectionStatus} that the service accepts. */
+exports.KnownPrivateLinkServiceConnectionStatus = void 0;
 (function (KnownPrivateLinkServiceConnectionStatus) {
+    /** Pending */
     KnownPrivateLinkServiceConnectionStatus["Pending"] = "Pending";
+    /** Approved */
     KnownPrivateLinkServiceConnectionStatus["Approved"] = "Approved";
+    /** Rejected */
     KnownPrivateLinkServiceConnectionStatus["Rejected"] = "Rejected";
+    /** Disconnected */
     KnownPrivateLinkServiceConnectionStatus["Disconnected"] = "Disconnected";
 })(exports.KnownPrivateLinkServiceConnectionStatus || (exports.KnownPrivateLinkServiceConnectionStatus = {}));
+/** Known values of {@link PrivateEndpointConnectionProvisioningState} that the service accepts. */
+exports.KnownPrivateEndpointConnectionProvisioningState = void 0;
 (function (KnownPrivateEndpointConnectionProvisioningState) {
+    /** Succeeded */
     KnownPrivateEndpointConnectionProvisioningState["Succeeded"] = "Succeeded";
+    /** Creating */
     KnownPrivateEndpointConnectionProvisioningState["Creating"] = "Creating";
+    /** Updating */
     KnownPrivateEndpointConnectionProvisioningState["Updating"] = "Updating";
+    /** Deleting */
     KnownPrivateEndpointConnectionProvisioningState["Deleting"] = "Deleting";
+    /** Failed */
     KnownPrivateEndpointConnectionProvisioningState["Failed"] = "Failed";
 })(exports.KnownPrivateEndpointConnectionProvisioningState || (exports.KnownPrivateEndpointConnectionProvisioningState = {}));
+/** Known values of {@link PublicNetworkAccess} that the service accepts. */
+exports.KnownPublicNetworkAccess = void 0;
 (function (KnownPublicNetworkAccess) {
+    /** Enabled */
     KnownPublicNetworkAccess["Enabled"] = "Enabled";
+    /** Disabled */
     KnownPublicNetworkAccess["Disabled"] = "Disabled";
 })(exports.KnownPublicNetworkAccess || (exports.KnownPublicNetworkAccess = {}));
+/** Known values of {@link RequiredNsgRules} that the service accepts. */
+exports.KnownRequiredNsgRules = void 0;
 (function (KnownRequiredNsgRules) {
+    /** AllRules */
     KnownRequiredNsgRules["AllRules"] = "AllRules";
+    /** NoAzureDatabricksRules */
     KnownRequiredNsgRules["NoAzureDatabricksRules"] = "NoAzureDatabricksRules";
+    /** NoAzureServiceRules */
     KnownRequiredNsgRules["NoAzureServiceRules"] = "NoAzureServiceRules";
 })(exports.KnownRequiredNsgRules || (exports.KnownRequiredNsgRules = {}));
+/** Known values of {@link CreatedByType} that the service accepts. */
+exports.KnownCreatedByType = void 0;
 (function (KnownCreatedByType) {
+    /** User */
     KnownCreatedByType["User"] = "User";
+    /** Application */
     KnownCreatedByType["Application"] = "Application";
+    /** ManagedIdentity */
     KnownCreatedByType["ManagedIdentity"] = "ManagedIdentity";
+    /** Key */
     KnownCreatedByType["Key"] = "Key";
 })(exports.KnownCreatedByType || (exports.KnownCreatedByType = {}));
+/** Known values of {@link PeeringState} that the service accepts. */
+exports.KnownPeeringState = void 0;
 (function (KnownPeeringState) {
+    /** Initiated */
     KnownPeeringState["Initiated"] = "Initiated";
+    /** Connected */
     KnownPeeringState["Connected"] = "Connected";
+    /** Disconnected */
     KnownPeeringState["Disconnected"] = "Disconnected";
 })(exports.KnownPeeringState || (exports.KnownPeeringState = {}));
+/** Known values of {@link PeeringProvisioningState} that the service accepts. */
+exports.KnownPeeringProvisioningState = void 0;
 (function (KnownPeeringProvisioningState) {
+    /** Succeeded */
     KnownPeeringProvisioningState["Succeeded"] = "Succeeded";
+    /** Updating */
     KnownPeeringProvisioningState["Updating"] = "Updating";
+    /** Deleting */
     KnownPeeringProvisioningState["Deleting"] = "Deleting";
+    /** Failed */
     KnownPeeringProvisioningState["Failed"] = "Failed";
 })(exports.KnownPeeringProvisioningState || (exports.KnownPeeringProvisioningState = {}));
 
@@ -1629,20 +1747,31 @@ class WorkspacesImpl {
             [Symbol.asyncIterator]() {
                 return this;
             },
-            byPage: () => {
-                return this.listByResourceGroupPagingPage(resourceGroupName, options);
+            byPage: (settings) => {
+                if (settings === null || settings === void 0 ? void 0 : settings.maxPageSize) {
+                    throw new Error("maxPageSize is not supported by this operation.");
+                }
+                return this.listByResourceGroupPagingPage(resourceGroupName, options, settings);
             }
         };
     }
-    listByResourceGroupPagingPage(resourceGroupName, options) {
+    listByResourceGroupPagingPage(resourceGroupName, options, settings) {
         return tslib.__asyncGenerator(this, arguments, function* listByResourceGroupPagingPage_1() {
-            let result = yield tslib.__await(this._listByResourceGroup(resourceGroupName, options));
-            yield yield tslib.__await(result.value || []);
-            let continuationToken = result.nextLink;
+            let result;
+            let continuationToken = settings === null || settings === void 0 ? void 0 : settings.continuationToken;
+            if (!continuationToken) {
+                result = yield tslib.__await(this._listByResourceGroup(resourceGroupName, options));
+                let page = result.value || [];
+                continuationToken = result.nextLink;
+                setContinuationToken(page, continuationToken);
+                yield yield tslib.__await(page);
+            }
             while (continuationToken) {
                 result = yield tslib.__await(this._listByResourceGroupNext(resourceGroupName, continuationToken, options));
                 continuationToken = result.nextLink;
-                yield yield tslib.__await(result.value || []);
+                let page = result.value || [];
+                setContinuationToken(page, continuationToken);
+                yield yield tslib.__await(page);
             }
         });
     }
@@ -1677,20 +1806,31 @@ class WorkspacesImpl {
             [Symbol.asyncIterator]() {
                 return this;
             },
-            byPage: () => {
-                return this.listBySubscriptionPagingPage(options);
+            byPage: (settings) => {
+                if (settings === null || settings === void 0 ? void 0 : settings.maxPageSize) {
+                    throw new Error("maxPageSize is not supported by this operation.");
+                }
+                return this.listBySubscriptionPagingPage(options, settings);
             }
         };
     }
-    listBySubscriptionPagingPage(options) {
+    listBySubscriptionPagingPage(options, settings) {
         return tslib.__asyncGenerator(this, arguments, function* listBySubscriptionPagingPage_1() {
-            let result = yield tslib.__await(this._listBySubscription(options));
-            yield yield tslib.__await(result.value || []);
-            let continuationToken = result.nextLink;
+            let result;
+            let continuationToken = settings === null || settings === void 0 ? void 0 : settings.continuationToken;
+            if (!continuationToken) {
+                result = yield tslib.__await(this._listBySubscription(options));
+                let page = result.value || [];
+                continuationToken = result.nextLink;
+                setContinuationToken(page, continuationToken);
+                yield yield tslib.__await(page);
+            }
             while (continuationToken) {
                 result = yield tslib.__await(this._listBySubscriptionNext(continuationToken, options));
                 continuationToken = result.nextLink;
-                yield yield tslib.__await(result.value || []);
+                let page = result.value || [];
+                setContinuationToken(page, continuationToken);
+                yield yield tslib.__await(page);
             }
         });
     }
@@ -1719,7 +1859,7 @@ class WorkspacesImpl {
      * @param options The options parameters.
      */
     get(resourceGroupName, workspaceName, options) {
-        return this.client.sendOperationRequest({ resourceGroupName, workspaceName, options }, getOperationSpec);
+        return this.client.sendOperationRequest({ resourceGroupName, workspaceName, options }, getOperationSpec$3);
     }
     /**
      * Deletes the workspace.
@@ -1751,11 +1891,13 @@ class WorkspacesImpl {
                     }
                 };
             });
-            const lro = new LroImpl(sendOperation, { resourceGroupName, workspaceName, options }, deleteOperationSpec);
-            return new coreLro.LroEngine(lro, {
+            const lro = new LroImpl(sendOperation, { resourceGroupName, workspaceName, options }, deleteOperationSpec$2);
+            const poller = new coreLro.LroEngine(lro, {
                 resumeFrom: options === null || options === void 0 ? void 0 : options.resumeFrom,
                 intervalInMs: options === null || options === void 0 ? void 0 : options.updateIntervalInMs
             });
+            yield poller.poll();
+            return poller;
         });
     }
     /**
@@ -1801,11 +1943,13 @@ class WorkspacesImpl {
                     }
                 };
             });
-            const lro = new LroImpl(sendOperation, { resourceGroupName, workspaceName, parameters, options }, createOrUpdateOperationSpec);
-            return new coreLro.LroEngine(lro, {
+            const lro = new LroImpl(sendOperation, { resourceGroupName, workspaceName, parameters, options }, createOrUpdateOperationSpec$1);
+            const poller = new coreLro.LroEngine(lro, {
                 resumeFrom: options === null || options === void 0 ? void 0 : options.resumeFrom,
                 intervalInMs: options === null || options === void 0 ? void 0 : options.updateIntervalInMs
             });
+            yield poller.poll();
+            return poller;
         });
     }
     /**
@@ -1853,10 +1997,12 @@ class WorkspacesImpl {
                 };
             });
             const lro = new LroImpl(sendOperation, { resourceGroupName, workspaceName, parameters, options }, updateOperationSpec);
-            return new coreLro.LroEngine(lro, {
+            const poller = new coreLro.LroEngine(lro, {
                 resumeFrom: options === null || options === void 0 ? void 0 : options.resumeFrom,
                 intervalInMs: options === null || options === void 0 ? void 0 : options.updateIntervalInMs
             });
+            yield poller.poll();
+            return poller;
         });
     }
     /**
@@ -1906,8 +2052,8 @@ class WorkspacesImpl {
     }
 }
 // Operation Specifications
-const serializer = coreClient.createSerializer(Mappers, /* isXml */ false);
-const getOperationSpec = {
+const serializer$5 = coreClient__namespace.createSerializer(Mappers, /* isXml */ false);
+const getOperationSpec$3 = {
     path: "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Databricks/workspaces/{workspaceName}",
     httpMethod: "GET",
     responses: {
@@ -1926,9 +2072,9 @@ const getOperationSpec = {
         subscriptionId
     ],
     headerParameters: [accept],
-    serializer
+    serializer: serializer$5
 };
-const deleteOperationSpec = {
+const deleteOperationSpec$2 = {
     path: "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Databricks/workspaces/{workspaceName}",
     httpMethod: "DELETE",
     responses: {
@@ -1948,9 +2094,9 @@ const deleteOperationSpec = {
         subscriptionId
     ],
     headerParameters: [accept],
-    serializer
+    serializer: serializer$5
 };
-const createOrUpdateOperationSpec = {
+const createOrUpdateOperationSpec$1 = {
     path: "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Databricks/workspaces/{workspaceName}",
     httpMethod: "PUT",
     responses: {
@@ -1980,7 +2126,7 @@ const createOrUpdateOperationSpec = {
     ],
     headerParameters: [accept, contentType],
     mediaType: "json",
-    serializer
+    serializer: serializer$5
 };
 const updateOperationSpec = {
     path: "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Databricks/workspaces/{workspaceName}",
@@ -2012,7 +2158,7 @@ const updateOperationSpec = {
     ],
     headerParameters: [accept, contentType],
     mediaType: "json",
-    serializer
+    serializer: serializer$5
 };
 const listByResourceGroupOperationSpec = {
     path: "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Databricks/workspaces",
@@ -2032,7 +2178,7 @@ const listByResourceGroupOperationSpec = {
         subscriptionId
     ],
     headerParameters: [accept],
-    serializer
+    serializer: serializer$5
 };
 const listBySubscriptionOperationSpec = {
     path: "/subscriptions/{subscriptionId}/providers/Microsoft.Databricks/workspaces",
@@ -2048,7 +2194,7 @@ const listBySubscriptionOperationSpec = {
     queryParameters: [apiVersion],
     urlParameters: [$host, subscriptionId],
     headerParameters: [accept],
-    serializer
+    serializer: serializer$5
 };
 const listByResourceGroupNextOperationSpec = {
     path: "{nextLink}",
@@ -2069,7 +2215,7 @@ const listByResourceGroupNextOperationSpec = {
         nextLink
     ],
     headerParameters: [accept],
-    serializer
+    serializer: serializer$5
 };
 const listBySubscriptionNextOperationSpec = {
     path: "{nextLink}",
@@ -2089,7 +2235,7 @@ const listBySubscriptionNextOperationSpec = {
         nextLink
     ],
     headerParameters: [accept],
-    serializer
+    serializer: serializer$5
 };
 
 /*
@@ -2122,20 +2268,31 @@ class OperationsImpl {
             [Symbol.asyncIterator]() {
                 return this;
             },
-            byPage: () => {
-                return this.listPagingPage(options);
+            byPage: (settings) => {
+                if (settings === null || settings === void 0 ? void 0 : settings.maxPageSize) {
+                    throw new Error("maxPageSize is not supported by this operation.");
+                }
+                return this.listPagingPage(options, settings);
             }
         };
     }
-    listPagingPage(options) {
+    listPagingPage(options, settings) {
         return tslib.__asyncGenerator(this, arguments, function* listPagingPage_1() {
-            let result = yield tslib.__await(this._list(options));
-            yield yield tslib.__await(result.value || []);
-            let continuationToken = result.nextLink;
+            let result;
+            let continuationToken = settings === null || settings === void 0 ? void 0 : settings.continuationToken;
+            if (!continuationToken) {
+                result = yield tslib.__await(this._list(options));
+                let page = result.value || [];
+                continuationToken = result.nextLink;
+                setContinuationToken(page, continuationToken);
+                yield yield tslib.__await(page);
+            }
             while (continuationToken) {
                 result = yield tslib.__await(this._listNext(continuationToken, options));
                 continuationToken = result.nextLink;
-                yield yield tslib.__await(result.value || []);
+                let page = result.value || [];
+                setContinuationToken(page, continuationToken);
+                yield yield tslib.__await(page);
             }
         });
     }
@@ -2162,7 +2319,7 @@ class OperationsImpl {
      * @param options The options parameters.
      */
     _list(options) {
-        return this.client.sendOperationRequest({ options }, listOperationSpec);
+        return this.client.sendOperationRequest({ options }, listOperationSpec$3);
     }
     /**
      * ListNext
@@ -2170,12 +2327,12 @@ class OperationsImpl {
      * @param options The options parameters.
      */
     _listNext(nextLink, options) {
-        return this.client.sendOperationRequest({ nextLink, options }, listNextOperationSpec);
+        return this.client.sendOperationRequest({ nextLink, options }, listNextOperationSpec$2);
     }
 }
 // Operation Specifications
-const serializer$1 = coreClient.createSerializer(Mappers, /* isXml */ false);
-const listOperationSpec = {
+const serializer$4 = coreClient__namespace.createSerializer(Mappers, /* isXml */ false);
+const listOperationSpec$3 = {
     path: "/providers/Microsoft.Databricks/operations",
     httpMethod: "GET",
     responses: {
@@ -2189,9 +2346,9 @@ const listOperationSpec = {
     queryParameters: [apiVersion],
     urlParameters: [$host],
     headerParameters: [accept],
-    serializer: serializer$1
+    serializer: serializer$4
 };
-const listNextOperationSpec = {
+const listNextOperationSpec$2 = {
     path: "{nextLink}",
     httpMethod: "GET",
     responses: {
@@ -2205,7 +2362,7 @@ const listNextOperationSpec = {
     queryParameters: [apiVersion],
     urlParameters: [$host, nextLink],
     headerParameters: [accept],
-    serializer: serializer$1
+    serializer: serializer$4
 };
 
 /*
@@ -2240,20 +2397,31 @@ class PrivateLinkResourcesImpl {
             [Symbol.asyncIterator]() {
                 return this;
             },
-            byPage: () => {
-                return this.listPagingPage(resourceGroupName, workspaceName, options);
+            byPage: (settings) => {
+                if (settings === null || settings === void 0 ? void 0 : settings.maxPageSize) {
+                    throw new Error("maxPageSize is not supported by this operation.");
+                }
+                return this.listPagingPage(resourceGroupName, workspaceName, options, settings);
             }
         };
     }
-    listPagingPage(resourceGroupName, workspaceName, options) {
+    listPagingPage(resourceGroupName, workspaceName, options, settings) {
         return tslib.__asyncGenerator(this, arguments, function* listPagingPage_1() {
-            let result = yield tslib.__await(this._list(resourceGroupName, workspaceName, options));
-            yield yield tslib.__await(result.value || []);
-            let continuationToken = result.nextLink;
+            let result;
+            let continuationToken = settings === null || settings === void 0 ? void 0 : settings.continuationToken;
+            if (!continuationToken) {
+                result = yield tslib.__await(this._list(resourceGroupName, workspaceName, options));
+                let page = result.value || [];
+                continuationToken = result.nextLink;
+                setContinuationToken(page, continuationToken);
+                yield yield tslib.__await(page);
+            }
             while (continuationToken) {
                 result = yield tslib.__await(this._listNext(resourceGroupName, workspaceName, continuationToken, options));
                 continuationToken = result.nextLink;
-                yield yield tslib.__await(result.value || []);
+                let page = result.value || [];
+                setContinuationToken(page, continuationToken);
+                yield yield tslib.__await(page);
             }
         });
     }
@@ -2282,7 +2450,7 @@ class PrivateLinkResourcesImpl {
      * @param options The options parameters.
      */
     _list(resourceGroupName, workspaceName, options) {
-        return this.client.sendOperationRequest({ resourceGroupName, workspaceName, options }, listOperationSpec$1);
+        return this.client.sendOperationRequest({ resourceGroupName, workspaceName, options }, listOperationSpec$2);
     }
     /**
      * Get the specified private link resource for the given group id (sub-resource)
@@ -2292,7 +2460,7 @@ class PrivateLinkResourcesImpl {
      * @param options The options parameters.
      */
     get(resourceGroupName, workspaceName, groupId, options) {
-        return this.client.sendOperationRequest({ resourceGroupName, workspaceName, groupId, options }, getOperationSpec$1);
+        return this.client.sendOperationRequest({ resourceGroupName, workspaceName, groupId, options }, getOperationSpec$2);
     }
     /**
      * ListNext
@@ -2306,8 +2474,8 @@ class PrivateLinkResourcesImpl {
     }
 }
 // Operation Specifications
-const serializer$2 = coreClient.createSerializer(Mappers, /* isXml */ false);
-const listOperationSpec$1 = {
+const serializer$3 = coreClient__namespace.createSerializer(Mappers, /* isXml */ false);
+const listOperationSpec$2 = {
     path: "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Databricks/workspaces/{workspaceName}/privateLinkResources",
     httpMethod: "GET",
     responses: {
@@ -2326,9 +2494,9 @@ const listOperationSpec$1 = {
         subscriptionId
     ],
     headerParameters: [accept],
-    serializer: serializer$2
+    serializer: serializer$3
 };
-const getOperationSpec$1 = {
+const getOperationSpec$2 = {
     path: "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Databricks/workspaces/{workspaceName}/privateLinkResources/{groupId}",
     httpMethod: "GET",
     responses: {
@@ -2348,7 +2516,7 @@ const getOperationSpec$1 = {
         groupId
     ],
     headerParameters: [accept],
-    serializer: serializer$2
+    serializer: serializer$3
 };
 const listNextOperationSpec$1 = {
     path: "{nextLink}",
@@ -2370,7 +2538,7 @@ const listNextOperationSpec$1 = {
         nextLink
     ],
     headerParameters: [accept],
-    serializer: serializer$2
+    serializer: serializer$3
 };
 
 /*
@@ -2405,20 +2573,31 @@ class PrivateEndpointConnectionsImpl {
             [Symbol.asyncIterator]() {
                 return this;
             },
-            byPage: () => {
-                return this.listPagingPage(resourceGroupName, workspaceName, options);
+            byPage: (settings) => {
+                if (settings === null || settings === void 0 ? void 0 : settings.maxPageSize) {
+                    throw new Error("maxPageSize is not supported by this operation.");
+                }
+                return this.listPagingPage(resourceGroupName, workspaceName, options, settings);
             }
         };
     }
-    listPagingPage(resourceGroupName, workspaceName, options) {
+    listPagingPage(resourceGroupName, workspaceName, options, settings) {
         return tslib.__asyncGenerator(this, arguments, function* listPagingPage_1() {
-            let result = yield tslib.__await(this._list(resourceGroupName, workspaceName, options));
-            yield yield tslib.__await(result.value || []);
-            let continuationToken = result.nextLink;
+            let result;
+            let continuationToken = settings === null || settings === void 0 ? void 0 : settings.continuationToken;
+            if (!continuationToken) {
+                result = yield tslib.__await(this._list(resourceGroupName, workspaceName, options));
+                let page = result.value || [];
+                continuationToken = result.nextLink;
+                setContinuationToken(page, continuationToken);
+                yield yield tslib.__await(page);
+            }
             while (continuationToken) {
                 result = yield tslib.__await(this._listNext(resourceGroupName, workspaceName, continuationToken, options));
                 continuationToken = result.nextLink;
-                yield yield tslib.__await(result.value || []);
+                let page = result.value || [];
+                setContinuationToken(page, continuationToken);
+                yield yield tslib.__await(page);
             }
         });
     }
@@ -2447,7 +2626,7 @@ class PrivateEndpointConnectionsImpl {
      * @param options The options parameters.
      */
     _list(resourceGroupName, workspaceName, options) {
-        return this.client.sendOperationRequest({ resourceGroupName, workspaceName, options }, listOperationSpec$2);
+        return this.client.sendOperationRequest({ resourceGroupName, workspaceName, options }, listOperationSpec$1);
     }
     /**
      * Get a private endpoint connection properties for a workspace
@@ -2462,7 +2641,7 @@ class PrivateEndpointConnectionsImpl {
             workspaceName,
             privateEndpointConnectionName,
             options
-        }, getOperationSpec$2);
+        }, getOperationSpec$1);
     }
     /**
      * Update the status of a private endpoint connection with the specified name
@@ -2503,10 +2682,12 @@ class PrivateEndpointConnectionsImpl {
                 privateEndpointConnection,
                 options
             }, createOperationSpec);
-            return new coreLro.LroEngine(lro, {
+            const poller = new coreLro.LroEngine(lro, {
                 resumeFrom: options === null || options === void 0 ? void 0 : options.resumeFrom,
                 intervalInMs: options === null || options === void 0 ? void 0 : options.updateIntervalInMs
             });
+            yield poller.poll();
+            return poller;
         });
     }
     /**
@@ -2560,10 +2741,12 @@ class PrivateEndpointConnectionsImpl {
                 privateEndpointConnectionName,
                 options
             }, deleteOperationSpec$1);
-            return new coreLro.LroEngine(lro, {
+            const poller = new coreLro.LroEngine(lro, {
                 resumeFrom: options === null || options === void 0 ? void 0 : options.resumeFrom,
                 intervalInMs: options === null || options === void 0 ? void 0 : options.updateIntervalInMs
             });
+            yield poller.poll();
+            return poller;
         });
     }
     /**
@@ -2587,12 +2770,12 @@ class PrivateEndpointConnectionsImpl {
      * @param options The options parameters.
      */
     _listNext(resourceGroupName, workspaceName, nextLink, options) {
-        return this.client.sendOperationRequest({ resourceGroupName, workspaceName, nextLink, options }, listNextOperationSpec$2);
+        return this.client.sendOperationRequest({ resourceGroupName, workspaceName, nextLink, options }, listNextOperationSpec);
     }
 }
 // Operation Specifications
-const serializer$3 = coreClient.createSerializer(Mappers, /* isXml */ false);
-const listOperationSpec$2 = {
+const serializer$2 = coreClient__namespace.createSerializer(Mappers, /* isXml */ false);
+const listOperationSpec$1 = {
     path: "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Databricks/workspaces/{workspaceName}/privateEndpointConnections",
     httpMethod: "GET",
     responses: {
@@ -2611,9 +2794,9 @@ const listOperationSpec$2 = {
         subscriptionId
     ],
     headerParameters: [accept],
-    serializer: serializer$3
+    serializer: serializer$2
 };
-const getOperationSpec$2 = {
+const getOperationSpec$1 = {
     path: "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Databricks/workspaces/{workspaceName}/privateEndpointConnections/{privateEndpointConnectionName}",
     httpMethod: "GET",
     responses: {
@@ -2633,7 +2816,7 @@ const getOperationSpec$2 = {
         privateEndpointConnectionName
     ],
     headerParameters: [accept],
-    serializer: serializer$3
+    serializer: serializer$2
 };
 const createOperationSpec = {
     path: "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Databricks/workspaces/{workspaceName}/privateEndpointConnections/{privateEndpointConnectionName}",
@@ -2666,7 +2849,7 @@ const createOperationSpec = {
     ],
     headerParameters: [accept, contentType],
     mediaType: "json",
-    serializer: serializer$3
+    serializer: serializer$2
 };
 const deleteOperationSpec$1 = {
     path: "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Databricks/workspaces/{workspaceName}/privateEndpointConnections/{privateEndpointConnectionName}",
@@ -2689,9 +2872,9 @@ const deleteOperationSpec$1 = {
         privateEndpointConnectionName
     ],
     headerParameters: [accept],
-    serializer: serializer$3
+    serializer: serializer$2
 };
-const listNextOperationSpec$2 = {
+const listNextOperationSpec = {
     path: "{nextLink}",
     httpMethod: "GET",
     responses: {
@@ -2711,7 +2894,7 @@ const listNextOperationSpec$2 = {
         nextLink
     ],
     headerParameters: [accept],
-    serializer: serializer$3
+    serializer: serializer$2
 };
 
 /*
@@ -2733,18 +2916,18 @@ class OutboundNetworkDependenciesEndpointsImpl {
     /**
      * Gets the list of endpoints that VNET Injected Workspace calls Azure Databricks Control Plane. You
      * must configure outbound access with these endpoints. For more information, see
-     * https://docs.microsoft.com/en-us/azure/databricks/administration-guide/cloud-configurations/azure/udr
+     * https://docs.microsoft.com/azure/databricks/administration-guide/cloud-configurations/azure/udr
      * @param resourceGroupName The name of the resource group. The name is case insensitive.
      * @param workspaceName The name of the workspace.
      * @param options The options parameters.
      */
     list(resourceGroupName, workspaceName, options) {
-        return this.client.sendOperationRequest({ resourceGroupName, workspaceName, options }, listOperationSpec$3);
+        return this.client.sendOperationRequest({ resourceGroupName, workspaceName, options }, listOperationSpec);
     }
 }
 // Operation Specifications
-const serializer$4 = coreClient.createSerializer(Mappers, /* isXml */ false);
-const listOperationSpec$3 = {
+const serializer$1 = coreClient__namespace.createSerializer(Mappers, /* isXml */ false);
+const listOperationSpec = {
     path: "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Databricks/workspaces/{workspaceName}/outboundNetworkDependenciesEndpoints",
     httpMethod: "GET",
     responses: {
@@ -2773,7 +2956,7 @@ const listOperationSpec$3 = {
         subscriptionId
     ],
     headerParameters: [accept],
-    serializer: serializer$4
+    serializer: serializer$1
 };
 
 /*
@@ -2808,20 +2991,31 @@ class VNetPeeringImpl {
             [Symbol.asyncIterator]() {
                 return this;
             },
-            byPage: () => {
-                return this.listByWorkspacePagingPage(resourceGroupName, workspaceName, options);
+            byPage: (settings) => {
+                if (settings === null || settings === void 0 ? void 0 : settings.maxPageSize) {
+                    throw new Error("maxPageSize is not supported by this operation.");
+                }
+                return this.listByWorkspacePagingPage(resourceGroupName, workspaceName, options, settings);
             }
         };
     }
-    listByWorkspacePagingPage(resourceGroupName, workspaceName, options) {
+    listByWorkspacePagingPage(resourceGroupName, workspaceName, options, settings) {
         return tslib.__asyncGenerator(this, arguments, function* listByWorkspacePagingPage_1() {
-            let result = yield tslib.__await(this._listByWorkspace(resourceGroupName, workspaceName, options));
-            yield yield tslib.__await(result.value || []);
-            let continuationToken = result.nextLink;
+            let result;
+            let continuationToken = settings === null || settings === void 0 ? void 0 : settings.continuationToken;
+            if (!continuationToken) {
+                result = yield tslib.__await(this._listByWorkspace(resourceGroupName, workspaceName, options));
+                let page = result.value || [];
+                continuationToken = result.nextLink;
+                setContinuationToken(page, continuationToken);
+                yield yield tslib.__await(page);
+            }
             while (continuationToken) {
                 result = yield tslib.__await(this._listByWorkspaceNext(resourceGroupName, workspaceName, continuationToken, options));
                 continuationToken = result.nextLink;
-                yield yield tslib.__await(result.value || []);
+                let page = result.value || [];
+                setContinuationToken(page, continuationToken);
+                yield yield tslib.__await(page);
             }
         });
     }
@@ -2851,7 +3045,7 @@ class VNetPeeringImpl {
      * @param options The options parameters.
      */
     get(resourceGroupName, workspaceName, peeringName, options) {
-        return this.client.sendOperationRequest({ resourceGroupName, workspaceName, peeringName, options }, getOperationSpec$3);
+        return this.client.sendOperationRequest({ resourceGroupName, workspaceName, peeringName, options }, getOperationSpec);
     }
     /**
      * Deletes the workspace vNetPeering.
@@ -2884,11 +3078,13 @@ class VNetPeeringImpl {
                     }
                 };
             });
-            const lro = new LroImpl(sendOperation, { resourceGroupName, workspaceName, peeringName, options }, deleteOperationSpec$2);
-            return new coreLro.LroEngine(lro, {
+            const lro = new LroImpl(sendOperation, { resourceGroupName, workspaceName, peeringName, options }, deleteOperationSpec);
+            const poller = new coreLro.LroEngine(lro, {
                 resumeFrom: options === null || options === void 0 ? void 0 : options.resumeFrom,
                 intervalInMs: options === null || options === void 0 ? void 0 : options.updateIntervalInMs
             });
+            yield poller.poll();
+            return poller;
         });
     }
     /**
@@ -2942,11 +3138,13 @@ class VNetPeeringImpl {
                 peeringName,
                 virtualNetworkPeeringParameters,
                 options
-            }, createOrUpdateOperationSpec$1);
-            return new coreLro.LroEngine(lro, {
+            }, createOrUpdateOperationSpec);
+            const poller = new coreLro.LroEngine(lro, {
                 resumeFrom: options === null || options === void 0 ? void 0 : options.resumeFrom,
                 intervalInMs: options === null || options === void 0 ? void 0 : options.updateIntervalInMs
             });
+            yield poller.poll();
+            return poller;
         });
     }
     /**
@@ -2984,8 +3182,8 @@ class VNetPeeringImpl {
     }
 }
 // Operation Specifications
-const serializer$5 = coreClient.createSerializer(Mappers, /* isXml */ false);
-const getOperationSpec$3 = {
+const serializer = coreClient__namespace.createSerializer(Mappers, /* isXml */ false);
+const getOperationSpec = {
     path: "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Databricks/workspaces/{workspaceName}/virtualNetworkPeerings/{peeringName}",
     httpMethod: "GET",
     responses: {
@@ -3006,9 +3204,9 @@ const getOperationSpec$3 = {
         peeringName
     ],
     headerParameters: [accept],
-    serializer: serializer$5
+    serializer
 };
-const deleteOperationSpec$2 = {
+const deleteOperationSpec = {
     path: "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Databricks/workspaces/{workspaceName}/virtualNetworkPeerings/{peeringName}",
     httpMethod: "DELETE",
     responses: {
@@ -3029,9 +3227,9 @@ const deleteOperationSpec$2 = {
         peeringName
     ],
     headerParameters: [accept],
-    serializer: serializer$5
+    serializer
 };
-const createOrUpdateOperationSpec$1 = {
+const createOrUpdateOperationSpec = {
     path: "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Databricks/workspaces/{workspaceName}/virtualNetworkPeerings/{peeringName}",
     httpMethod: "PUT",
     responses: {
@@ -3062,7 +3260,7 @@ const createOrUpdateOperationSpec$1 = {
     ],
     headerParameters: [accept, contentType],
     mediaType: "json",
-    serializer: serializer$5
+    serializer
 };
 const listByWorkspaceOperationSpec = {
     path: "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Databricks/workspaces/{workspaceName}/virtualNetworkPeerings",
@@ -3083,7 +3281,7 @@ const listByWorkspaceOperationSpec = {
         subscriptionId
     ],
     headerParameters: [accept],
-    serializer: serializer$5
+    serializer
 };
 const listByWorkspaceNextOperationSpec = {
     path: "{nextLink}",
@@ -3105,7 +3303,7 @@ const listByWorkspaceNextOperationSpec = {
         nextLink
     ],
     headerParameters: [accept],
-    serializer: serializer$5
+    serializer
 };
 
 /*
@@ -3115,7 +3313,7 @@ const listByWorkspaceNextOperationSpec = {
  * Code generated by Microsoft (R) AutoRest Code Generator.
  * Changes may cause incorrect behavior and will be lost if the code is regenerated.
  */
-class AzureDatabricksManagementClient extends coreClient.ServiceClient {
+class AzureDatabricksManagementClient extends coreClient__namespace.ServiceClient {
     /**
      * Initializes a new instance of the AzureDatabricksManagementClient class.
      * @param credentials Subscription credentials which uniquely identify client subscription.
@@ -3123,6 +3321,7 @@ class AzureDatabricksManagementClient extends coreClient.ServiceClient {
      * @param options The parameter options
      */
     constructor(credentials, subscriptionId, options) {
+        var _a, _b, _c;
         if (credentials === undefined) {
             throw new Error("'credentials' cannot be null");
         }
@@ -3137,17 +3336,35 @@ class AzureDatabricksManagementClient extends coreClient.ServiceClient {
             requestContentType: "application/json; charset=utf-8",
             credential: credentials
         };
-        const packageDetails = `azsdk-js-arm-databricks/3.0.0-beta.2`;
+        const packageDetails = `azsdk-js-arm-databricks/3.0.0-beta.4`;
         const userAgentPrefix = options.userAgentOptions && options.userAgentOptions.userAgentPrefix
             ? `${options.userAgentOptions.userAgentPrefix} ${packageDetails}`
             : `${packageDetails}`;
-        if (!options.credentialScopes) {
-            options.credentialScopes = ["https://management.azure.com/.default"];
-        }
         const optionsWithDefaults = Object.assign(Object.assign(Object.assign({}, defaults), options), { userAgentOptions: {
                 userAgentPrefix
-            }, baseUri: options.endpoint || "https://management.azure.com" });
+            }, endpoint: (_b = (_a = options.endpoint) !== null && _a !== void 0 ? _a : options.baseUri) !== null && _b !== void 0 ? _b : "https://management.azure.com" });
         super(optionsWithDefaults);
+        let bearerTokenAuthenticationPolicyFound = false;
+        if ((options === null || options === void 0 ? void 0 : options.pipeline) && options.pipeline.getOrderedPolicies().length > 0) {
+            const pipelinePolicies = options.pipeline.getOrderedPolicies();
+            bearerTokenAuthenticationPolicyFound = pipelinePolicies.some((pipelinePolicy) => pipelinePolicy.name ===
+                coreRestPipeline__namespace.bearerTokenAuthenticationPolicyName);
+        }
+        if (!options ||
+            !options.pipeline ||
+            options.pipeline.getOrderedPolicies().length == 0 ||
+            !bearerTokenAuthenticationPolicyFound) {
+            this.pipeline.removePolicy({
+                name: coreRestPipeline__namespace.bearerTokenAuthenticationPolicyName
+            });
+            this.pipeline.addPolicy(coreRestPipeline__namespace.bearerTokenAuthenticationPolicy({
+                credential: credentials,
+                scopes: (_c = optionsWithDefaults.credentialScopes) !== null && _c !== void 0 ? _c : `${optionsWithDefaults.endpoint}/.default`,
+                challengeCallbacks: {
+                    authorizeRequestOnChallenge: coreClient__namespace.authorizeRequestOnClaimChallenge
+                }
+            }));
+        }
         // Parameter assignments
         this.subscriptionId = subscriptionId;
         // Assigning values to Constant parameters
@@ -3159,8 +3376,37 @@ class AzureDatabricksManagementClient extends coreClient.ServiceClient {
         this.privateEndpointConnections = new PrivateEndpointConnectionsImpl(this);
         this.outboundNetworkDependenciesEndpoints = new OutboundNetworkDependenciesEndpointsImpl(this);
         this.vNetPeering = new VNetPeeringImpl(this);
+        this.addCustomApiVersionPolicy(options.apiVersion);
+    }
+    /** A function that adds a policy that sets the api-version (or equivalent) to reflect the library version. */
+    addCustomApiVersionPolicy(apiVersion) {
+        if (!apiVersion) {
+            return;
+        }
+        const apiVersionPolicy = {
+            name: "CustomApiVersionPolicy",
+            sendRequest(request, next) {
+                return tslib.__awaiter(this, void 0, void 0, function* () {
+                    const param = request.url.split("?");
+                    if (param.length > 1) {
+                        const newParams = param[1].split("&").map((item) => {
+                            if (item.indexOf("api-version") > -1) {
+                                return "api-version=" + apiVersion;
+                            }
+                            else {
+                                return item;
+                            }
+                        });
+                        request.url = param[0] + "?" + newParams.join("&");
+                    }
+                    return next(request);
+                });
+            }
+        };
+        this.pipeline.addPolicy(apiVersionPolicy);
     }
 }
 
 exports.AzureDatabricksManagementClient = AzureDatabricksManagementClient;
+exports.getContinuationToken = getContinuationToken;
 //# sourceMappingURL=index.js.map