Changeset 8185 for main/waeup.kofa/trunk/src/waeup/kofa/utils/tests

Timestamp:

16 Apr 2012, 23:17:35 (13 years ago)

Author:

uli

Message:

Add datetime/timezone helpers urgently needed.

File:

• main/waeup.kofa/trunk/src/waeup/kofa/utils/tests/test_helpers.py (modified) (3 diffs)

main/waeup.kofa/trunk/src/waeup/kofa/utils/tests/test_helpers.py

@@ -19 +19 @@
 ##
 
+import datetime
 import os
+import pytz
 import shutil
 import tempfile
@@ -262 +264 @@
         self.assertEqual(sorted(result5), ['baz'])
         return
+
+class DateTimeHelpersTestCase(unittest.TestCase):
+
+    def test_now(self):
+        tz_berlin = pytz.timezone('Europe/Berlin')
+        result1 = helpers.now()
+        result2 = helpers.now(tz_berlin)
+        self.assertEqual(result1.tzinfo, pytz.utc)
+        self.assertFalse(result2.tzinfo == pytz.utc)
+        self.assertFalse(result2.tzinfo is None)
+        return
+
+    def test_to_timezone(self):
+        fmt = '%Y-%m-%d %H:%M:%S %Z%z'
+        tz_berlin = pytz.timezone('Europe/Berlin')
+        tz_lagos = pytz.timezone('Africa/Lagos')
+        dt1 = datetime.datetime(2012, 1, 1, 0, 0)
+        dt2 = datetime.datetime(2012, 1, 1, 0, 0, tzinfo=tz_berlin)
+        dt3 = datetime.datetime(2012, 6, 1, 0, 0, tzinfo=tz_lagos)
+        dt4 = datetime.datetime(2012, 6, 1, 0, 0, tzinfo=tz_berlin)
+        result1 = helpers.to_timezone(dt1)
+        result2 = helpers.to_timezone(dt1, pytz.utc)
+        result3 = helpers.to_timezone(dt2)
+        result4 = helpers.to_timezone(dt2, tz_lagos)
+        result5 = helpers.to_timezone(dt3, tz_berlin)
+        result6 = helpers.to_timezone(dt4, tz_lagos)
+        self.assertEqual(
+            result1.strftime(fmt), '2012-01-01 00:00:00 UTC+0000')
+        self.assertEqual(
+            result2.strftime(fmt), '2012-01-01 00:00:00 UTC+0000')
+        self.assertEqual(
+            result3.strftime(fmt), '2011-12-31 23:00:00 UTC+0000')
+        self.assertEqual(
+            result4.strftime(fmt), '2012-01-01 00:00:00 WAT+0100')
+        self.assertEqual(
+            result5.strftime(fmt), '2012-06-01 01:46:00 CEST+0200')
+        self.assertEqual(
+            result6.strftime(fmt), '2012-06-01 00:00:00 WAT+0100')
+        return
+
 
 def test_suite():
@@ -275 +317 @@
         FileSizeTestCase,
         IfaceNamesTestCase,
+        DateTimeHelpersTestCase,
         ]:
         suite.addTests(