update policies

Author: l0lawrence

sdk/storage/azure-storage-file-share/azure/storage/fileshare/_shared/policies.py

@@ -467,20 +467,25 @@ def _validate_content_response(
             # Raises exception if missing
             content_length = int(response.http_response.headers[CONTENT_LENGTH_HEADER])
 
-            # Patch response to return response iterator wrapped in structured message decoder
-            original_iter_bytes = response.http_response.iter_bytes
-
-            def wrapped_iter_bytes(*args, **kwargs):
-                iterator = original_iter_bytes(*args, **kwargs)
-                decoder = decoder_cls(iterator, content_length, block_size=DATA_BLOCK_SIZE)
-                # Only set request/response if present on iterator
-                if hasattr(iterator, "request"):
-                    decoder.request = iterator.request  # type: ignore
-                if hasattr(iterator, "response"):
-                    decoder.response = iterator.response  # type: ignore
-                return decoder
-
-            response.http_response.iter_bytes = wrapped_iter_bytes
+            def _make_wrapper(original):
+                def wrapped(*args, **kwargs):
+                    iterator = original(*args, **kwargs)
+                    decoder = decoder_cls(iterator, content_length, block_size=DATA_BLOCK_SIZE)
+                    if hasattr(iterator, "request"):
+                        decoder.request = iterator.request  # type: ignore
+                    if hasattr(iterator, "response"):
+                        decoder.response = iterator.response  # type: ignore
+                    return decoder
+                return wrapped
+
+            # Patch response to return response iterator wrapped in structured message decoder.
+            # TypeSpec-generated code calls iter_bytes()/iter_raw() instead of stream_download().
+            if hasattr(response.http_response, "iter_bytes"):
+                response.http_response.iter_bytes = _make_wrapper(response.http_response.iter_bytes)
+            if hasattr(response.http_response, "iter_raw"):
+                response.http_response.iter_raw = _make_wrapper(response.http_response.iter_raw)
+            if hasattr(response.http_response, "stream_download"):
+                response.http_response.stream_download = _make_wrapper(response.http_response.stream_download)
 
 
 class StorageContentValidation(SansIOHTTPPolicy):