@azure/cosmos 4.5.0 → 4.6.0

package/dist/commonjs/ClientContext.js

@@ -22,6 +22,7 @@ const CosmosDbDiagnosticLevel_js_1 = require("./diagnostics/CosmosDbDiagnosticLe
 const core_util_1 = require("@azure/core-util");
 const platform_js_1 = require("./common/platform.js");
 const partitionKeyRangeCache_js_1 = require("./routing/partitionKeyRangeCache.js");
+const constants_js_2 = require("./common/constants.js");
 const logger = (0, logger_1.createClientLogger)("ClientContext");
 const QueryJsonContentType = "application/query+json";
 const HttpHeaders = constants_js_1.Constants.HttpHeaders;
@@ -29,14 +30,27 @@ const HttpHeaders = constants_js_1.Constants.HttpHeaders;
  * @hidden
  */
 class ClientContext {
+    cosmosClientOptions;
+    globalEndpointManager;
+    clientConfig;
+    diagnosticLevel;
+    globalPartitionEndpointManager;
+    sessionContainer;
+    connectionPolicy;
+    pipeline;
+    diagnosticWriter;
+    diagnosticFormatter;
+    partitionKeyDefinitionCache; // TODO: PartitionKeyDefinitionCache
+    /** @internal */
+    partitionKeyRangeCache;
+    /** boolean flag to support operations with client-side encryption */
+    enableEncryption = false;
     constructor(cosmosClientOptions, globalEndpointManager, clientConfig, diagnosticLevel, globalPartitionEndpointManager) {
         this.cosmosClientOptions = cosmosClientOptions;
         this.globalEndpointManager = globalEndpointManager;
         this.clientConfig = clientConfig;
         this.diagnosticLevel = diagnosticLevel;
         this.globalPartitionEndpointManager = globalPartitionEndpointManager;
-        /** boolean flag to support operations with client-side encryption */
-        this.enableEncryption = false;
         if (cosmosClientOptions.clientEncryptionOptions) {
             this.enableEncryption = true;
         }
@@ -47,16 +61,40 @@ class ClientContext {
         if (cosmosClientOptions.aadCredentials) {
             this.pipeline = (0, core_rest_pipeline_1.createEmptyPipeline)();
             const hrefEndpoint = (0, checkURL_js_1.sanitizeEndpoint)(cosmosClientOptions.endpoint);
-            const scope = `${hrefEndpoint}/.default`;
+            // Use custom AAD scope if provided, otherwise use account-based scope
+            const accountScope = `${hrefEndpoint}/.default`;
+            const primaryScope = cosmosClientOptions.aadScope || accountScope;
+            const fallbackScope = constants_js_2.AAD_DEFAULT_SCOPE;
             this.pipeline.addPolicy((0, core_rest_pipeline_1.bearerTokenAuthenticationPolicy)({
                 credential: cosmosClientOptions.aadCredentials,
-                scopes: scope,
+                scopes: primaryScope,
                 challengeCallbacks: {
                     async authorizeRequest({ request, getAccessToken }) {
-                        const tokenResponse = await getAccessToken([scope], {});
-                        const AUTH_PREFIX = `type=aad&ver=1.0&sig=`;
-                        const authorizationToken = `${AUTH_PREFIX}${tokenResponse.token}`;
-                        request.headers.set("Authorization", authorizationToken);
+                        try {
+                            const tokenResponse = await getAccessToken([primaryScope], {});
+                            const authorizationToken = `${constants_js_2.AAD_AUTH_PREFIX}${tokenResponse.token}`;
+                            request.headers.set(constants_js_1.Constants.HttpHeaders.Authorization, authorizationToken);
+                        }
+                        catch (error) {
+                            // If no custom scope is provided and we get AADSTS500011 error,
+                            // fallback to the default Cosmos scope
+                            if (!cosmosClientOptions.aadScope &&
+                                error?.message?.includes(constants_js_2.AAD_RESOURCE_NOT_FOUND_ERROR)) {
+                                try {
+                                    const fallbackTokenResponse = await getAccessToken([fallbackScope], {});
+                                    const authorizationToken = `${constants_js_2.AAD_AUTH_PREFIX}${fallbackTokenResponse.token}`;
+                                    request.headers.set(constants_js_1.Constants.HttpHeaders.Authorization, authorizationToken);
+                                }
+                                catch (fallbackError) {
+                                    // If fallback also fails, throw the original error
+                                    throw error;
+                                }
+                            }
+                            else {
+                                // If custom scope is provided or error is not AADSTS500011, throw the original error
+                                throw error;
+                            }
+                        }
                     },
                 },
             }));
@@ -67,10 +105,16 @@ class ClientContext {
     /** @hidden */
     async read({ path, resourceType, resourceId, options = {}, partitionKey, diagnosticNode, partitionKeyRangeId, }) {
         try {
-            const request = Object.assign(Object.assign({}, this.getContextDerivedPropsForRequestCreation()), { method: constants_js_1.HTTPMethod.get, path, operationType: constants_js_1.OperationType.Read, resourceId,
+            const request = {
+                ...this.getContextDerivedPropsForRequestCreation(),
+                method: constants_js_1.HTTPMethod.get,
+                path,
+                operationType: constants_js_1.OperationType.Read,
+                resourceId,
                 options,
                 resourceType,
-                partitionKey });
+                partitionKey,
+            };
             diagnosticNode.addData({
                 operationType: constants_js_1.OperationType.Read,
                 resourceType,
@@ -85,7 +129,13 @@ class ClientContext {
             }
             this.applySessionToken(request);
             // read will use ReadEndpoint since it uses GET operation
-            request.endpoint = await this.globalEndpointManager.resolveServiceEndpoint(diagnosticNode, request.resourceType, request.operationType);
+            request.endpoint = await this.globalEndpointManager.resolveServiceEndpointInternal({
+                diagnosticNode: diagnosticNode,
+                resourceType: request.resourceType,
+                operationType: request.operationType,
+                startServiceEndpointIndex: 0,
+                excludedLocations: options?.excludedLocations,
+            });
             const response = await (0, Plugin_js_1.executePlugins)(diagnosticNode, request, RequestHandler_js_1.RequestHandler.request, Plugin_js_1.PluginOn.operation);
             this.captureSessionToken(undefined, path, constants_js_1.OperationType.Read, response.headers);
             return response;
@@ -98,10 +148,18 @@ class ClientContext {
     async queryFeed({ path, resourceType, resourceId, resultFn, query, options, diagnosticNode, partitionKeyRangeId, partitionKey, startEpk, endEpk, correlatedActivityId, }) {
         // Query operations will use ReadEndpoint even though it uses
         // GET(for queryFeed) and POST(for regular query operations)
-        const request = Object.assign(Object.assign({}, this.getContextDerivedPropsForRequestCreation()), { method: constants_js_1.HTTPMethod.get, path, operationType: constants_js_1.OperationType.Query, partitionKeyRangeId,
+        const request = {
+            ...this.getContextDerivedPropsForRequestCreation(),
+            method: constants_js_1.HTTPMethod.get,
+            path,
+            operationType: constants_js_1.OperationType.Query,
+            partitionKeyRangeId,
             resourceId,
             resourceType,
-            options, body: query, partitionKey });
+            options,
+            body: query,
+            partitionKey,
+        };
         diagnosticNode.addData({
             operationType: constants_js_1.OperationType.Query,
             resourceType,
@@ -110,7 +168,13 @@ class ClientContext {
         if (query !== undefined) {
             request.method = constants_js_1.HTTPMethod.post;
         }
-        request.endpoint = await this.globalEndpointManager.resolveServiceEndpoint(diagnosticNode, request.resourceType, request.operationType);
+        request.endpoint = await this.globalEndpointManager.resolveServiceEndpointInternal({
+            diagnosticNode: diagnosticNode,
+            resourceType: request.resourceType,
+            operationType: request.operationType,
+            startServiceEndpointIndex: 0,
+            excludedLocations: options?.excludedLocations,
+        });
         request.headers = await this.buildHeaders(request);
         if (startEpk !== undefined && endEpk !== undefined) {
             request.headers[HttpHeaders.StartEpk] = startEpk;
@@ -140,14 +204,27 @@ class ClientContext {
         return this.processQueryFeedResponse(response, !!query, resultFn);
     }
     async getQueryPlan(path, resourceType, resourceId, query, options = {}, diagnosticNode, correlatedActivityId) {
-        const request = Object.assign(Object.assign({}, this.getContextDerivedPropsForRequestCreation()), { method: constants_js_1.HTTPMethod.post, path, operationType: constants_js_1.OperationType.Read, resourceId,
+        const request = {
+            ...this.getContextDerivedPropsForRequestCreation(),
+            method: constants_js_1.HTTPMethod.post,
+            path,
+            operationType: constants_js_1.OperationType.Read,
+            resourceId,
             resourceType,
-            options, body: query });
+            options,
+            body: query,
+        };
         diagnosticNode.addData({
             operationType: constants_js_1.OperationType.Read,
             resourceType,
         });
-        request.endpoint = await this.globalEndpointManager.resolveServiceEndpoint(diagnosticNode, request.resourceType, request.operationType);
+        request.endpoint = await this.globalEndpointManager.resolveServiceEndpointInternal({
+            diagnosticNode: diagnosticNode,
+            resourceType: request.resourceType,
+            operationType: request.operationType,
+            startServiceEndpointIndex: 0,
+            excludedLocations: options?.excludedLocations,
+        });
         request.headers = await this.buildHeaders(request);
         if (correlatedActivityId !== undefined) {
             request.headers[HttpHeaders.CorrelatedActivityId] = correlatedActivityId;
@@ -183,11 +260,16 @@ class ClientContext {
     }
     async delete({ path, resourceType, resourceId, options = {}, partitionKey, method = constants_js_1.HTTPMethod.delete, diagnosticNode, partitionKeyRangeId, }) {
         try {
-            const request = Object.assign(Object.assign({}, this.getContextDerivedPropsForRequestCreation()), { method: method, operationType: constants_js_1.OperationType.Delete, path,
+            const request = {
+                ...this.getContextDerivedPropsForRequestCreation(),
+                method: method,
+                operationType: constants_js_1.OperationType.Delete,
+                path,
                 resourceType,
                 options,
                 resourceId,
-                partitionKey });
+                partitionKey,
+            };
             diagnosticNode.addData({
                 operationType: constants_js_1.OperationType.Delete,
                 resourceType,
@@ -196,7 +278,13 @@ class ClientContext {
             request.partitionKeyRangeId = partitionKeyRangeId;
             this.applySessionToken(request);
             // deleteResource will use WriteEndpoint since it uses DELETE operation
-            request.endpoint = await this.globalEndpointManager.resolveServiceEndpoint(diagnosticNode, request.resourceType, request.operationType);
+            request.endpoint = await this.globalEndpointManager.resolveServiceEndpointInternal({
+                diagnosticNode: diagnosticNode,
+                resourceType: request.resourceType,
+                operationType: request.operationType,
+                startServiceEndpointIndex: 0,
+                excludedLocations: options?.excludedLocations,
+            });
             const response = await (0, Plugin_js_1.executePlugins)(diagnosticNode, request, RequestHandler_js_1.RequestHandler.request, Plugin_js_1.PluginOn.operation);
             if ((0, helper_js_1.parseLink)(path).type !== "colls") {
                 this.captureSessionToken(undefined, path, constants_js_1.OperationType.Delete, response.headers);
@@ -213,12 +301,17 @@ class ClientContext {
     }
     async patch({ body, path, resourceType, resourceId, options = {}, partitionKey, diagnosticNode, partitionKeyRangeId, }) {
         try {
-            const request = Object.assign(Object.assign({}, this.getContextDerivedPropsForRequestCreation()), { method: constants_js_1.HTTPMethod.patch, operationType: constants_js_1.OperationType.Patch, path,
+            const request = {
+                ...this.getContextDerivedPropsForRequestCreation(),
+                method: constants_js_1.HTTPMethod.patch,
+                operationType: constants_js_1.OperationType.Patch,
+                path,
                 resourceType,
                 body,
                 resourceId,
                 options,
-                partitionKey });
+                partitionKey,
+            };
             diagnosticNode.addData({
                 operationType: constants_js_1.OperationType.Patch,
                 resourceType,
@@ -227,7 +320,13 @@ class ClientContext {
             request.partitionKeyRangeId = partitionKeyRangeId;
             this.applySessionToken(request);
             // patch will use WriteEndpoint
-            request.endpoint = await this.globalEndpointManager.resolveServiceEndpoint(diagnosticNode, request.resourceType, request.operationType);
+            request.endpoint = await this.globalEndpointManager.resolveServiceEndpointInternal({
+                diagnosticNode: diagnosticNode,
+                resourceType: request.resourceType,
+                operationType: request.operationType,
+                startServiceEndpointIndex: 0,
+                excludedLocations: options?.excludedLocations,
+            });
             const response = await (0, Plugin_js_1.executePlugins)(diagnosticNode, request, RequestHandler_js_1.RequestHandler.request, Plugin_js_1.PluginOn.operation);
             this.captureSessionToken(undefined, path, constants_js_1.OperationType.Patch, response.headers);
             return response;
@@ -239,12 +338,17 @@ class ClientContext {
     }
     async create({ body, path, resourceType, resourceId, diagnosticNode, options = {}, partitionKey, partitionKeyRangeId, }) {
         try {
-            const request = Object.assign(Object.assign({}, this.getContextDerivedPropsForRequestCreation()), { method: constants_js_1.HTTPMethod.post, operationType: constants_js_1.OperationType.Create, path,
+            const request = {
+                ...this.getContextDerivedPropsForRequestCreation(),
+                method: constants_js_1.HTTPMethod.post,
+                operationType: constants_js_1.OperationType.Create,
+                path,
                 resourceType,
                 resourceId,
                 body,
                 options,
-                partitionKey });
+                partitionKey,
+            };
             diagnosticNode.addData({
                 operationType: constants_js_1.OperationType.Create,
                 resourceType,
@@ -253,7 +357,13 @@ class ClientContext {
             request.partitionKeyRangeId = partitionKeyRangeId;
             // create will use WriteEndpoint since it uses POST operation
             this.applySessionToken(request);
-            request.endpoint = await this.globalEndpointManager.resolveServiceEndpoint(diagnosticNode, request.resourceType, request.operationType);
+            request.endpoint = await this.globalEndpointManager.resolveServiceEndpointInternal({
+                diagnosticNode: diagnosticNode,
+                resourceType: request.resourceType,
+                operationType: request.operationType,
+                startServiceEndpointIndex: 0,
+                excludedLocations: options?.excludedLocations,
+            });
             const response = await (0, Plugin_js_1.executePlugins)(diagnosticNode, request, RequestHandler_js_1.RequestHandler.request, Plugin_js_1.PluginOn.operation);
             this.captureSessionToken(undefined, path, constants_js_1.OperationType.Create, response.headers);
             return response;
@@ -301,12 +411,17 @@ class ClientContext {
     }
     async replace({ body, path, resourceType, resourceId, options = {}, partitionKey, diagnosticNode, partitionKeyRangeId, }) {
         try {
-            const request = Object.assign(Object.assign({}, this.getContextDerivedPropsForRequestCreation()), { method: constants_js_1.HTTPMethod.put, operationType: constants_js_1.OperationType.Replace, path,
+            const request = {
+                ...this.getContextDerivedPropsForRequestCreation(),
+                method: constants_js_1.HTTPMethod.put,
+                operationType: constants_js_1.OperationType.Replace,
+                path,
                 resourceType,
                 body,
                 resourceId,
                 options,
-                partitionKey });
+                partitionKey,
+            };
             diagnosticNode.addData({
                 operationType: constants_js_1.OperationType.Replace,
                 resourceType,
@@ -315,7 +430,13 @@ class ClientContext {
             request.partitionKeyRangeId = partitionKeyRangeId;
             this.applySessionToken(request);
             // replace will use WriteEndpoint since it uses PUT operation
-            request.endpoint = await this.globalEndpointManager.resolveServiceEndpoint(diagnosticNode, request.resourceType, request.operationType);
+            request.endpoint = await this.globalEndpointManager.resolveServiceEndpointInternal({
+                diagnosticNode: diagnosticNode,
+                resourceType: request.resourceType,
+                operationType: request.operationType,
+                startServiceEndpointIndex: 0,
+                excludedLocations: options?.excludedLocations,
+            });
             const response = await (0, Plugin_js_1.executePlugins)(diagnosticNode, request, RequestHandler_js_1.RequestHandler.request, Plugin_js_1.PluginOn.operation);
             this.captureSessionToken(undefined, path, constants_js_1.OperationType.Replace, response.headers);
             return response;
@@ -327,12 +448,17 @@ class ClientContext {
     }
     async upsert({ body, path, resourceType, resourceId, options = {}, partitionKey, diagnosticNode, partitionKeyRangeId, }) {
         try {
-            const request = Object.assign(Object.assign({}, this.getContextDerivedPropsForRequestCreation()), { method: constants_js_1.HTTPMethod.post, operationType: constants_js_1.OperationType.Upsert, path,
+            const request = {
+                ...this.getContextDerivedPropsForRequestCreation(),
+                method: constants_js_1.HTTPMethod.post,
+                operationType: constants_js_1.OperationType.Upsert,
+                path,
                 resourceType,
                 body,
                 resourceId,
                 options,
-                partitionKey });
+                partitionKey,
+            };
             diagnosticNode.addData({
                 operationType: constants_js_1.OperationType.Upsert,
                 resourceType,
@@ -342,7 +468,13 @@ class ClientContext {
             request.headers[HttpHeaders.IsUpsert] = true;
             this.applySessionToken(request);
             // upsert will use WriteEndpoint since it uses POST operation
-            request.endpoint = await this.globalEndpointManager.resolveServiceEndpoint(diagnosticNode, request.resourceType, request.operationType);
+            request.endpoint = await this.globalEndpointManager.resolveServiceEndpointInternal({
+                diagnosticNode: diagnosticNode,
+                resourceType: request.resourceType,
+                operationType: request.operationType,
+                startServiceEndpointIndex: 0,
+                excludedLocations: options?.excludedLocations,
+            });
             const response = await (0, Plugin_js_1.executePlugins)(diagnosticNode, request, RequestHandler_js_1.RequestHandler.request, Plugin_js_1.PluginOn.operation);
             this.captureSessionToken(undefined, path, constants_js_1.OperationType.Upsert, response.headers);
             return response;
@@ -360,7 +492,17 @@ class ClientContext {
         }
         const path = (0, helper_js_1.getPathFromLink)(sprocLink);
         const id = (0, helper_js_1.getIdFromLink)(sprocLink);
-        const request = Object.assign(Object.assign({}, this.getContextDerivedPropsForRequestCreation()), { method: constants_js_1.HTTPMethod.post, operationType: constants_js_1.OperationType.Execute, path, resourceType: constants_js_1.ResourceType.sproc, options, resourceId: id, body: params, partitionKey });
+        const request = {
+            ...this.getContextDerivedPropsForRequestCreation(),
+            method: constants_js_1.HTTPMethod.post,
+            operationType: constants_js_1.OperationType.Execute,
+            path,
+            resourceType: constants_js_1.ResourceType.sproc,
+            options,
+            resourceId: id,
+            body: params,
+            partitionKey,
+        };
         diagnosticNode.addData({
             operationType: constants_js_1.OperationType.Execute,
             resourceType: constants_js_1.ResourceType.sproc,
@@ -368,7 +510,13 @@ class ClientContext {
         request.headers = await this.buildHeaders(request);
         request.partitionKeyRangeId = partitionKeyRangeId;
         // executeStoredProcedure will use WriteEndpoint since it uses POST operation
-        request.endpoint = await this.globalEndpointManager.resolveServiceEndpoint(diagnosticNode, request.resourceType, request.operationType);
+        request.endpoint = await this.globalEndpointManager.resolveServiceEndpointInternal({
+            diagnosticNode: diagnosticNode,
+            resourceType: request.resourceType,
+            operationType: request.operationType,
+            startServiceEndpointIndex: 0,
+            excludedLocations: options?.excludedLocations,
+        });
         const response = await (0, Plugin_js_1.executePlugins)(diagnosticNode, request, RequestHandler_js_1.RequestHandler.request, Plugin_js_1.PluginOn.operation);
         return response;
     }
@@ -379,7 +527,15 @@ class ClientContext {
      */
     async getDatabaseAccount(diagnosticNode, options = {}) {
         const endpoint = options.urlConnection || this.cosmosClientOptions.endpoint;
-        const request = Object.assign(Object.assign({}, this.getContextDerivedPropsForRequestCreation()), { endpoint, method: constants_js_1.HTTPMethod.get, operationType: constants_js_1.OperationType.Read, path: "", resourceType: constants_js_1.ResourceType.none, options });
+        const request = {
+            ...this.getContextDerivedPropsForRequestCreation(),
+            endpoint,
+            method: constants_js_1.HTTPMethod.get,
+            operationType: constants_js_1.OperationType.Read,
+            path: "",
+            resourceType: constants_js_1.ResourceType.none,
+            options,
+        };
         diagnosticNode.addData({
             operationType: constants_js_1.OperationType.Read,
             resourceType: constants_js_1.ResourceType.none,
@@ -410,10 +566,17 @@ class ClientContext {
     }
     async batch({ body, path, partitionKey, resourceId, options = {}, diagnosticNode, partitionKeyRangeId, }) {
         try {
-            const request = Object.assign(Object.assign({}, this.getContextDerivedPropsForRequestCreation()), { method: constants_js_1.HTTPMethod.post, operationType: constants_js_1.OperationType.Batch, path,
-                body, resourceType: constants_js_1.ResourceType.item, resourceId,
+            const request = {
+                ...this.getContextDerivedPropsForRequestCreation(),
+                method: constants_js_1.HTTPMethod.post,
+                operationType: constants_js_1.OperationType.Batch,
+                path,
+                body,
+                resourceType: constants_js_1.ResourceType.item,
+                resourceId,
                 options,
-                partitionKey });
+                partitionKey,
+            };
             diagnosticNode.addData({
                 operationType: constants_js_1.OperationType.Batch,
                 resourceType: constants_js_1.ResourceType.item,
@@ -423,7 +586,13 @@ class ClientContext {
             request.headers[HttpHeaders.IsBatchRequest] = true;
             request.headers[HttpHeaders.IsBatchAtomic] = true;
             this.applySessionToken(request);
-            request.endpoint = await this.globalEndpointManager.resolveServiceEndpoint(diagnosticNode, request.resourceType, request.operationType);
+            request.endpoint = await this.globalEndpointManager.resolveServiceEndpointInternal({
+                diagnosticNode: diagnosticNode,
+                resourceType: request.resourceType,
+                operationType: request.operationType,
+                startServiceEndpointIndex: 0,
+                excludedLocations: options?.excludedLocations,
+            });
             const response = await (0, Plugin_js_1.executePlugins)(diagnosticNode, request, RequestHandler_js_1.RequestHandler.request, Plugin_js_1.PluginOn.operation);
             this.captureSessionToken(undefined, path, constants_js_1.OperationType.Batch, response.headers);
             response.diagnostics = diagnosticNode.toDiagnostic(this.getClientConfig());
@@ -435,11 +604,17 @@ class ClientContext {
         }
     }
     async bulk({ body, path, partitionKeyRangeId, resourceId, bulkOptions = {}, options = {}, diagnosticNode, }) {
-        var _a;
         try {
-            const request = Object.assign(Object.assign({}, this.getContextDerivedPropsForRequestCreation()), { method: constants_js_1.HTTPMethod.post, operationType: constants_js_1.OperationType.Batch, path,
-                body, resourceType: constants_js_1.ResourceType.item, resourceId,
-                options });
+            const request = {
+                ...this.getContextDerivedPropsForRequestCreation(),
+                method: constants_js_1.HTTPMethod.post,
+                operationType: constants_js_1.OperationType.Batch,
+                path,
+                body,
+                resourceType: constants_js_1.ResourceType.item,
+                resourceId,
+                options,
+            };
             diagnosticNode.addData({
                 operationType: constants_js_1.OperationType.Batch,
                 resourceType: constants_js_1.ResourceType.item,
@@ -449,9 +624,15 @@ class ClientContext {
             request.headers[HttpHeaders.IsBatchRequest] = true;
             request.headers[HttpHeaders.PartitionKeyRangeID] = partitionKeyRangeId;
             request.headers[HttpHeaders.IsBatchAtomic] = false;
-            request.headers[HttpHeaders.BatchContinueOnError] = (_a = bulkOptions.continueOnError) !== null && _a !== void 0 ? _a : true;
+            request.headers[HttpHeaders.BatchContinueOnError] = bulkOptions.continueOnError ?? true;
             this.applySessionToken(request);
-            request.endpoint = await this.globalEndpointManager.resolveServiceEndpoint(diagnosticNode, request.resourceType, request.operationType);
+            request.endpoint = await this.globalEndpointManager.resolveServiceEndpointInternal({
+                diagnosticNode: diagnosticNode,
+                resourceType: request.resourceType,
+                operationType: request.operationType,
+                startServiceEndpointIndex: 0,
+                excludedLocations: options?.excludedLocations,
+            });
             const response = await (0, Plugin_js_1.executePlugins)(diagnosticNode, request, RequestHandler_js_1.RequestHandler.request, Plugin_js_1.PluginOn.operation);
             this.captureSessionToken(undefined, path, constants_js_1.OperationType.Batch, response.headers);
             return response;
@@ -521,7 +702,10 @@ class ClientContext {
     buildHeaders(requestContext) {
         return (0, request_js_1.getHeaders)({
             clientOptions: this.cosmosClientOptions,
-            defaultHeaders: Object.assign(Object.assign({}, this.cosmosClientOptions.defaultHeaders), requestContext.options.initialHeaders),
+            defaultHeaders: {
+                ...this.cosmosClientOptions.defaultHeaders,
+                ...requestContext.options.initialHeaders,
+            },
             verb: requestContext.method,
             path: requestContext.path,
             resourceId: requestContext.resourceId,
@@ -574,8 +758,8 @@ class ClientContext {
      * @internal
      */
     isPartitionLevelFailOverEnabled() {
-        return (this.connectionPolicy.enablePartitionLevelFailover ||
-            this.connectionPolicy.enablePartitionLevelCircuitBreaker);
+        return (this.globalEndpointManager.enablePartitionLevelFailover ||
+            this.globalEndpointManager.enablePartitionLevelCircuitBreaker);
     }
 }
 exports.ClientContext = ClientContext;