Merged revisions 68319 via svnmerge from
svn+ssh://pythondev@svn.python.org/python/trunk
........
r68319 | antoine.pitrou | 2009-01-04 22:29:23 +0100 (dim., 04 janv. 2009) | 3 lines
Issue #4272: Add an optional argument to the GzipFile constructor to override the timestamp in the gzip stream.
........
diff --git a/Lib/test/test_gzip.py b/Lib/test/test_gzip.py
index d28c024..e758826 100644
--- a/Lib/test/test_gzip.py
+++ b/Lib/test/test_gzip.py
@@ -6,6 +6,7 @@
from test import support
import os
import gzip
+import struct
data1 = b""" int length=DEFAULTALLOC, err = Z_OK;
@@ -160,6 +161,68 @@
self.assertEqual(f.name, self.filename)
f.close()
+ def test_mtime(self):
+ mtime = 123456789
+ fWrite = gzip.GzipFile(self.filename, 'w', mtime = mtime)
+ fWrite.write(data1)
+ fWrite.close()
+
+ fRead = gzip.GzipFile(self.filename)
+ dataRead = fRead.read()
+ self.assertEqual(dataRead, data1)
+ self.assert_(hasattr(fRead, 'mtime'))
+ self.assertEqual(fRead.mtime, mtime)
+ fRead.close()
+
+ def test_metadata(self):
+ mtime = 123456789
+
+ fWrite = gzip.GzipFile(self.filename, 'w', mtime = mtime)
+ fWrite.write(data1)
+ fWrite.close()
+
+ fRead = open(self.filename, 'rb')
+
+ # see RFC 1952: http://www.faqs.org/rfcs/rfc1952.html
+
+ idBytes = fRead.read(2)
+ self.assertEqual(idBytes, b'\x1f\x8b') # gzip ID
+
+ cmByte = fRead.read(1)
+ self.assertEqual(cmByte, b'\x08') # deflate
+
+ flagsByte = fRead.read(1)
+ self.assertEqual(flagsByte, b'\x08') # only the FNAME flag is set
+
+ mtimeBytes = fRead.read(4)
+ self.assertEqual(mtimeBytes, struct.pack('<i', mtime)) # little-endian
+
+ xflByte = fRead.read(1)
+ self.assertEqual(xflByte, b'\x02') # maximum compression
+
+ osByte = fRead.read(1)
+ self.assertEqual(osByte, b'\xff') # OS "unknown" (OS-independent)
+
+ # Since the FNAME flag is set, the zero-terminated filename follows.
+ # RFC 1952 specifies that this is the name of the input file, if any.
+ # However, the gzip module defaults to storing the name of the output
+ # file in this field.
+ expected = self.filename.encode('Latin-1') + b'\x00'
+ nameBytes = fRead.read(len(expected))
+ self.assertEqual(nameBytes, expected)
+
+ # Since no other flags were set, the header ends here.
+ # Rather than process the compressed data, let's seek to the trailer.
+ fRead.seek(os.stat(self.filename).st_size - 8)
+
+ crc32Bytes = fRead.read(4) # CRC32 of uncompressed data [data1]
+ self.assertEqual(crc32Bytes, b'\xaf\xd7d\x83')
+
+ isizeBytes = fRead.read(4)
+ self.assertEqual(isizeBytes, struct.pack('<i', len(data1)))
+
+ fRead.close()
+
def test_main(verbose=None):
support.run_unittest(TestGzip)