Use mktime for Python 3.2 compatibility

1 files changed, 12 insertions, 1 deletions

diff --git a/tests/test_pkg_resources.py b/tests/test_pkg_resources.py
index 8ce77255..ed414096 100644
--- a/tests/test_pkg_resources.py
+++ b/tests/test_pkg_resources.py
@@ -3,6 +3,7 @@ import tempfile
 import os
 import zipfile
 import datetime
+import time
 
 import pkg_resources
 
@@ -11,6 +12,16 @@ try:
 except NameError:
     unicode = str
 
+def timestamp(dt):
+    """
+    Return a timestamp for a local, naive datetime instance.
+    """
+    try:
+        return dt.timestamp()
+    except AttributeError:
+        # Python 3.2 and earlier
+        return time.mktime(dt.timetuple())
+
 class EggRemover(unicode):
     def __call__(self):
         if self in sys.path:
@@ -77,7 +88,7 @@ class TestZipProvider(object):
         f = open(filename, 'w')
         f.write('hello, world?')
         f.close()
-        ts = self.ref_time.timestamp()
+        ts = timestamp(self.ref_time)
         os.utime(filename, (ts, ts))
         filename = zp.get_resource_filename(manager, 'data.dat')
         f = open(filename)