Added offline version of test_decompress_compressed_header (#22112)

zhzhh97 · rakshith91 · zhihengzhu · web-flow · commit 0bcf6ad31df2 · 2022-08-02T08:53:21.000-07:00
* initial * Added offline version of test_decompress_compressed_header * Fixed according to the feedback * Updated by using _pipeline.run * Fix CI Check Error * Fix Merge Conflict * Fix Merge Conflict * Fix Merge Conflict * Fixed based on feedback * Added string() back, deleted debugging code * Update test_streaming.py * Added string() back, deleted debugging code * Revert "Added string() back, deleted debugging code" This reverts commit fd207f2. * don't touch async file for this PR * Update streams.py * Update streams.py Co-authored-by: Rakshith Bhyravabhotla <rakshith.bhyravabhotla@gmail.com> Co-authored-by: zhihengzhu <zhuzhihen97@outlook.com> Co-authored-by: Krista Pratico <krpratic@microsoft.com> Co-authored-by: Xiang Yan <xiangsjtu@gmail.com>
diff --git a/sdk/core/azure-core/tests/test_streaming.py b/sdk/core/azure-core/tests/test_streaming.py
@@ -163,6 +163,7 @@ def test_compress_plain_header(http_request):
     decoded = content.decode('utf-8')
     assert decoded == "test"
 
+@pytest.mark.live_test_only
 @pytest.mark.parametrize("http_request", HTTP_REQUESTS)
 def test_decompress_compressed_header(http_request):
     # expect plain text
@@ -178,6 +179,18 @@ def test_decompress_compressed_header(http_request):
     decoded = content.decode('utf-8')
     assert decoded == "test"
 
+@pytest.mark.parametrize("http_request", HTTP_REQUESTS)
+def test_decompress_compressed_header_offline(port, http_request):
+    client = PipelineClient("")
+    request = http_request(method="GET", url="http://localhost:{}/streams/decompress_header".format(port))
+    with RequestsTransport() as sender:
+        response = client._pipeline.run(request, stream=True).http_response
+        response.raise_for_status()
+        data = response.stream_download(sender, decompress=True)
+        content = b"".join(list(data))
+        decoded = content.decode('utf-8')
+        assert decoded == "test"
+
 @pytest.mark.parametrize("http_request", HTTP_REQUESTS)
 def test_compress_compressed_header(http_request):
     # expect compressed text
diff --git a/sdk/core/azure-core/tests/testserver_tests/coretestserver/coretestserver/test_routes/streams.py b/sdk/core/azure-core/tests/testserver_tests/coretestserver/coretestserver/test_routes/streams.py
@@ -6,6 +6,7 @@
 # -------------------------------------------------------------------------
 import os
 import gzip
+import tempfile
 from flask import (
     Response,
     Blueprint,
@@ -18,7 +19,6 @@ def __iter__(self):
         yield b"Hello, "
         yield b"world!"
 
-
 def streaming_body():
     yield b"Hello, "
     yield b"world!"
@@ -29,7 +29,7 @@ def stream_json_error():
 
 def streaming_test():
     yield b"test"
-    
+
 def stream_compressed_header_error():
     yield b'test'
 
@@ -41,7 +41,7 @@ def stream_compressed_no_header():
         yield fd.read()
     
     os.remove("test.tar.gz")
-
+    
 @streams_api.route('/basic', methods=['GET'])
 def basic():
     return Response(streaming_body(), status=200)
@@ -67,4 +67,15 @@ def compressed_no_header():
 @streams_api.route('/compressed', methods=['GET'])
 def compressed():
     return Response(stream_compressed_header_error(), status=300, headers={"Content-Encoding": "gzip"})
+
+def compressed_stream():
+    with tempfile.TemporaryFile(mode='w+b') as f:
+        gzf = gzip.GzipFile(mode='w+b', fileobj=f)
+        gzf.write(b"test")
+        gzf.flush()
+        f.seek(0)
+        yield f.read()
     
+@streams_api.route('/decompress_header', methods=['GET'])
+def decompress_header():
+    return Response(compressed_stream(), status=200, headers={"Content-Encoding": "gzip"})
