diff --git a/linkcheck/tests/test_decorators.py b/linkcheck/tests/test_decorators.py new file mode 100644 index 00000000..d6be9e07 --- /dev/null +++ b/linkcheck/tests/test_decorators.py @@ -0,0 +1,56 @@ +# -*- coding: iso-8859-1 -*- +# Copyright (C) 2005-2006 Bastian Kleineidam +# +# This program is free software; you can redistribute it and/or modify +# it under the terms of the GNU General Public License as published by +# the Free Software Foundation; either version 2 of the License, or +# (at your option) any later version. +# +# This program is distributed in the hope that it will be useful, +# but WITHOUT ANY WARRANTY; without even the implied warranty of +# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the +# GNU General Public License for more details. +# +# You should have received a copy of the GNU General Public License +# along with this program; if not, write to the Free Software +# Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA. +""" +Test decorators. +""" + +import unittest +import time +import cStringIO as StringIO +import linkcheck.decorators + + +class TestDecorators (unittest.TestCase): + """ + Test decorators. + """ + + def test_timeit (self): + @linkcheck.decorators.timed() + def f (): + return 42 + self.assertEquals(f(), 42) + + def test_timeit2 (self): + log = StringIO.StringIO() + @linkcheck.decorators.timed(log=log, limit=0) + def f (): + time.sleep(1) + return 42 + self.assertEquals(f(), 42) + self.assert_(log.getvalue()) + + +def test_suite (): + """ + Build and return a TestSuite. + """ + return unittest.makeSuite(TestDecorators) + + +if __name__ == '__main__': + unittest.main()