Passing Response Hook to Client Connection (#43288)

* passing response hook to client connection

* removing pylint disable

---------

Co-authored-by: Andrew Mathew <[email protected]>

Author: andrewmathew1

sdk/cosmos/azure-cosmos/azure/cosmos/aio/_container.py

@@ -1513,7 +1513,7 @@ async def get_throughput(
         return _deserialize_throughput(throughput=throughput_properties)
 
     @distributed_trace_async
-    async def replace_throughput( # pylint: disable=unused-argument
+    async def replace_throughput(
         self,
         throughput: Union[int, ThroughputProperties],
         *,
@@ -1547,7 +1547,7 @@ async def replace_throughput( # pylint: disable=unused-argument
         new_offer = throughput_properties[0].copy()
         _replace_throughput(throughput=throughput, new_throughput_properties=new_offer)
         data = await self.client_connection.ReplaceOffer(offer_link=throughput_properties[0]["_self"],
-                                                         offer=throughput_properties[0], **kwargs)
+                                                         offer=throughput_properties[0], response_hook=response_hook, **kwargs)
 
         return ThroughputProperties(offer_throughput=data["content"]["offerThroughput"], properties=data)
 

sdk/cosmos/azure-cosmos/azure/cosmos/container.py

@@ -1698,7 +1698,7 @@ def get_throughput(
         return _deserialize_throughput(throughput=throughput_properties)
 
     @distributed_trace
-    def replace_throughput( # pylint: disable=unused-argument
+    def replace_throughput(
         self,
         throughput: Union[int, ThroughputProperties],
         *,
@@ -1730,7 +1730,7 @@ def replace_throughput( # pylint: disable=unused-argument
         new_throughput_properties = throughput_properties[0].copy()
         _replace_throughput(throughput=throughput, new_throughput_properties=new_throughput_properties)
         data = self.client_connection.ReplaceOffer(
-            offer_link=throughput_properties[0]["_self"], offer=throughput_properties[0], **kwargs)
+            offer_link=throughput_properties[0]["_self"], offer=throughput_properties[0], response_hook=response_hook, **kwargs)
 
         return ThroughputProperties(offer_throughput=data["content"]["offerThroughput"], properties=data)