add default endpoint region in the condition when applicable (Azure#34078)

Co-authored-by: annie-mac <[email protected]>

Author: xinlian12

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

@@ -37,6 +37,7 @@
 import reactor.core.publisher.Mono;
 
 import java.net.URI;
+import java.util.ArrayList;
 import java.util.Arrays;
 import java.util.Collections;
 import java.util.List;
@@ -157,6 +158,13 @@ private Mono<IFaultInjectionRuleInternal> getEffectiveServerErrorRule(
                 List<URI> regionEndpoints = this.getRegionEndpoints(rule.getCondition());
                 effectiveCondition.setRegionEndpoints(regionEndpoints);
 
+                if (StringUtils.isEmpty(rule.getCondition().getRegion())) {
+                    // if region is not specific configured, then also add the defaultEndpoint
+                    List<URI> regionEndpointsWithDefault = new ArrayList<>(regionEndpoints);
+                    regionEndpointsWithDefault.add(this.globalEndpointManager.getDefaultEndpoint());
+                    effectiveCondition.setRegionEndpoints(regionEndpointsWithDefault);
+                }
+
                 // TODO: add handling for gateway mode
 
                 // Direct connection mode, populate physical addresses

sdk/cosmos/azure-cosmos-test/src/test/java/com/azure/cosmos/test/faultinjection/FaultInjectionServerErrorRuleTests.java

@@ -4,6 +4,7 @@
 package com.azure.cosmos.test.faultinjection;
 
 import com.azure.cosmos.BridgeInternal;
+import com.azure.cosmos.ConsistencyLevel;
 import com.azure.cosmos.CosmosAsyncClient;
 import com.azure.cosmos.CosmosAsyncContainer;
 import com.azure.cosmos.CosmosClientBuilder;
@@ -147,8 +148,7 @@ public void faultInjectionServerErrorRuleTests_OperationType(OperationType opera
             CosmosFaultInjectionHelper.configureFaultInjectionRules(cosmosAsyncContainer, Arrays.asList(serverGoneErrorRule)).block();
 
             assertThat(serverGoneErrorRule.getAddresses().size()).isZero();
-            assertThat(serverGoneErrorRule.getRegionEndpoints().size()).isEqualTo(readRegionMap.size());
-            assertThat(serverGoneErrorRule.getRegionEndpoints().size() == this.readRegionMap.size()
+            assertThat(serverGoneErrorRule.getRegionEndpoints().size() == this.readRegionMap.size() + 1
                && serverGoneErrorRule.getRegionEndpoints().containsAll(this.readRegionMap.values()));
 
             CosmosDiagnostics cosmosDiagnostics = this.performDocumentOperation(cosmosAsyncContainer, operationType, createdItem);
@@ -164,8 +164,7 @@ public void faultInjectionServerErrorRuleTests_OperationType(OperationType opera
             serverGoneErrorRule.disable();
             CosmosFaultInjectionHelper.configureFaultInjectionRules(cosmosAsyncContainer, Arrays.asList(serverTooManyRequestsErrorRule)).block();
             assertThat(serverGoneErrorRule.getAddresses().size()).isZero();
-            assertThat(serverGoneErrorRule.getRegionEndpoints().size()).isEqualTo(readRegionMap.size());
-            assertThat(serverGoneErrorRule.getRegionEndpoints().size() == this.readRegionMap.size()
+            assertThat(serverGoneErrorRule.getRegionEndpoints().size() == this.readRegionMap.size() + 1
                 && serverGoneErrorRule.getRegionEndpoints().containsAll(this.readRegionMap.values()));
 
             cosmosDiagnostics = this.performDocumentOperation(cosmosAsyncContainer, operationType, createdItem);
@@ -250,6 +249,7 @@ public void faultInjectionServerErrorRuleTests_OperationTypeImpactAddresses(Oper
                 .key(TestConfigurations.MASTER_KEY)
                 .endpoint(TestConfigurations.HOST)
                 .preferredRegions(preferredRegionList)
+                .consistencyLevel(ConsistencyLevel.EVENTUAL)
                 .buildAsyncClient();
 
             CosmosAsyncContainer container =
@@ -258,7 +258,7 @@ public void faultInjectionServerErrorRuleTests_OperationTypeImpactAddresses(Oper
                     .getContainer(this.cosmosAsyncContainer.getId());
 
             CosmosFaultInjectionHelper.configureFaultInjectionRules(container, Arrays.asList(writeRegionServerGoneErrorRule)).block();
-            assertThat(writeRegionServerGoneErrorRule.getRegionEndpoints().size()).isEqualTo(1);
+            assertThat(writeRegionServerGoneErrorRule.getRegionEndpoints().size()).isEqualTo(2);
 
             CosmosDiagnostics cosmosDiagnostics = this.performDocumentOperation(container, operationType, createdItem);
             if (operationType.isWriteOperation()) {

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

@@ -127,6 +127,10 @@ public URI resolveFaultInjectionServiceEndpoint(String region, boolean writeOnly
         return this.locationCache.resolveFaultInjectionEndpoint(region, writeOnly);
     }
 
+    public URI getDefaultEndpoint() {
+        return this.locationCache.getDefaultEndpoint();
+    }
+
     public void markEndpointUnavailableForRead(URI endpoint) {
         logger.debug("Marking endpoint {} unavailable for read",endpoint);
         this.locationCache.markEndpointUnavailableForRead(endpoint);;

sdk/cosmos/azure-cosmos/src/main/java/com/azure/cosmos/implementation/routing/LocationCache.java

@@ -217,6 +217,10 @@ public URI resolveFaultInjectionEndpoint(String region, boolean writeOnly) {
         throw new IllegalArgumentException("Can not find service endpoint for region " + region);
     }
 
+    public URI getDefaultEndpoint() {
+        return this.defaultEndpoint;
+    }
+
     public boolean shouldRefreshEndpoints(Utils.ValueHolder<Boolean> canRefreshInBackground) {
         canRefreshInBackground.v = true;
         DatabaseAccountLocationsInfo currentLocationInfo = this.locationInfo;