diff options
Diffstat (limited to 'src/isodate/tests')
-rw-r--r-- | src/isodate/tests/__init__.py | 7 | ||||
-rw-r--r-- | src/isodate/tests/test_strf.py | 126 |
2 files changed, 131 insertions, 2 deletions
diff --git a/src/isodate/tests/__init__.py b/src/isodate/tests/__init__.py index 67512d0..1f1da2d 100644 --- a/src/isodate/tests/__init__.py +++ b/src/isodate/tests/__init__.py @@ -29,7 +29,8 @@ Collect all test suites into one TestSuite instance. ''' import unittest -from isodate.tests import test_date, test_time, test_datetime, test_duration +from isodate.tests import (test_date, test_time, test_datetime, test_duration, + test_strf) def test_suite(): ''' @@ -39,7 +40,9 @@ def test_suite(): test_date.test_suite(), test_time.test_suite(), test_datetime.test_suite(), - test_duration.test_suite()]) + test_duration.test_suite(), + test_strf.test_suite(), + ]) if __name__ == '__main__': unittest.main(defaultTest='test_suite') diff --git a/src/isodate/tests/test_strf.py b/src/isodate/tests/test_strf.py new file mode 100644 index 0000000..dc10c73 --- /dev/null +++ b/src/isodate/tests/test_strf.py @@ -0,0 +1,126 @@ +############################################################################## +# Copyright 2009, Gerhard Weis +# All rights reserved. +# +# Redistribution and use in source and binary forms, with or without +# modification, are permitted provided that the following conditions are met: +# +# * Redistributions of source code must retain the above copyright notice, +# this list of conditions and the following disclaimer. +# * Redistributions in binary form must reproduce the above copyright notice, +# this list of conditions and the following disclaimer in the documentation +# and/or other materials provided with the distribution. +# * Neither the name of the authors nor the names of its contributors +# may be used to endorse or promote products derived from this software +# without specific prior written permission. +# +# THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS" +# AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE +# IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE +# ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER OR CONTRIBUTORS BE +# LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR +# CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF +# SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS +# INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN +# CONTRACT, STRICT LIABILITY, OR TORT +############################################################################## +''' +Test cases for the isodate module. +''' +import unittest +import time +from datetime import datetime, timedelta +from isodate import strftime +from isodate import LOCAL +from isodate import DT_EXT_COMPLETE +from isodate import tzinfo + +# dict to store orig values +ORIG = {} + + +# local time zone mock function +def localtime_mock(secs): + """ + mock time.localtime so that it always returns a time_struct with tm_idst=1 + """ + tt = ORIG['localtime'](secs) + # befor 2000 everything is dst, after 2000 no dst. + if tt.tm_year < 2000: + dst = 1 + else: + dst = 0 + tt = (tt.tm_year, tt.tm_mon, tt.tm_mday, + tt.tm_hour, tt.tm_min, tt.tm_sec, + tt.tm_wday, tt.tm_yday, dst) + return time.struct_time(tt) + + +TEST_CASES = ((datetime(2012, 12, 25, 13, 30, 0, 0, LOCAL), DT_EXT_COMPLETE, + "2012-12-25T13:30:00+10:00"), + # DST ON + (datetime(1999, 12, 25, 13, 30, 0, 0, LOCAL), DT_EXT_COMPLETE, + "1999-12-25T13:30:00+11:00"), + ) + + +def create_testcase(dt, format, expectation): + """ + Create a TestCase class for a specific test. + + This allows having a separate TestCase for each test tuple from the + TEST_CASES list, so that a failed test won't stop other tests. + """ + + class TestDate(unittest.TestCase): + ''' + A test case template to test ISO date formatting. + ''' + + def test_format(self): + ''' + Take date object and create ISO string from it. + This is the reverse test to test_parse. + ''' + if expectation is None: + self.assertRaises(AttributeError, + strftime(dt, format)) + else: + self.assertEqual(strftime(dt, format), + expectation) + + return unittest.TestLoader().loadTestsFromTestCase(TestDate) + + +def setUpModule(): + ORIG['STDOFFSET'] = tzinfo.STDOFFSET + ORIG['DSTOFFSET'] = tzinfo.DSTOFFSET + ORIG['DSTDIFF'] = tzinfo.DSTDIFF + ORIG['localtime'] = time.localtime + # ovveride all saved values with fixtures. + # calculate LOCAL TZ offset, so that this test runs in every time zone + tzinfo.STDOFFSET = timedelta(seconds=36000) # assume we are in +10:00 + tzinfo.DSTOFFSET = timedelta(seconds=39600) # assume DST = +11:00 + tzinfo.DSTDIFF = tzinfo.DSTOFFSET - tzinfo.STDOFFSET + time.localtime = localtime_mock + + +def tearDownModule(): + # restore test fixtures + tzinfo.STDOFFSET = ORIG['STDOFFSET'] + tzinfo.DSTOFFSET = ORIG['DSTOFFSET'] + tzinfo.DSTDIFF = ORIG['DSTDIFF'] + time.localtime = ORIG['localtime'] + + +def test_suite(): + ''' + Construct a TestSuite instance for all test cases. + ''' + suite = unittest.TestSuite() + for dt, format, expectation in TEST_CASES: + suite.addTest(create_testcase(dt, format, expectation)) + return suite + +if __name__ == '__main__': + unittest.main(defaultTest='test_suite') |