update unittest

Author: 5ec1cff

Lib/test/test_zipfile/test_core.py

@@ -3449,71 +3449,26 @@ def test_too_short(self):
 
 class StoreSeekReadTest(unittest.TestCase):
     def test_store_seek_read(self):
-        class StatIO(io.RawIOBase):
-            def __init__(self, buf: bytes):
+        from _pyio import BytesIO
+        class StatIO(BytesIO):
+            def __init__(self):
+                super().__init__()
                 self.bytes_read = 0
-                self.buffer = buf
-                self.pos = 0
-                self.eof = False
-
-            def readinto(self, buffer, /):
-                if self.eof:
-                    return 0
-                sz = min(len(buffer), len(self.buffer) - self.pos)
-                buffer[:sz] = self.buffer[self.pos:self.pos + sz]
-                self.pos += sz
-                self.bytes_read += sz
-                self.eof = self.pos == len(self.buffer)
-                return sz
-
-            def readall(self):
-                if self.eof:
-                    return b''
-                self.eof = True
-                self.bytes_read += len(self.buffer) - self.pos
-                ret = self.buffer[self.pos:]
-                self.pos = len(self.buffer)
-                return ret
-
-            def seek(self, offset, whence=os.SEEK_SET, /):
-                if whence == os.SEEK_CUR:
-                    new_pos = self.pos + offset
-                elif whence == os.SEEK_SET:
-                    new_pos = offset
-                elif whence == os.SEEK_END:
-                    new_pos = len(self.buffer) + offset
-                else:
-                    raise ValueError("unsupported whence")
-
-                if new_pos < 0:
-                    new_pos = 0
-                elif new_pos > len(self.buffer):
-                    new_pos = len(self.buffer)
-
-                self.eof = new_pos == len(self.buffer)
-                self.pos = new_pos
-                return new_pos
-
-            def tell(self):
-                return self.pos
-
-            def readable(self):
-                return True
-
-            def writable(self):
-                return False
 
-            def seekable(self):
-                return True
+            def read(self, size=-1):
+                bs = super().read(size)
+                self.bytes_read += len(bs)
+                print('read sz', len(bs))
+                return bs
 
             def get_bytes_read(self):
                 return self.bytes_read
 
-        bio = io.BytesIO()
+        sio = StatIO()
         txt = b'0123456789'*10000
 
         # Check seek on a file
-        with zipfile.ZipFile(bio, "w") as zipf:
+        with zipfile.ZipFile(sio, "w") as zipf:
             zipf.writestr("foo.txt", txt) # 100000 bytes
 
         sio = StatIO(bio.getvalue())

Misc/NEWS.d/next/Library/2024-12-21-03-20-12.gh-issue-128131.QpPmNt.rst

@@ -1 +1 @@
-Support forward seek in uncompressed unencrypted :class:`!zipfile.ZipExtFile`.
+Support fast forward seek in uncompressed unencrypted :class:`!zipfile.ZipExtFile`.