Fixing CI / live tests pipeline.

sdk/cosmos/azure-cosmos-test/src/main/java/com/azure/cosmos/test/implementation/faultinjection/FaultInjectionRuleProcessor.java

@@ -23,6 +23,7 @@
 import com.azure.cosmos.implementation.directconnectivity.AddressSelector;
 import com.azure.cosmos.implementation.directconnectivity.rntbd.RntbdUtils;
 import com.azure.cosmos.implementation.feedranges.FeedRangeInternal;
+import com.azure.cosmos.implementation.routing.LocationCache;
 import com.azure.cosmos.implementation.routing.PartitionKeyRangeIdentity;
 import com.azure.cosmos.test.faultinjection.FaultInjectionCondition;
 import com.azure.cosmos.test.faultinjection.FaultInjectionConnectionErrorResult;
@@ -447,6 +448,7 @@ private Mono<List<URI>> resolvePhysicalAddresses(
                                     null);
 
                                 faultInjectionAddressRequest.requestContext.locationEndpointToRoute = regionEndpoint;
+                                faultInjectionAddressRequest.requestContext.consolidatedRegionalEndpointToRoute = new LocationCache.ConsolidatedRegionalEndpoint(regionEndpoint, null);
                                 faultInjectionAddressRequest.setPartitionKeyRangeIdentity(new PartitionKeyRangeIdentity(pkRangeId));
 
                                 if (isWriteOnly) {

sdk/cosmos/azure-cosmos-tests/src/test/java/com/azure/cosmos/implementation/RegionScopedSessionContainerTest.java

@@ -1774,6 +1774,7 @@ private static RxDocumentServiceRequest createRequestEntity(OperationType operat
             resourceType);
 
         request.requestContext.locationEndpointToRoute = locationEndpointToRoute;
+        request.requestContext.consolidatedRegionalEndpointToRoute = new LocationCache.ConsolidatedRegionalEndpoint(locationEndpointToRoute, null);
 
         return request;
     }

sdk/cosmos/azure-cosmos/src/main/java/com/azure/cosmos/implementation/DocumentServiceRequestContext.java

@@ -146,6 +146,7 @@ public DocumentServiceRequestContext clone() {
         context.usePreferredLocations = this.usePreferredLocations;
         context.locationIndexToRoute = this.locationIndexToRoute;
         context.locationEndpointToRoute = this.locationEndpointToRoute;
+        context.consolidatedRegionalEndpointToRoute = this.consolidatedRegionalEndpointToRoute;
         context.performLocalRefreshOnGoneException = this.performLocalRefreshOnGoneException;
         context.effectivePartitionKey = this.effectivePartitionKey;
         context.performedBackgroundAddressRefresh = this.performedBackgroundAddressRefresh;

sdk/cosmos/azure-cosmos/src/main/java/com/azure/cosmos/implementation/directconnectivity/rntbd/RntbdOpenConnectionsHandler.java

@@ -10,6 +10,7 @@
 import com.azure.cosmos.implementation.RxDocumentServiceRequest;
 import com.azure.cosmos.implementation.apachecommons.lang.StringUtils;
 import com.azure.cosmos.implementation.directconnectivity.Uri;
+import com.azure.cosmos.implementation.routing.LocationCache;
 import org.slf4j.Logger;
 import org.slf4j.LoggerFactory;
 import reactor.core.publisher.Flux;
@@ -77,6 +78,7 @@ private RxDocumentServiceRequest getOpenConnectionRequest(String collectionRid,
         RxDocumentServiceRequest openConnectionRequest =
             RxDocumentServiceRequest.create(null, OperationType.Create, ResourceType.Connection);
         openConnectionRequest.requestContext.locationEndpointToRoute = serviceEndpoint;
+        openConnectionRequest.requestContext.consolidatedRegionalEndpointToRoute = new LocationCache.ConsolidatedRegionalEndpoint(serviceEndpoint, null);
         openConnectionRequest.requestContext.resolvedCollectionRid = collectionRid;
         openConnectionRequest.faultInjectionRequestContext.setLocationEndpointToRoute(serviceEndpoint);
 

sdk/cosmos/azure-cosmos/src/main/java/com/azure/cosmos/implementation/faultinjection/FaultInjectionRequestContext.java

@@ -3,6 +3,8 @@
 
 package com.azure.cosmos.implementation.faultinjection;
 
+import com.azure.cosmos.implementation.routing.LocationCache;
+
 import java.net.URI;
 import java.util.ArrayList;
 import java.util.List;
@@ -23,6 +25,7 @@ public class FaultInjectionRequestContext {
     private final AtomicBoolean addressForceRefreshed;
 
     private volatile URI locationEndpointToRoute;
+    private volatile LocationCache.ConsolidatedRegionalEndpoint consolidatedRegionalEndpointToRoute;
 
     /***
      * This usually is called during retries.
@@ -96,6 +99,7 @@ public String getFaultInjectionRuleId(long transportRequestId) {
 
     public void setLocationEndpointToRoute(URI locationEndpointToRoute) {
         this.locationEndpointToRoute = locationEndpointToRoute;
+        this.consolidatedRegionalEndpointToRoute = new LocationCache.ConsolidatedRegionalEndpoint(locationEndpointToRoute, null);
     }
 
     public URI getLocationEndpointToRoute() {