Split up requests and urllib3 unit tests.

Signed-off-by: dblock <dblock@amazon.com>
opensearch-project · Oct 23, 2023 · 8d5aeb0 · 8d5aeb0
1 parent df5b29b
commit 8d5aeb0
Show file tree

Hide file tree

Showing 6 changed files with 1,179 additions and 1,237 deletions.
diff --git a/test_opensearchpy/test_async/test_connection.py b/test_opensearchpy/test_async/test_connection.py
@@ -48,11 +48,6 @@
 pytestmark = pytest.mark.asyncio
 
 
-def gzip_decompress(data):
-    buf = gzip.GzipFile(fileobj=io.BytesIO(data), mode="rb")
-    return buf.read()
-
-
 class TestAIOHttpConnection:
     async def _get_mock_connection(
         self,
@@ -130,7 +125,8 @@ async def test_http_compression(self):
 
         _, kwargs = con.session.request.call_args
 
-        assert gzip_decompress(kwargs["data"]) == b"{}"
+        buf = gzip.GzipFile(fileobj=io.BytesIO(kwargs["data"]), mode="rb")
+        assert buf.read() == b"{}"
         assert kwargs["headers"]["accept-encoding"] == "gzip,deflate"
         assert kwargs["headers"]["content-encoding"] == "gzip"