test: Add test for reference cycle in gzip.GzipFile

Mr-Sunglasses · Mar 6, 2025 · Mar 6, 2025 · Mar 6, 2025 · Mar 6, 2025 · Mar 6, 2025
commit 79b7f6ca15d27e1461005213632ddb3aa0fd6704
diff --git a/Lib/test/test_gzip.py b/Lib/test/test_gzip.py
@@ -467,6 +467,25 @@ def sizes():
             self.assertEqual(f.read(100), b'')
             self.assertEqual(nread, len(uncompressed))
 
+    def test_no_refloop_with_writestream(self):
+        """Test that the _WriteBufferStream __del__ method prevents reference loops."""
+        import gc
+
+        gc.collect()
+
+        before_count = len(gc.get_objects())
+
+        with io.BytesIO() as bio:
+            with gzip.GzipFile(fileobj=bio, mode="wb") as f:
+                f.write(b"test data")
+
+        gc.collect(0)
+
+        # Check that all objects were properly cleaned up
+        after_count = len(gc.get_objects())
+        self.assertLess(after_count - before_count, 10,
+                        "Too many objects remain after GzipFile cleanup")
+
     def test_textio_readlines(self):
         # Issue #10791: TextIOWrapper.readlines() fails when wrapping GzipFile.
         lines = (data1 * 50).decode("ascii").splitlines(keepends=True)