apify
diff --git a/‎src/apify/scrapy/cache.py
Copy file name to clipboardExpand all lines: src/apify/scrapy/cache.py
+4-3Lines changed: 4 additions & 3 deletions b/‎src/apify/scrapy/cache.py
Copy file name to clipboardExpand all lines: src/apify/scrapy/cache.py
+4-3Lines changed: 4 additions & 3 deletions
diff --git a/‎tests/unit/scrapy/test_cache.py
Copy file name to clipboardExpand all lines: tests/unit/scrapy/test_cache.py
+6-4Lines changed: 6 additions & 4 deletions b/‎tests/unit/scrapy/test_cache.py
Copy file name to clipboardExpand all lines: tests/unit/scrapy/test_cache.py
+6-4Lines changed: 6 additions & 4 deletions
@@ -155,9 +155,10 @@ def store_response(self, _: Spider, request: Request, response: Response) -> Non
 
 def to_gzip(data: dict, mtime: int | None = None) -> bytes:
     """Dump a dictionary to a gzip-compressed byte stream."""
-    with io.BytesIO() as byte_stream, gzip.GzipFile(fileobj=byte_stream, mode='wb', mtime=mtime) as gzip_file:
-        pickle.dump(data, gzip_file, protocol=4)
-    return byte_stream.getvalue()
+    with io.BytesIO() as byte_stream:
+        with gzip.GzipFile(fileobj=byte_stream, mode='wb', mtime=mtime) as gzip_file:
+            pickle.dump(data, gzip_file, protocol=4)
+        return byte_stream.getvalue()
 
 
 def from_gzip(gzip_bytes: bytes) -> dict:
 
@@ -2,26 +2,28 @@
 
 from apify.scrapy.cache import from_gzip, read_gzip_time, to_gzip
 
+FIXTURE_DICT = {'name': 'Alice'}
+
 FIXTURE_BYTES = (
     b'\x1f\x8b\x08\x00\x00\x00\x00\x00\x02\xffk`\x99*\xcc\x00\x01\xb5SzX\xf2\x12s'
     b'S\xa7\xf4\xb0:\xe6d&\xa7N)\xd6\x03\x00\x1c\xe8U\x9c\x1e\x00\x00\x00'
 )
 
 
 def test_gzip() -> None:
-    assert from_gzip(to_gzip({'name': 'Alice'})) == {'name': 'Alice'}
+    assert from_gzip(to_gzip(FIXTURE_DICT)) == FIXTURE_DICT
 
 
 def test_to_gzip() -> None:
-    data_bytes = to_gzip({'name': 'Alice'}, mtime=0)
+    data_bytes = to_gzip(FIXTURE_DICT, mtime=0)
 
     assert data_bytes == FIXTURE_BYTES
 
 
 def test_from_gzip() -> None:
     data_dict = from_gzip(FIXTURE_BYTES)
 
-    assert data_dict == {'name': 'Alice'}
+    assert data_dict == FIXTURE_DICT
 
 
 def test_read_gzip_time() -> None:
@@ -30,6 +32,6 @@ def test_read_gzip_time() -> None:
 
 def test_read_gzip_time_non_zero() -> None:
     current_time = int(time())
-    data_bytes = to_gzip({'name': 'Alice'}, mtime=current_time)
+    data_bytes = to_gzip(FIXTURE_DICT, mtime=current_time)
 
     assert read_gzip_time(data_bytes) == current_time