Refactor unit tests

botocore/compress.py

@@ -16,9 +16,10 @@
 
 """
 
-import gzip
 import io
 import logging
+from gzip import GzipFile
+from gzip import compress as gzip_compress
 
 from botocore.compat import urlencode
 from botocore.utils import determine_content_length
@@ -88,9 +89,9 @@ def _get_body_size(body):
 
 def _gzip_compress_body(body):
     if isinstance(body, str):
-        return gzip.compress(body.encode('utf-8'))
+        return gzip_compress(body.encode('utf-8'))
     elif isinstance(body, (bytes, bytearray)):
-        return gzip.compress(body)
+        return gzip_compress(body)
     elif hasattr(body, 'read'):
         if hasattr(body, 'seek') and hasattr(body, 'tell'):
             current_position = body.tell()
@@ -102,7 +103,7 @@ def _gzip_compress_body(body):
 
 def _gzip_compress_fileobj(body):
     compressed_obj = io.BytesIO()
-    with gzip.GzipFile(fileobj=compressed_obj, mode='wb') as gz:
+    with GzipFile(fileobj=compressed_obj, mode='wb') as gz:
         while True:
             chunk = body.read(8192)
             if not chunk:

tests/__init__.py

@@ -587,6 +587,30 @@ def __exit__(self, *args, **kwargs):
         self.datetime_patcher.stop()
 
 
+class PatchObject(ContextDecorator):
+    """
+    Context manager for patching out objects in specific classes for tests.
+
+    :type module: module
+    :param module: reference to imported module to patch
+
+    :type obj: str
+    :param obj: object within the module to replace.
+
+    :type new: Any
+    :param new: Any functional replacement for object to be patched.
+    """
+
+    def __init__(self, module, obj, new):
+        self.obj_patcher = mock.patch.object(module, obj, new)
+
+    def __enter__(self, *args, **kwargs):
+        self.obj_patcher.start()
+
+    def __exit__(self, *args, **kwargs):
+        self.obj_patcher.stop()
+
+
 def patch_load_service_model(
     session, monkeypatch, service_model_json, ruleset_json
 ):