tests

Author: Wulian233

Lib/test/test_zipfile/_path/test_path.py

@@ -1,6 +1,7 @@
 import io
 import itertools
 import contextlib
+import os
 import pathlib
 import pickle
 import stat

Lib/test/test_zipfile/test_core.py

@@ -1834,6 +1834,30 @@ def test_write_unicode_filenames(self):
             self.assertEqual(zf.filelist[0].filename, "foo.txt")
             self.assertEqual(zf.filelist[1].filename, "\xf6.txt")
 
+    def test_write_with_source_date_epoch(self):
+        os.environ['SOURCE_DATE_EPOCH'] = "1727268221"
+        data = "Testing SOURCE_DATE_EPOCH"
+
+        with zipfile.ZipFile(self.TESTFN, "w") as zf:
+            zf.writestr("test_source_date_epoch.txt", data)
+
+        with zipfile.ZipFile(self.TESTFN, "r") as zf:
+            zip_info = zf.getinfo("test_source_date_epoch.txt")
+            self.assertEqual(zip_info.date_time, time.gmtime(int(os.environ['SOURCE_DATE_EPOCH']))[:6])
+
+    def test_write_without_source_date_epoch(self):
+        if 'SOURCE_DATE_EPOCH' in os.environ:
+            del os.environ['SOURCE_DATE_EPOCH']
+        data = "Testing without SOURCE_DATE_EPOCH"
+
+        with zipfile.ZipFile(self.TESTFN, "w") as zf:
+            zf.writestr("test_no_source_date_epoch.txt", data)
+
+        with zipfile.ZipFile(self.TESTFN, "r") as zf:
+            zip_info = zf.getinfo("test_no_source_date_epoch.txt")
+            current_time = time.gmtime()[:6]
+            self.assertNotEqual(zip_info.date_time, current_time)
+
     def create_zipfile_with_extra_data(self, filename, extra_data_name):
         with zipfile.ZipFile(TESTFN, mode='w') as zf:
             filename_encoded = filename.encode("utf-8")