Add test for utils.timestamp

dbmigrator/tests/test_utils.py

@@ -6,8 +6,13 @@
 # See LICENCE.txt for details.
 # ###
 
+import datetime
 import os.path
 import unittest
+try:
+    from unittest import mock
+except ImportError:
+    import mock
 
 import psycopg2
 
@@ -203,3 +208,12 @@ def test_rollback_migration(self):
             stdout, 'Rolling back migration {} {}\n'.format(version, name))
 
         self.assertIn((version, name), after_migrations)
+
+    def test_timestamp(self):
+        from ..utils import timestamp
+
+        now = datetime.datetime(2016, 2, 28, 22, 51, 56)
+
+        with mock.patch('datetime.datetime') as mock_datetime:
+            mock_datetime.utcnow.return_value = now
+            self.assertEqual(timestamp(), '20160228225156')

setup.py

@@ -16,6 +16,9 @@
 tests_require = [
     ]
 
+if sys.version_info.major < 3:
+    tests_require.append('mock')
+
 LONG_DESC = '\n\n~~~~\n\n'.join([open('README.rst').read(),
                                  open('CHANGELOG.rst').read()])