Cosmos: Fix #2824 by forcing 'enable_content_response_on_write' on for read item (#2894)

Author: analogrelay

sdk/cosmos/azure_data_cosmos/src/clients/container_client.rs

@@ -490,7 +490,11 @@ impl ContainerClient {
         item_id: &str,
         options: Option<ItemOptions<'_>>,
     ) -> azure_core::Result<Response<T>> {
-        let options = options.unwrap_or_default();
+        let mut options = options.unwrap_or_default();
+
+        // Read APIs should always return the item, ignoring whatever the user set.
+        options.enable_content_response_on_write = true;
+
         let link = self.items_link.item(item_id);
         let url = self.pipeline.url(&link);
         let mut req = Request::new(url, Method::Get);

sdk/cosmos/azure_data_cosmos/src/constants.rs

@@ -35,6 +35,7 @@ pub const POST_TRIGGER_INCLUDE: HeaderName =
     HeaderName::from_static("x-ms-documentdb-post-trigger-include");
 pub const SESSION_TOKEN: HeaderName = HeaderName::from_static("x-ms-session-token");
 pub const INDEXING_DIRECTIVE: HeaderName = HeaderName::from_static("x-ms-indexing-directive");
+pub const SUB_STATUS: HeaderName = HeaderName::from_static("x-ms-substatus");
 
 pub const QUERY_CONTENT_TYPE: ContentType = ContentType::from_static("application/query+json");
 

sdk/cosmos/azure_data_cosmos/src/pipeline/mod.rs

@@ -66,7 +66,9 @@ impl CosmosPipeline {
         resource_link: ResourceLink,
     ) -> azure_core::Result<BufResponse> {
         let ctx = ctx.with_value(resource_link);
-        self.pipeline.send(&ctx, request).await
+        let r = self.pipeline.send(&ctx, request).await?;
+        let r = azure_core::http::check_success(r).await?;
+        Ok(r)
     }
 
     pub async fn send<T>(
@@ -77,7 +79,7 @@ impl CosmosPipeline {
     ) -> azure_core::Result<Response<T>> {
         self.send_raw(ctx, request, resource_link)
             .await
-            .map(|r| r.into())
+            .map(Into::into)
     }
 
     pub fn send_query_request<T: DeserializeOwned + Send>(
@@ -108,6 +110,7 @@ impl CosmosPipeline {
                 }
 
                 let resp = pipeline.send(&ctx, &mut req).await?;
+                let resp = azure_core::http::check_success(resp).await?;
                 let page = FeedPage::<T>::from_response(resp).await?;
 
                 Ok(page.into())