Owner Resource Not Found (Gateway V2 Connection Mode and Direct Connection Mode): Remaps sub-status to 1003 for requests to child resources against non-existent container.

sdk/cosmos/azure-cosmos-tests/src/test/java/com/azure/cosmos/faultinjection/FaultInjectionTestBase.java

@@ -10,18 +10,25 @@
 import com.azure.cosmos.TestObject;
 import com.azure.cosmos.implementation.OperationType;
 import com.azure.cosmos.models.CosmosBatch;
+import com.azure.cosmos.models.CosmosBulkOperationResponse;
+import com.azure.cosmos.models.CosmosBulkOperations;
 import com.azure.cosmos.models.CosmosChangeFeedRequestOptions;
 import com.azure.cosmos.models.CosmosItemIdentity;
+import com.azure.cosmos.models.CosmosItemOperation;
 import com.azure.cosmos.models.CosmosPatchOperations;
 import com.azure.cosmos.models.CosmosQueryRequestOptions;
 import com.azure.cosmos.models.FeedRange;
 import com.azure.cosmos.models.FeedResponse;
 import com.azure.cosmos.models.PartitionKey;
 import com.azure.cosmos.rx.TestSuiteBase;
+import reactor.core.publisher.Flux;
 
+import java.util.ArrayList;
 import java.util.Arrays;
 import java.util.List;
 
+import static org.assertj.core.api.Assertions.assertThat;
+
 public abstract class FaultInjectionTestBase extends TestSuiteBase {
     public FaultInjectionTestBase(CosmosClientBuilder cosmosClientBuilder) {
         super(cosmosClientBuilder);
@@ -32,6 +39,24 @@ protected CosmosDiagnostics performDocumentOperation(
         OperationType operationType,
         TestObject createdItem,
         boolean isReadMany) {
+
+        return performDocumentOperation(
+            cosmosAsyncContainer,
+            operationType,
+            createdItem,
+            isReadMany,
+            true,
+            false);
+    }
+
+    protected CosmosDiagnostics performDocumentOperation(
+        CosmosAsyncContainer cosmosAsyncContainer,
+        OperationType operationType,
+        TestObject createdItem,
+        boolean isReadMany,
+        boolean fetchFeedRangesBeforehandForChangeFeed,
+        boolean isBulkOperation) {
+
         try {
             if (operationType == OperationType.Query && !isReadMany) {
                 CosmosQueryRequestOptions queryRequestOptions = new CosmosQueryRequestOptions();
@@ -87,19 +112,56 @@ protected CosmosDiagnostics performDocumentOperation(
                 }
 
                 if (operationType == OperationType.Batch) {
-                    CosmosBatch batch = CosmosBatch.createCosmosBatch(new PartitionKey(createdItem.getId()));
 
-                    batch.upsertItemOperation(createdItem);
-                    batch.readItemOperation(createdItem.getId());
+                    if (isBulkOperation) {
+
+                        List<CosmosItemOperation> cosmosItemOperations = new ArrayList<>();
+
+                        CosmosItemOperation cosmosItemOperation = CosmosBulkOperations.getReadItemOperation(
+                            createdItem.getId(),
+                            new PartitionKey(createdItem.getId()),
+                            TestObject.class);
+
+                        cosmosItemOperations.add(cosmosItemOperation);
 
-                    return cosmosAsyncContainer.executeCosmosBatch(batch).block().getDiagnostics();
+                        Flux<CosmosItemOperation> operationsFlux = Flux.fromIterable(cosmosItemOperations);
+
+                        CosmosBulkOperationResponse<Object> response = cosmosAsyncContainer.executeBulkOperations(operationsFlux).blockLast();
+
+                        assertThat(response).isNotNull();
+
+                        return response.getResponse().getCosmosDiagnostics();
+                    } else {
+                        CosmosBatch batch = CosmosBatch.createCosmosBatch(new PartitionKey(createdItem.getId()));
+
+                        batch.upsertItemOperation(createdItem);
+                        batch.readItemOperation(createdItem.getId());
+
+                        return cosmosAsyncContainer.executeCosmosBatch(batch).block().getDiagnostics();
+                    }
                 }
             }
 
             if (operationType == OperationType.ReadFeed) {
-                List<FeedRange> feedRanges = cosmosAsyncContainer.getFeedRanges().block();
+
+                if (fetchFeedRangesBeforehandForChangeFeed) {
+                    List<FeedRange> feedRanges = cosmosAsyncContainer.getFeedRanges().block();
+
+                    assertThat(feedRanges).isNotNull();
+                    assertThat(feedRanges.size()).isGreaterThan(0);
+
+                    CosmosChangeFeedRequestOptions changeFeedRequestOptions =
+                        CosmosChangeFeedRequestOptions.createForProcessingFromBeginning(feedRanges.get(0));
+
+                    FeedResponse<TestObject> firstPage =  cosmosAsyncContainer
+                        .queryChangeFeed(changeFeedRequestOptions, TestObject.class)
+                        .byPage()
+                        .blockFirst();
+                    return firstPage.getCosmosDiagnostics();
+                }
+
                 CosmosChangeFeedRequestOptions changeFeedRequestOptions =
-                    CosmosChangeFeedRequestOptions.createForProcessingFromBeginning(feedRanges.get(0));
+                    CosmosChangeFeedRequestOptions.createForProcessingFromBeginning(FeedRange.forFullRange());
 
                 FeedResponse<TestObject> firstPage =  cosmosAsyncContainer
                     .queryChangeFeed(changeFeedRequestOptions, TestObject.class)

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

@@ -38,7 +38,7 @@ public void staledException(int statusCode, int subStatusCode, boolean expectRet
 
         RxCollectionCache rxCollectionCache = Mockito.mock(RxCollectionCache.class);
         Mockito
-            .when(rxCollectionCache.resolveByNameAsync(Mockito.any(), Mockito.any(), Mockito.isNull()))
+            .when(rxCollectionCache.resolveByNameAsync(Mockito.any(), Mockito.any(), Mockito.isNull(), Mockito.isNull(), Mockito.isNull()))
             .thenReturn(Mono.just(documentCollection));
         doNothing().when(rxCollectionCache).refresh(Mockito.any(), Mockito.any(), Mockito.isNull());
 
@@ -51,7 +51,8 @@ public void staledException(int statusCode, int subStatusCode, boolean expectRet
             null,
             null,
             sessionContainer,
-            TestUtils.mockDiagnosticsClientContext()
+            TestUtils.mockDiagnosticsClientContext(),
+            null
         );
 
         CosmosException exception = BridgeInternal.createCosmosException(statusCode);
@@ -72,7 +73,7 @@ public void suppressRetryForExternalCollectionRid() {
 
         RxCollectionCache rxCollectionCache = Mockito.mock(RxCollectionCache.class);
         Mockito
-            .when(rxCollectionCache.resolveByNameAsync(Mockito.any(), Mockito.any(), Mockito.isNull()))
+            .when(rxCollectionCache.resolveByNameAsync(Mockito.any(), Mockito.any(), Mockito.isNull(), Mockito.isNull(), Mockito.isNull()))
             .thenReturn(Mono.just(documentCollection));
         doNothing().when(rxCollectionCache).refresh(Mockito.any(), Mockito.any(), Mockito.isNull());
 
@@ -88,7 +89,8 @@ public void suppressRetryForExternalCollectionRid() {
             null,
             customHeaders,
             sessionContainer,
-            TestUtils.mockDiagnosticsClientContext()
+            TestUtils.mockDiagnosticsClientContext(),
+            null
         );
 
         InvalidPartitionException invalidPartitionException = new InvalidPartitionException();
@@ -109,7 +111,7 @@ public void cleanSessionToken() {
 
         RxCollectionCache rxCollectionCache = Mockito.mock(RxCollectionCache.class);
         Mockito
-            .when(rxCollectionCache.resolveByNameAsync(Mockito.any(), Mockito.any(), Mockito.isNull()))
+            .when(rxCollectionCache.resolveByNameAsync(Mockito.any(), Mockito.any(), Mockito.isNull(), Mockito.isNull(), Mockito.isNull()))
             .thenReturn(Mono.just(documentCollection))
             .thenReturn(Mono.just(documentCollectionAfterRefresh));
 
@@ -125,7 +127,8 @@ public void cleanSessionToken() {
             null,
             null,
             sessionContainer,
-            TestUtils.mockDiagnosticsClientContext()
+            TestUtils.mockDiagnosticsClientContext(),
+            null
         );
 
         InvalidPartitionException invalidPartitionException = new InvalidPartitionException();

sdk/cosmos/azure-cosmos/CHANGELOG.md

@@ -9,6 +9,7 @@
 #### Bugs Fixed
 
 #### Other Changes
+* Remaps sub-status to 1003 for requests to child resources against non-existent container. - [PR 47604](https://github.com/Azure/azure-sdk-for-java/pull/47604)
 
 ### 4.76.0 (2025-12-09)
 

sdk/cosmos/azure-cosmos/src/main/java/com/azure/cosmos/CosmosException.java

@@ -674,6 +674,16 @@ public void setRequestUri(CosmosException cosmosException, Uri requestUri) {
                     public Uri getRequestUri(CosmosException cosmosException) {
                         return cosmosException.getRequestUri();
                     }
+
+                    @Override
+                    public void setSubStatusCode(CosmosException cosmosException, int subStatusCode) {
+                        Map<String, String> responseHeaders = cosmosException.getResponseHeaders();
+
+                        responseHeaders.put(HttpConstants.HttpHeaders.SUB_STATUS, String.valueOf(subStatusCode));
+                        cosmosException.setSubStatusCode(subStatusCode);
+                    }
+
+
                 });
     }
 

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

@@ -1596,6 +1596,7 @@ public interface CosmosExceptionAccessor {
             List<String> getFaultInjectionEvaluationResults(CosmosException cosmosException);
             void setRequestUri(CosmosException cosmosException, Uri requestUri);
             Uri getRequestUri(CosmosException cosmosException);
+            void setSubStatusCode(CosmosException cosmosException, int subStatusCode);
         }
     }
 

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

@@ -191,6 +191,9 @@ public class RxDocumentClientImpl implements AsyncDocumentClient, IAuthorization
     private static final ImplementationBridgeHelpers.CosmosBulkExecutionOptionsHelper.CosmosBulkExecutionOptionsAccessor bulkExecutionOptionsAccessor =
         ImplementationBridgeHelpers.CosmosBulkExecutionOptionsHelper.getCosmosBulkExecutionOptionsAccessor();
 
+    private static final ImplementationBridgeHelpers.CosmosExceptionHelper.CosmosExceptionAccessor cosmosExceptionAccessor =
+        ImplementationBridgeHelpers.CosmosExceptionHelper.getCosmosExceptionAccessor();
+
     private static final String tempMachineId = "uuid:" + UUIDs.nonBlockingRandomUUID();
     private static final AtomicInteger activeClientsCnt = new AtomicInteger(0);
     private static final Map<String, Integer> clientMap = new ConcurrentHashMap<>();
@@ -1272,7 +1275,9 @@ private <T> Flux<FeedResponse<T>> createQuery(
             qryOptAccessor.getProperties(nonNullQueryOptions),
             qryOptAccessor.getHeaders(nonNullQueryOptions),
             this.sessionContainer,
-            diagnosticsFactory);
+            diagnosticsFactory,
+            ResourceType.Document
+        );
 
         return
             ObservableHelper.fluxInlineIfPossibleAsObs(
@@ -2609,7 +2614,8 @@ private Mono<ResourceResponse<Document>> createDocumentCore(
             this.getRetryPolicyForPointOperation(
                 scopedDiagnosticsFactory,
                 nonNullRequestOptions,
-                collectionLink);
+                collectionLink
+            );
 
         AtomicReference<RxDocumentServiceRequest> requestReference = new AtomicReference<>();
 
@@ -2996,7 +3002,8 @@ private Mono<ResourceResponse<Document>> upsertDocumentCore(
             this.getRetryPolicyForPointOperation(
                 scopedDiagnosticsFactory,
                 nonNullRequestOptions,
-                collectionLink);;
+                collectionLink
+            );
         AtomicReference<RxDocumentServiceRequest> requestReference = new AtomicReference<>();
 
         Consumer<CosmosException> gwModeE2ETimeoutDiagnosticHandler
@@ -3138,7 +3145,8 @@ private Mono<ResourceResponse<Document>> replaceDocumentCore(
             this.getRetryPolicyForPointOperation(
                 scopedDiagnosticsFactory,
                 nonNullRequestOptions,
-                Utils.getCollectionName(documentLink));
+                Utils.getCollectionName(documentLink)
+            );
         AtomicReference<RxDocumentServiceRequest> requestReference = new AtomicReference<>();
 
         Consumer<CosmosException> gwModeE2ETimeoutDiagnosticHandler
@@ -3471,7 +3479,8 @@ private Mono<ResourceResponse<Document>> patchDocumentCore(
             this.getRetryPolicyForPointOperation(
                 scopedDiagnosticsFactory,
                 nonNullRequestOptions,
-                Utils.getCollectionName(documentLink));
+                Utils.getCollectionName(documentLink)
+            );
 
         AtomicReference<RxDocumentServiceRequest> requestReference = new AtomicReference<>();
 
@@ -3684,7 +3693,8 @@ private Mono<ResourceResponse<Document>> deleteDocumentCore(
             this.getRetryPolicyForPointOperation(
                 scopedDiagnosticsFactory,
                 nonNullRequestOptions,
-                Utils.getCollectionName(documentLink));
+                Utils.getCollectionName(documentLink)
+            );
 
         AtomicReference<RxDocumentServiceRequest> requestReference = new AtomicReference<>();
 
@@ -3876,7 +3886,8 @@ private Mono<ResourceResponse<Document>> readDocumentCore(
             this.getRetryPolicyForPointOperation(
                 scopedDiagnosticsFactory,
                 nonNullRequestOptions,
-                Utils.getCollectionName(documentLink));
+                Utils.getCollectionName(documentLink)
+            );
 
         AtomicReference<RxDocumentServiceRequest> requestReference = new AtomicReference<>();
 
@@ -4025,7 +4036,9 @@ public <T> Mono<FeedResponse<T>> readMany(
             qryOptAccessor.getProperties(state.getQueryOptions()),
             qryOptAccessor.getHeaders(state.getQueryOptions()),
             this.sessionContainer,
-            diagnosticsFactory);
+            diagnosticsFactory,
+            ResourceType.Document
+        );
 
         return ObservableHelper
             .inlineIfPossibleAsObs(
@@ -4722,7 +4735,9 @@ public <T> Flux<FeedResponse<T>> queryDocumentChangeFeedFromPagedFlux(
             changeFeedOptionsAccessor.getProperties(state.getChangeFeedOptions()),
             changeFeedOptionsAccessor.getHeaders(state.getChangeFeedOptions()),
             this.sessionContainer,
-            diagnosticsFactory);
+            diagnosticsFactory,
+            ResourceType.Document
+        );
 
         return ObservableHelper
             .fluxInlineIfPossibleAsObs(
@@ -4746,7 +4761,7 @@ private <T> Flux<FeedResponse<T>> queryDocumentChangeFeedFromPagedFluxInternal(
         DiagnosticsClientContext diagnosticsClientContext) {
 
         return this.getCollectionCache()
-            .resolveByNameAsync(null, collectionLink, null)
+            .resolveByNameAsync(null, collectionLink, null, null, ResourceType.Document)
             .flatMapMany(collection -> {
                 if (collection == null) {
                     throw new IllegalStateException("Collection can not be null");
@@ -4867,7 +4882,9 @@ public <T> Flux<FeedResponse<T>> readAllDocuments(
                 qryOptAccessor.getProperties(effectiveOptions),
                 qryOptAccessor.getHeaders(effectiveOptions),
                 this.sessionContainer,
-                diagnosticsFactory);
+                diagnosticsFactory,
+                ResourceType.Document
+            );
 
             Flux<FeedResponse<T>> innerFlux = ObservableHelper.fluxInlineIfPossibleAsObs(
                 () -> {
@@ -5196,7 +5213,9 @@ public Mono<CosmosBatchResponse> executeBatchRequest(String collectionLink,
                 nonNullRequestOptions.getProperties(),
                 nonNullRequestOptions.getHeaders(),
                 this.sessionContainer,
-                scopedDiagnosticsFactory);
+                scopedDiagnosticsFactory,
+                ResourceType.Document
+            );
         }
 
         final DocumentClientRetryPolicy finalRetryPolicy = documentClientRetryPolicy;
@@ -6633,7 +6652,9 @@ public Mono<List<FeedRange>> getFeedRanges(String collectionLink, boolean forceR
             new HashMap<>(),
             new HashMap<>(),
             this.sessionContainer,
-            null);
+            null,
+            ResourceType.PartitionKeyRange
+        );
 
         RxDocumentServiceRequest request = RxDocumentServiceRequest.create(
             this,
@@ -8020,7 +8041,9 @@ private DocumentClientRetryPolicy getRetryPolicyForPointOperation(
             requestOptions.getProperties(),
             requestOptions.getHeaders(),
             this.sessionContainer,
-            diagnosticsClientContext);
+            diagnosticsClientContext,
+            ResourceType.Document
+        );
 
         return requestRetryPolicy;
     }

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

@@ -25,8 +25,6 @@
 import com.azure.cosmos.implementation.http.HttpTransportSerializer;
 import com.azure.cosmos.implementation.http.ReactorNettyRequestRecord;
 import com.azure.cosmos.implementation.interceptor.ITransportClientInterceptor;
-import com.azure.cosmos.implementation.perPartitionCircuitBreaker.GlobalPartitionEndpointManagerForPerPartitionCircuitBreaker;
-import com.azure.cosmos.implementation.perPartitionCircuitBreaker.LocationSpecificHealthContext;
 import com.azure.cosmos.implementation.routing.PartitionKeyInternal;
 import com.azure.cosmos.implementation.routing.PartitionKeyInternalHelper;
 import com.azure.cosmos.implementation.routing.PartitionKeyRangeIdentity;
@@ -635,7 +633,7 @@ private Mono<RxDocumentServiceResponse> toDocumentServiceResponse(Mono<HttpRespo
                         CrossRegionAvailabilityContextForRxDocumentServiceRequest availabilityStrategyContextForReq =
                             request.requestContext.getCrossRegionAvailabilityContext();
 
-                        if (availabilityStrategyContextForReq.getAvailabilityStrategyContext().isAvailabilityStrategyEnabled() && !availabilityStrategyContextForReq.getAvailabilityStrategyContext().isHedgedRequest()) {
+                        if (availabilityStrategyContextForReq.getAvailabilityStrategyContext() != null && availabilityStrategyContextForReq.getAvailabilityStrategyContext().isAvailabilityStrategyEnabled() && !availabilityStrategyContextForReq.getAvailabilityStrategyContext().isHedgedRequest()) {
 
                             BridgeInternal.setRequestTimeline(oce, reactorNettyRequestRecord.takeTimelineSnapshot());